init

backend/utils/douban_utils.py

@@ -0,0 +1,54 @@
+import requests
+import time
+import os
+
+class DoubanManager:
+    def __init__(self, config):
+        self.cookie = config.get('cookie', '')
+
+    def test_connection(self):
+        if not self.cookie:
+            return False, "未配置 Cookie"
+        # 简单测试，可以访问一个需要登录的页面
+        headers = {'Cookie': self.cookie}
+        try:
+            res = requests.get('https://www.douban.com/mine/', headers=headers, timeout=10)
+            if '登录' in res.text:
+                 return False, "Cookie 已失效"
+            return True, "Cookie 有效"
+        except Exception as e:
+            return False, f"网络请求失败: {e}"
+
+    def scrape(self, media_path):
+        """
+        模拟耗时的刮削任务
+        :param media_path: 要刮削的媒体文件夹路径
+        """
+        print(f"开始刮削: {media_path}")
+        # 1. 提取剧名
+        series_name = os.path.basename(media_path)
+        print(f"提取剧名: {series_name}")
+        
+        # 2. 在豆瓣搜索 (模拟)
+        time.sleep(5) 
+        print("模拟豆瓣搜索中...")
+        
+        # 3. 获取信息并写入 NFO 和海报 (模拟)
+        time.sleep(5)
+        nfo_path = os.path.join(media_path, 'movie.nfo')
+        poster_path = os.path.join(media_path, 'poster.jpg')
+        
+        with open(nfo_path, 'w', encoding='utf-8') as f:
+            f.write(f"<movie><title>{series_name}</title><plot>这是一个模拟刮削的剧情简介。</plot></movie>")
+        
+        # 模拟下载海报
+        try:
+            # 使用一个占位图
+            img_data = requests.get("https://via.placeholder.com/300x450.png?text=Poster").content
+            with open(poster_path, 'wb') as handler:
+                handler.write(img_data)
+        except Exception as e:
+            print(f"下载海报失败: {e}")
+
+        print(f"刮削完成: {media_path}")
+