Merge branch 'jxxghp:main' into main

app/api/endpoints/message.py

@@ -49,7 +49,7 @@ def web_message(text: str, current_user: User = Depends(get_current_active_super
     """
     MessageChain().handle_message(
         channel=MessageChannel.Web,
-        userid=current_user.id,
+        userid=current_user.name,
         username=current_user.name,
         text=text
     )

app/api/servcookie.py

@@ -14,6 +14,8 @@ from app.utils.common import get_decrypted_cookie_data
 
 class GzipRequest(Request):
 
+    _body: bytes = b""
+
     async def body(self) -> bytes:
         if not hasattr(self, "_body"):
             body = await super().body()
@@ -66,10 +68,9 @@ async def update_cookie(req: schemas.CookieData):
     content = json.dumps({"encrypted": req.encrypted})
     with open(file_path, encoding="utf-8", mode="w") as file:
         file.write(content)
-    read_content = None
     with open(file_path, encoding="utf-8", mode="r") as file:
         read_content = file.read()
-    if (read_content == content):
+    if read_content == content:
         return {"action": "done"}
     else:
         return {"action": "error"}

app/chain/__init__.py

@@ -437,12 +437,13 @@ class ChainBase(metaclass=ABCMeta):
         :param medias:  媒体列表
         :return: 成功或失败
         """
-        self.messagehelper.put(message, role="user")
+        note_list = [media.to_dict() for media in medias]
+        self.messagehelper.put(message, role="user", note=note_list)
         self.messageoper.add(channel=message.channel, mtype=message.mtype,
                              title=message.title, text=message.text,
                              image=message.image, link=message.link,
                              userid=message.userid, action=1,
-                             note=[media.to_dict() for media in medias])
+                             note=note_list)
         return self.run_module("post_medias_message", message=message, medias=medias)
 
     def post_torrents_message(self, message: Notification, torrents: List[Context]) -> Optional[bool]:
@@ -452,12 +453,13 @@ class ChainBase(metaclass=ABCMeta):
         :param torrents:  种子列表
         :return: 成功或失败
         """
-        self.messagehelper.put(message, role="user")
+        note_list = [torrent.torrent_info.to_dict() for torrent in torrents]
+        self.messagehelper.put(message, role="user", note=note_list)
         self.messageoper.add(channel=message.channel, mtype=message.mtype,
                              title=message.title, text=message.text,
                              image=message.image, link=message.link,
                              userid=message.userid, action=1,
-                             note=[torrent.to_dict() for torrent in torrents])
+                             note=note_list)
         return self.run_module("post_torrents_message", message=message, torrents=torrents)
 
     def scrape_metadata(self, path: Path, mediainfo: MediaInfo, transfer_type: str,

app/core/context.py

@@ -615,6 +615,7 @@ class MediaInfo:
         """
         返回字典
         """
+        self.clear()
         dicts = asdict(self)
         dicts["type"] = self.type.value if self.type else None
         dicts["detail_link"] = self.detail_link

app/db/message_oper.py

@@ -50,7 +50,7 @@ class MessageOper(DbOper):
             "userid": userid,
             "action": action,
             "reg_time": time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()),
-            "note": json.dumps(note or {})
+            "note": json.dumps(note) if note else ''
         })
         Message(**kwargs).create(self._db)
 

app/helper/message.py

@@ -14,11 +14,12 @@ class MessageHelper(metaclass=Singleton):
         self.sys_queue = queue.Queue()
         self.user_queue = queue.Queue()
 
-    def put(self, message: Any, role: str = "sys"):
+    def put(self, message: Any, role: str = "sys", note: dict = None):
         """
         存消息
         :param message: 消息
         :param role: 消息通道 sys/user
+        :param note: 附件json
         """
         if role == "sys":
             self.sys_queue.put(message)
@@ -28,6 +29,7 @@ class MessageHelper(metaclass=Singleton):
             elif hasattr(message, "to_dict"):
                 content = message.to_dict()
                 content['date'] = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
+                content['note'] = json.dumps(note) if note else None
                 self.user_queue.put(json.dumps(content))
 
     def get(self, role: str = "sys") -> Optional[str]:

app/helper/resource.py

@@ -55,7 +55,7 @@ class ResourceHelper(metaclass=Singleton):
             target = resource.get("target")
             version = resource.get("version")
             # 判断平台
-            if platform and platform != SystemUtils.platform:
+            if platform and platform != SystemUtils.platform():
                 continue
             # 判断本地是否存在
             local_path = self._base_dir / target / rname

app/utils/system.py

@@ -71,8 +71,8 @@ class SystemUtils:
         """
         return True if platform.machine() == 'aarch64' else False
 
-    @property
-    def platform(self) -> str:
+    @staticmethod
+    def platform() -> str:
         """
         获取系统平台
         """