Update app.py

app.py

@@ -25,18 +25,26 @@ def human_readable_size(size):
         index += 1
     return f"{size:.2f} {units[index]}"
 
+def configure_session():
+    """配置libtorrent会话"""
+    settings = lt.settings_pack()
+    settings.set_str(lt.settings_pack.listen_interfaces, '0.0.0.0:6881')
+    settings.set_int(lt.settings_ppack.alert_mask, lt.alert.category_t.all_categories)
+    return lt.session(settings)
+
 def download_task(magnet_uri, download_path, status):
     """后台下载任务"""
     try:
-        ses = lt.session()
-        ses.listen_on(6881, 6891)
+        ses = configure_session()
+        status.status = "解析磁力链接..."
         
-        params = {
-            'save_path': download_path,
-            'storage_mode': lt.storage_mode_t(2)
-        }
+        # 创建添加参数
+        params = lt.parse_magnet_uri(magnet_uri)
+        params.save_path = download_path
+        params.storage_mode = lt.storage_mode_t.storage_mode_sparse
         
-        handle = lt.add_magnet_uri(ses, magnet_uri, params)
+        # 异步添加种子
+        handle = ses.add_torrent(params)
         status.status = "获取元数据..."
         
         # 等待元数据，带超时机制
@@ -46,6 +54,7 @@ def download_task(magnet_uri, download_path, status):
             if time.time() - start > timeout:
                 raise TimeoutError("获取元数据超时")
             time.sleep(0.5)
+            ses.post_torrent_updates()
         
         torrent_info = handle.get_torrent_info()
         status.total_size = torrent_info.total_size()
@@ -55,6 +64,7 @@ def download_task(magnet_uri, download_path, status):
         start_time = time.time()
         last_downloaded = 0
         while not handle.is_seed():
+            ses.post_torrent_updates()
             s = handle.status()
             
             # 计算下载速度
@@ -122,8 +132,11 @@ def main():
             cols = st.columns(4)
             cols[0].metric("总大小", human_readable_size(status.total_size))
             cols[1].metric("已下载", human_readable_size(status.downloaded_size))
-            cols[2].metric("速度", f"{status.download_rate/1024:.2f} MB/s" if status.download_rate > 1024 else f"{status.download_rate:.2f} KB/s")
-            cols[3].metric("剩余时间", f"{status.remaining_time:.1f}秒" if status.remaining_time else "--")
+            cols[2].metric("速度", 
+                f"{status.download_rate/1024:.2f} MB/s" if status.download_rate > 1024 
+                else f"{status.download_rate:.2f} KB/s")
+            cols[3].metric("剩余时间", 
+                f"{status.remaining_time:.1f}秒" if status.remaining_time else "--")
             
             st.caption(f"状态: {status.status}")
         
@@ -139,9 +152,6 @@ def main():
                 file_name=os.path.basename(status.file_path),
                 mime="application/octet-stream"
             )
-        
-        # 可选：清理文件
-        # os.remove(status.file_path)
 
 if __name__ == "__main__":
     main()