admin
83cbfaf03f
- 修复Redis认证 (配置密码) - 启动Python管理后台 (端口9531, 15个功能开关) - 统一版本号 0.2.7 - 更新docker-compose.yml (镜像版本/Redis URL/Admin服务)
90 lines
2.6 KiB
Python
90 lines
2.6 KiB
Python
"""
|
||
CloudSearch Transfer — 夸克网盘凭证管理 v1.0.0
|
||
|
||
夸克网盘使用 Cookie 直传，无需 token 刷新机制。
|
||
验证方式：检查 Cookie 字符串长度是否 >= 50。
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
import logging
|
||
from typing import Dict
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
class QuarkCredentialManager:
|
||
"""夸克网盘凭证管理器。
|
||
|
||
夸克网盘的上传/转存 API 直接从 Cookie 中读取认证信息，
|
||
无需 OAuth 或 refresh_token 刷新流程。
|
||
|
||
Attributes:
|
||
cookie: 存储的夸克 Cookie 字符串。
|
||
"""
|
||
|
||
# 夸克 Cookie 最小长度阈值（经验值，正常 Cookie 远超此长度）
|
||
MIN_COOKIE_LENGTH: int = 50
|
||
|
||
def __init__(self, cookie: str = "") -> None:
|
||
"""初始化凭证管理器。
|
||
|
||
Args:
|
||
cookie: 夸克网盘的 Cookie 字符串。
|
||
"""
|
||
self.cookie: str = cookie
|
||
|
||
def validate(self) -> bool:
|
||
"""验证 Cookie 是否满足最小长度要求。
|
||
|
||
Returns:
|
||
True 表示 Cookie 长度 >= MIN_COOKIE_LENGTH，否则为 False。
|
||
"""
|
||
if not self.cookie:
|
||
logger.warning("[QuarkCredential] Cookie is empty")
|
||
return False
|
||
|
||
valid = len(self.cookie) >= self.MIN_COOKIE_LENGTH
|
||
if not valid:
|
||
logger.warning(
|
||
"[QuarkCredential] 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 请求头。
|
||
|
||
夸克 API 需要在每次请求头中携带完整的 Cookie 字符串。
|
||
|
||
Returns:
|
||
包含 Cookie 字段的请求头字典。Cookie 无效时仍返回空字典。
|
||
"""
|
||
if not self.validate():
|
||
logger.warning("[QuarkCredential] Cannot build headers: cookie invalid")
|
||
return {}
|
||
|
||
return {
|
||
"Cookie": self.cookie,
|
||
}
|
||
|
||
def update_cookie(self, cookie: str) -> None:
|
||
"""更新 Cookie 字符串（用于手动刷新场景）。
|
||
|
||
Args:
|
||
cookie: 新的 Cookie 字符串。
|
||
"""
|
||
self.cookie = cookie
|
||
logger.info("[QuarkCredential] Cookie updated, new length=%d", len(cookie))
|
||
|
||
def __repr__(self) -> str:
|
||
return (
|
||
f"QuarkCredentialManager(cookie_len={len(self.cookie) if self.cookie else 0}, "
|
||
f"valid={self.validate()})"
|
||
)
|