55 lines
1.9 KiB
Python
55 lines
1.9 KiB
Python
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}")
|
|
|