timaa
83cbfaf03f
- 修复Redis认证 (配置密码) - 启动Python管理后台 (端口9531, 15个功能开关) - 统一版本号 0.2.7 - 更新docker-compose.yml (镜像版本/Redis URL/Admin服务)
96 lines
2.7 KiB
Python
96 lines
2.7 KiB
Python
"""
|
|
CloudSearch Transfer — UC网盘凭证管理 v1.0.0
|
|
|
|
UC网盘使用 Cookie 直传（与夸克高度相似），无需 token 刷新机制。
|
|
验证方式：检查 Cookie 字符串长度是否 >= 50。
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import logging
|
|
from typing import Dict
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class UcCredentialManager:
|
|
"""UC 网盘凭证管理器。
|
|
|
|
UC 网盘的转存 API 直接从 Cookie 中读取认证信息，
|
|
与夸克网盘机制完全一致，只是 API 域名不同（pc-api.uc.cn）。
|
|
|
|
Attributes:
|
|
cookie: 存储的 UC Cookie 字符串。
|
|
"""
|
|
|
|
# UC Cookie 最小长度阈值（与夸克一致）
|
|
MIN_COOKIE_LENGTH: int = 50
|
|
|
|
# UC 网盘 Referer
|
|
REFERER: str = "https://drive.uc.cn/"
|
|
|
|
def __init__(self, cookie: str = "") -> None:
|
|
"""初始化凭证管理器。
|
|
|
|
Args:
|
|
cookie: UC 网盘的 Cookie 字符串。
|
|
"""
|
|
self.cookie: str = cookie
|
|
|
|
def validate(self) -> bool:
|
|
"""验证 Cookie 是否满足最小长度要求。
|
|
|
|
Returns:
|
|
True 表示 Cookie 长度 >= MIN_COOKIE_LENGTH，否则为 False。
|
|
"""
|
|
if not self.cookie:
|
|
logger.warning("[UcCredential] Cookie is empty")
|
|
return False
|
|
|
|
valid = len(self.cookie) >= self.MIN_COOKIE_LENGTH
|
|
if not valid:
|
|
logger.warning(
|
|
"[UcCredential] Cookie too short: len=%d, min=%d",
|
|
len(self.cookie),
|
|
self.MIN_COOKIE_LENGTH,
|
|
)
|
|
return valid
|
|
|
|
def is_valid(self) -> bool:
|
|
"""validate() 的别名，便于适配器层调用。"""
|
|
return self.validate()
|
|
|
|
def get_headers(self) -> Dict[str, str]:
|
|
"""构建带 Cookie 认证的 HTTP 请求头。
|
|
|
|
UC API 需要在每次请求头中携带完整的 Cookie 字符串，
|
|
以及 Referer: https://drive.uc.cn/。
|
|
|
|
Returns:
|
|
包含 Cookie 和 Referer 字段的请求头字典。
|
|
Cookie 无效时仍返回空字典。
|
|
"""
|
|
if not self.validate():
|
|
logger.warning("[UcCredential] Cannot build headers: cookie invalid")
|
|
return {}
|
|
|
|
return {
|
|
"Cookie": self.cookie,
|
|
"Referer": self.REFERER,
|
|
}
|
|
|
|
def update_cookie(self, cookie: str) -> None:
|
|
"""更新 Cookie 字符串（用于手动刷新场景）。
|
|
|
|
Args:
|
|
cookie: 新的 Cookie 字符串。
|
|
"""
|
|
self.cookie = cookie
|
|
logger.info("[UcCredential] Cookie updated, new length=%d", len(cookie))
|
|
|
|
def __repr__(self) -> str:
|
|
return (
|
|
f"UcCredentialManager(cookie_len={len(self.cookie) if self.cookie else 0}, "
|
|
f"valid={self.validate()})"
|
|
)
|