add downloader info

app/modules/transmission/__init__.py

@@ -1,6 +1,7 @@
 from pathlib import Path
 from typing import Set, Tuple, Optional, Union, List
 
+from app import schemas
 from app.core.config import settings
 from app.core.metainfo import MetaInfo
 from app.log import logger
@@ -168,3 +169,15 @@ class TransmissionModule(_ModuleBase):
         :return: bool
         """
         return self.transmission.start_torrents(ids=hashs)
+
+    def downloader_info(self) -> schemas.DownloaderInfo:
+        """
+        下载器信息
+        """
+        info = self.transmission.transfer_info()
+        return schemas.DownloaderInfo(
+            download_speed=info.download_speed,
+            upload_speed=info.upload_speed,
+            download_size=info.current_stats.downloaded_bytes,
+            upload_size=info.current_stats.uploaded_bytes
+        )

app/modules/transmission/transmission.py

@@ -2,6 +2,7 @@ from typing import Optional, Union, Tuple, List
 
 import transmission_rpc
 from transmission_rpc import Client, Torrent, File
+from transmission_rpc.session import SessionStats
 
 from app.core.config import settings
 from app.log import logger
@@ -214,3 +215,13 @@ class Transmission(metaclass=Singleton):
         except Exception as err:
             logger.error(f"设置下载文件状态出错：{err}")
             return False
+
+    def transfer_info(self) -> Optional[SessionStats]:
+        """
+        获取传输信息
+        """
+        try:
+            return self.trc.session_stats()
+        except Exception as err:
+            logger.error(f"获取传输信息出错：{err}")
+            return None