GithubBackup/MoviePilot
1
0
Fork 0
mirror of https://github.com/jxxghp/MoviePilot.git synced 2026-09-08 17:08:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix：memory use

Browse Source
This commit is contained in:
jxxghp
2025-06-05 08:18:26 +08:00
parent 99334795b6
commit faaee09827
1 changed files with 23 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
app/chain/message.py
+23 -7
View File
@@ -1,4 +1,4 @@
import copy import gc
import re import re
from typing import Any, Optional, Dict, Union from typing import Any, Optional, Dict, Union
@@ -173,7 +173,7 @@ class MessageChain(ChainBase):
elif text.isdigit(): elif text.isdigit():
# 用户选择了具体的条目 # 用户选择了具体的条目
# 缓存 # 缓存
cache_data: dict = user_cache.get(userid) cache_data: dict = user_cache.get(userid).copy()
# 选择项目 # 选择项目
if not cache_data \ if not cache_data \
or not cache_data.get('items') \ or not cache_data.get('items') \
@@ -186,7 +186,7 @@ class MessageChain(ChainBase):
# 缓存类型 # 缓存类型
cache_type: str = cache_data.get('type') cache_type: str = cache_data.get('type')
# 缓存列表 # 缓存列表
cache_list: list = copy.deepcopy(cache_data.get('items')) cache_list: list = cache_data.get('items').copy()
# 选择 # 选择
if cache_type in ["Search", "ReSearch"]: if cache_type in ["Search", "ReSearch"]:
# 当前媒体信息 # 当前媒体信息
@@ -276,6 +276,10 @@ class MessageChain(ChainBase):
userid=userid, userid=userid,
total=len(contexts)) total=len(contexts))
# 清理内存
del cache_list
del cache_data
elif cache_type in ["Subscribe", "ReSubscribe"]: elif cache_type in ["Subscribe", "ReSubscribe"]:
# 订阅或洗版媒体 # 订阅或洗版媒体
mediainfo: MediaInfo = cache_list[_choice] mediainfo: MediaInfo = cache_list[_choice]
@@ -325,7 +329,7 @@ class MessageChain(ChainBase):
elif text.lower() == "p": elif text.lower() == "p":
# 上一页 # 上一页
cache_data: dict = user_cache.get(userid) cache_data: dict = user_cache.get(userid).copy()
if not cache_data: if not cache_data:
# 没有缓存 # 没有缓存
self.post_message(Notification( self.post_message(Notification(
@@ -341,7 +345,7 @@ class MessageChain(ChainBase):
_current_page -= 1 _current_page -= 1
cache_type: str = cache_data.get('type') cache_type: str = cache_data.get('type')
# 产生副本，避免修改原值 # 产生副本，避免修改原值
cache_list: list = copy.deepcopy(cache_data.get('items')) cache_list: list = cache_data.get('items').copy()
if _current_page == 0: if _current_page == 0:
start = 0 start = 0
end = self._page_size end = self._page_size
@@ -364,10 +368,13 @@ class MessageChain(ChainBase):
items=cache_list[start:end], items=cache_list[start:end],
userid=userid, userid=userid,
total=len(cache_list)) total=len(cache_list))
# 清理内存
del cache_list
del cache_data
elif text.lower() == "n": elif text.lower() == "n":
# 下一页 # 下一页
cache_data: dict = user_cache.get(userid) cache_data: dict = user_cache.get(userid).copy()
if not cache_data: if not cache_data:
# 没有缓存 # 没有缓存
self.post_message(Notification( self.post_message(Notification(
@@ -375,7 +382,7 @@ class MessageChain(ChainBase):
return return
cache_type: str = cache_data.get('type') cache_type: str = cache_data.get('type')
# 产生副本，避免修改原值 # 产生副本，避免修改原值
cache_list: list = copy.deepcopy(cache_data.get('items')) cache_list: list = cache_data.get('items').copy()
total = len(cache_list) total = len(cache_list)
# 加一页 # 加一页
cache_list = cache_list[ cache_list = cache_list[
@@ -400,6 +407,9 @@ class MessageChain(ChainBase):
source=source, source=source,
title=_current_meta.name, title=_current_meta.name,
items=cache_list, userid=userid, total=total) items=cache_list, userid=userid, total=total)
# 清理内存
del cache_list
del cache_data
else: else:
# 搜索或订阅 # 搜索或订阅
@@ -475,6 +485,12 @@ class MessageChain(ChainBase):
# 保存缓存 # 保存缓存
self.save_cache(user_cache, self._cache_file) self.save_cache(user_cache, self._cache_file)
# 清理内存
user_cache.clear()
del user_cache
gc.collect()
def __auto_download(self, channel: MessageChannel, source: str, cache_list: list[Context], def __auto_download(self, channel: MessageChannel, source: str, cache_list: list[Context],
userid: Union[str, int], username: str, userid: Union[str, int], username: str,
no_exists: Optional[Dict[Union[int, str], Dict[int, NotExistMediaInfo]]] = None): no_exists: Optional[Dict[Union[int, str], Dict[int, NotExistMediaInfo]]] = None):