admin
83cbfaf03f
- 修复Redis认证 (配置密码) - 启动Python管理后台 (端口9531, 15个功能开关) - 统一版本号 0.2.7 - 更新docker-compose.yml (镜像版本/Redis URL/Admin服务)
65 lines
2.2 KiB
Python
65 lines
2.2 KiB
Python
"""天翼云盘凭证管理 v1.0.0 — Cookie + 账号密码双模式"""
|
|
|
|
import re
|
|
import base64
|
|
import logging
|
|
from typing import Optional
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class Cloud189CredentialManager:
|
|
LOGIN_URL = "https://cloud.189.cn/api/portal/loginUrl.action"
|
|
SSO_URL = "https://open.e.189.cn/api/logbox/oauth2/ssoLogin.action"
|
|
|
|
def __init__(self, config):
|
|
self.config = config
|
|
self._cookie: Optional[str] = None
|
|
|
|
def validate(self) -> bool:
|
|
if self.config.cookie:
|
|
return len(self.config.cookie) >= 30
|
|
extra = self.config.extra or {}
|
|
return bool(extra.get("username") and extra.get("password"))
|
|
|
|
def get_headers(self) -> dict:
|
|
return {
|
|
"Cookie": self._cookie or self.config.cookie,
|
|
"Referer": "https://cloud.189.cn/",
|
|
}
|
|
|
|
def login_if_needed(self, session) -> bool:
|
|
"""如需账号密码登录，在此执行"""
|
|
if self.config.cookie:
|
|
self._cookie = self.config.cookie
|
|
return True
|
|
extra = self.config.extra or {}
|
|
username = extra.get("username", "")
|
|
password = extra.get("password", "")
|
|
if not username or not password:
|
|
return False
|
|
try:
|
|
logger.info("Attempting 189 cloud login...")
|
|
resp = session.get(self.LOGIN_URL, timeout=30)
|
|
data = resp.json()
|
|
login_url = data.get("toUrl", "")
|
|
session.cookies.clear()
|
|
sso_resp = session.post(
|
|
self.SSO_URL,
|
|
data={"account": username, "password": password,
|
|
"appKey": "cloud", "returnUrl": login_url},
|
|
timeout=30,
|
|
)
|
|
sso_data = sso_resp.json()
|
|
redirect_url = sso_data.get("toUrl", "")
|
|
if redirect_url:
|
|
session.get(redirect_url, timeout=30)
|
|
self._cookie = "; ".join(
|
|
f"{c.name}={c.value}" for c in session.cookies
|
|
)
|
|
logger.info("189 cloud login successful")
|
|
return bool(self._cookie)
|
|
except Exception as e:
|
|
logger.error(f"189 cloud login failed: {e}")
|
|
return False
|