Update app.py

app.py

@@ -26,26 +26,34 @@ def human_readable_size(size):
     return f"{size:.2f} {units[index]}"
 
 def configure_session():
-    """兼容不同版本的会话配置"""
-    try:
-        # 新版配置方式
+    """精确版本检测的会话配置"""
+    if hasattr(lt, 'settings_pack'):
+        # 新版本配置方式 (libtorrent 1.2.x+)
         settings = lt.settings_pack()
         settings.set_str(lt.settings_pack.listen_interfaces, '0.0.0.0:6881')
-        ses = lt.session(settings)
-    except AttributeError:
-        # 旧版配置方式
+        return lt.session(settings)
+    else:
+        # 旧版本配置方式 (libtorrent <1.2.x)
         ses = lt.session()
-        ses.listen_on(6881, 6891)
-    return ses
+        if hasattr(ses, 'listen_on'):
+            ses.listen_on(6881, 6891)
+        return ses
 
-def parse_magnet_compat(magnet_uri):
-    """兼容不同版本的磁力解析方法"""
-    try:
-        return lt.parse_magnet_uri(magnet_uri)
-    except AttributeError:
-        # 旧版处理方式
-        params = {'save_path': '.', 'storage_mode': lt.storage_mode_t(2)}
-        return (magnet_uri, params)
+def add_torrent_compat(ses, magnet_uri, download_path):
+    """精确版本检测的种子添加方法"""
+    if hasattr(lt, 'parse_magnet_uri'):
+        # 新版本添加方式
+        params = lt.parse_magnet_uri(magnet_uri)
+        params.save_path = download_path
+        if hasattr(lt.storage_mode_t, 'storage_mode_sparse'):
+            params.storage_mode = lt.storage_mode_t.storage_mode_sparse
+        return ses.add_torrent(params)
+    else:
+        # 旧版本添加方式
+        return lt.add_magnet_uri(ses, magnet_uri, {
+            'save_path': download_path,
+            'storage_mode': lt.storage_mode_t(2)
+        })
 
 def download_task(magnet_uri, download_path, status):
     """后台下载任务"""
@@ -53,29 +61,19 @@ def download_task(magnet_uri, download_path, status):
         ses = configure_session()
         status.status = "解析磁力链接..."
         
-        # 兼容解析磁力链接
-        try:
-            params = lt.parse_magnet_uri(magnet_uri)
-            params.save_path = download_path
-            params.storage_mode = lt.storage_mode_t.storage_mode_sparse
-            handle = ses.add_torrent(params)
-        except:
-            # 旧版处理方式
-            handle = lt.add_magnet_uri(ses, magnet_uri, {'save_path': download_path})
-        
+        # 添加种子
+        handle = add_torrent_compat(ses, magnet_uri, download_path)
         status.status = "获取元数据..."
         
         # 等待元数据，带超时机制
-        timeout = 30  # 30秒超时
+        timeout = 30
         start = time.time()
         while not handle.has_metadata():
             if time.time() - start > timeout:
                 raise TimeoutError("获取元数据超时")
             time.sleep(0.5)
-            try:
-                ses.post_torrent_updates()  # 新版状态更新
-            except:
-                pass
+            if hasattr(ses, 'post_torrent_updates'):
+                ses.post_torrent_updates()
         
         torrent_info = handle.get_torrent_info()
         status.total_size = torrent_info.total_size()
@@ -85,10 +83,8 @@ def download_task(magnet_uri, download_path, status):
         start_time = time.time()
         last_downloaded = 0
         while not handle.is_seed():
-            try:
-                ses.post_torrent_updates()  # 新版需要状态更新
-            except:
-                pass
+            if hasattr(ses, 'post_torrent_updates'):
+                ses.post_torrent_updates()
             
             s = handle.status()
             
@@ -118,16 +114,13 @@ def download_task(magnet_uri, download_path, status):
 def main():
     st.title("🕹️ 磁力链接下载器")
     
-    # 初始化session状态
     if 'download_status' not in st.session_state:
         st.session_state.download_status = DownloadStatus()
     
-    # 输入区域
     with st.form("magnet_form"):
         magnet_uri = st.text_input("磁力链接：", placeholder="magnet:?xt=urn:...")
         submitted = st.form_submit_button("开始下载")
     
-    # 创建下载目录
     download_path = Path("downloads")
     download_path.mkdir(exist_ok=True)
     
@@ -136,7 +129,6 @@ def main():
             st.error("无效的磁力链接格式")
             return
         
-        # 启动下载线程
         st.session_state.download_status = DownloadStatus()
         thread = threading.Thread(
             target=download_task,
@@ -144,16 +136,13 @@ def main():
         )
         thread.start()
     
-    # 实时状态显示
     status = st.session_state.download_status
     status_container = st.empty()
     
     while status.status not in ["下载完成", "错误"] and status.status != "等待中":
         with status_container.container():
-            # 进度条
             st.progress(status.progress, text=f"进度: {status.progress*100:.1f}%")
             
-            # 下载信息
             cols = st.columns(4)
             cols[0].metric("总大小", human_readable_size(status.total_size) if status.total_size > 0 else "--")
             cols[1].metric("已下载", human_readable_size(status.downloaded_size))
@@ -167,7 +156,6 @@ def main():
         
         time.sleep(0.5)
     
-    # 下载完成后的处理
     if status.status == "下载完成":
         st.success("🎉 下载完成！")
         with open(status.file_path, "rb") as f: