codfrm/MoviePilot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix 115 pan

Browse Source
This commit is contained in:
jxxghp
2024-06-19 13:02:04 +08:00
parent 7103b0334a
commit 060e2f225c
5 changed files with 196 additions and 103 deletions
Download Patch File Download Diff File Expand all files Collapse all files

164
app/helper/u115.py
View File

@ -1,10 +1,12 @@
from typing import Optional, Tuple, List
import base64
from typing import Optional, Tuple, Generator
import py115
from py115 import Cloud
from py115.types import LoginTarget, QrcodeSession, QrcodeStatus
from py115.types import LoginTarget, QrcodeSession, QrcodeStatus, Credential, File, DownloadTicket
from app.db.systemconfig_oper import SystemConfigOper
from app.log import logger
from app.schemas.types import SystemConfigKey
from app.utils.singleton import Singleton
from app.utils.system import SystemUtils
@ -15,32 +17,52 @@ class U115Helper(metaclass=Singleton):
115相关操作
"""
cloud: Cloud = None
session: QrcodeSession = None
cloud: Optional[Cloud] = None
_session: QrcodeSession = None
def __init__(self):
self.systemconfig = SystemConfigOper()
@property
def cookies(self):
def __init_cloud(self) -> bool:
"""
获取115认证参数并初始化参数格式
初始化Cloud
"""
return self.systemconfig.get(SystemConfigKey.User115Params) or {}
credential = self.credential
if not credential:
logger.warn("115未登录，请先登录！")
return False
try:
if not self.cloud:
self.cloud = py115.connect(credential)
except Exception as err:
logger.error(f"115连接失败，请重新扫码登录：{str(err)}")
self.clear_credential()
return False
return True
def save_credentail(self, cookies: dict):
@property
def credential(self) -> Optional[Credential]:
"""
获取已保存的115认证参数
"""
cookie_dict = self.systemconfig.get(SystemConfigKey.User115Params)
if not cookie_dict:
return None
return Credential.from_dict(cookie_dict)
def save_credentail(self, credential: Credential):
"""
设置115认证参数
"""
self.systemconfig.set(SystemConfigKey.User115Params, cookies)
self.systemconfig.set(SystemConfigKey.User115Params, credential.to_dict())
def clear_params(self):
def clear_credential(self):
"""
清除115认证参数
"""
self.systemconfig.delete(SystemConfigKey.User115Params)
def generate_qrcode(self) -> Optional[Tuple[dict, str]]:
def generate_qrcode(self) -> Optional[str]:
"""
生成二维码
"""
@ -58,67 +80,99 @@ class U115Helper(metaclass=Singleton):
try:
self.cloud = py115.connect()
self.session = self.cloud.qrcode_login(__get_os)
return self.session.image_data, ""
self._session = self.cloud.qrcode_login(__get_os())
image_bin = self._session.image_data
if not image_bin:
logger.warn("115生成二维码失败：未获取到二维码数据！")
return None
# 转换为base64图片格式
image_base64 = base64.b64encode(image_bin).decode()
return f"data:image/png;base64,{image_base64}"
except Exception as e:
return None, f"115生成二维码失败：{str(e)}"
logger.warn(f"115生成二维码失败：{str(e)}")
return None
def check_login(self, ck: str, t: str) -> Optional[Tuple[dict, str]]:
def check_login(self) -> Optional[Tuple[dict, str]]:
"""
二维码登录确认
"""
if not self.session:
return None, "请先生成二维码！"
if not self._session:
return {}, "请先生成二维码！"
try:
status = self.cloud.qrcode_poll(self.session)
if not self.cloud:
return {}, "请先生成二维码！"
status = self.cloud.qrcode_poll(self._session)
if status == QrcodeStatus.Done:
# 确认完成，保存认证信息
self.save_credentail(self.cloud.export_credentail())
result = {
"status": 1,
"tip": "登录成功！"
}
elif status == QrcodeStatus.Waiting:
return {
result = {
"status": 0,
"tip": "等待扫码确认..."
}, ""
"tip": "请使用微信或115客户端扫码"
}
elif status == QrcodeStatus.Expired:
return {
result = {
"status": -1,
"tip": "二维码已过期，请重新刷新！"
}, ""
}
self.cloud = None
elif status == QrcodeStatus.Failed:
return {
result = {
"status": -2,
"tip": "登录失败，请重试！"
}, ""
return None, "登录确认失败！"
}
self.cloud = None
else:
result = {
"status": -3,
"tip": "未知错误，请重试！"
}
self.cloud = None
return result, ""
except Exception as e:
return None, f"115登录确认失败：{str(e)}"
return {}, f"115登录确认失败：{str(e)}"
def list_files(self, parent_file_id: str = '0') -> List[dict]:
def list_files(self, parent_file_id: str = '0') -> Optional[Generator[File, None, None]]:
"""
浏览文件
"""
cookies = self.cookies
if not cookies:
return []
return self.cloud.storage().list(dir_id=parent_file_id)
if not self.__init_cloud():
return None
try:
return self.cloud.storage().list(dir_id=parent_file_id)
except Exception as e:
logger.error(f"浏览115文件失败：{str(e)}")
return None
def create_folder(self, parent_file_id: str, name: str) -> bool:
"""
创建目录
"""
cookies = self.cookies
if not cookies:
if not self.__init_cloud():
return False
return self.cloud.storage().make_dir(parent_file_id, name)
try:
self.cloud.storage().make_dir(parent_file_id, name)
return True
except Exception as e:
logger.error(f"创建115目录失败：{str(e)}")
return False
def delete_file(self, file_id: str) -> bool:
"""
删除文件
"""
cookies = self.cookies
if not cookies:
if not self.__init_cloud():
return False
return self.cloud.storage().delete(file_id)
try:
self.cloud.storage().delete(file_id)
return True
except Exception as e:
logger.error(f"删除115文件失败：{str(e)}")
return False
def get_file_detail(self, file_id: str) -> Optional[dict]:
"""
@ -130,13 +184,35 @@ class U115Helper(metaclass=Singleton):
"""
重命名文件
"""
cookies = self.cookies
if not cookies:
if not self.__init_cloud():
return False
return self.cloud.storage().rename(file_id, name)
try:
self.cloud.storage().rename(file_id, name)
return True
except Exception as e:
logger.error(f"重命名115文件失败：{str(e)}")
return False
def get_download_url(self, file_id: str) -> Optional[str]:
def download(self, pickcode: str) -> Optional[DownloadTicket]:
"""
获取下载链接
"""
pass
if not self.__init_cloud():
return None
try:
return self.cloud.storage().request_download(pickcode)
except Exception as e:
logger.error(f"115下载失败：{str(e)}")
return None
def get_storage(self) -> Optional[Tuple[int, int]]:
"""
获取存储空间
"""
if not self.__init_cloud():
return None
try:
return self.cloud.storage().space()
except Exception as e:
logger.error(f"获取115存储空间失败：{str(e)}")
return None