232 lines
7.7 KiB
Python
232 lines
7.7 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""
|
||
智能登录脚本 - 先静默尝试，失败再显示窗口
|
||
支持断点续传和错误恢复
|
||
"""
|
||
from DrissionPage import ChromiumPage, ChromiumOptions
|
||
import time
|
||
import json
|
||
import sys
|
||
from pathlib import Path
|
||
from datetime import datetime
|
||
|
||
SESSIONS_DIR = Path.home() / '.clawdbot' / 'browser-sessions'
|
||
ATTEMPTS_LOG = Path.home() / '.clawdbot' / 'browser-sessions' / 'attempts.json'
|
||
|
||
# 设置输出编码
|
||
if sys.platform == 'win32':
|
||
sys.stdout.reconfigure(encoding='utf-8', errors='replace')
|
||
sys.stderr.reconfigure(encoding='utf-8', errors='replace')
|
||
|
||
def log_attempt(url, success, method, notes=''):
|
||
"""记录尝试过的网站"""
|
||
SESSIONS_DIR.mkdir(parents=True, exist_ok=True)
|
||
attempts = []
|
||
if ATTEMPTS_LOG.exists():
|
||
attempts = json.loads(ATTEMPTS_LOG.read_text())
|
||
|
||
attempts.append({
|
||
'url': url,
|
||
'success': success,
|
||
'method': method, # headless / headed
|
||
'notes': notes,
|
||
'timestamp': time.time()
|
||
})
|
||
ATTEMPTS_LOG.write_text(json.dumps(attempts, indent=2, ensure_ascii=False))
|
||
|
||
def get_browser(headless=True):
|
||
"""获取浏览器实例"""
|
||
options = ChromiumOptions()
|
||
if headless:
|
||
options.headless()
|
||
options.set_argument('--disable-blink-features=AutomationControlled')
|
||
options.set_argument('--no-sandbox')
|
||
options.set_argument('--disable-dev-shm-usage')
|
||
options.set_user_agent(
|
||
'Mozilla/5.0 (Windows NT 10.0; Win64; x64) '
|
||
'AppleWebKit/537.36 (KHTML, like Gecko) '
|
||
'Chrome/120.0.0.0 Safari/537.36'
|
||
)
|
||
return ChromiumPage(options)
|
||
|
||
def save_session(page, name):
|
||
"""保存session"""
|
||
SESSIONS_DIR.mkdir(parents=True, exist_ok=True)
|
||
session_data = {
|
||
'name': name,
|
||
'url': page.url,
|
||
'title': page.title,
|
||
'cookies': dict(page.cookies()), # 用dict()转换
|
||
'timestamp': time.time()
|
||
}
|
||
path = SESSIONS_DIR / f'{name}.json'
|
||
path.write_text(json.dumps(session_data, indent=2, ensure_ascii=False))
|
||
print(f'Session saved: {path}')
|
||
return path
|
||
|
||
def load_session(page, name):
|
||
"""加载session"""
|
||
path = SESSIONS_DIR / f'{name}.json'
|
||
if not path.exists():
|
||
return False
|
||
data = json.loads(path.read_text())
|
||
for k, v in data.get('cookies', {}).items():
|
||
try:
|
||
page.set.cookies({k: v})
|
||
except:
|
||
pass
|
||
return True
|
||
|
||
def smart_login(url, session_name, account=None, password=None):
|
||
"""
|
||
智能登录 - 先静默尝试，失败再显示窗口
|
||
"""
|
||
print(f'=== 登录 {url} ===')
|
||
|
||
# 第一步：静默尝试加载已有session
|
||
print('1. 尝试静默加载已有session...')
|
||
page = get_browser(headless=True)
|
||
try:
|
||
page.get(url)
|
||
time.sleep(2)
|
||
|
||
if load_session(page, session_name):
|
||
page.refresh()
|
||
time.sleep(3)
|
||
# 检查是否登录成功（简单检查：不在登录页）
|
||
if 'login' not in page.url.lower() and 'passport' not in page.url.lower():
|
||
print('已有session有效，登录成功！')
|
||
log_attempt(url, True, 'headless', 'session_reuse')
|
||
return page
|
||
|
||
print('无有效session')
|
||
except Exception as e:
|
||
print(f'静默尝试失败: {e}')
|
||
finally:
|
||
page.quit()
|
||
|
||
# 第二步：如果有账号密码，尝试静默登录
|
||
if account and password:
|
||
print('2. 尝试静默自动登录...')
|
||
page = get_browser(headless=True)
|
||
try:
|
||
page.get(url)
|
||
time.sleep(3)
|
||
|
||
# 找输入框
|
||
inputs = page.eles('tag:input')
|
||
text_input = None
|
||
pwd_input = None
|
||
for inp in inputs:
|
||
t = inp.attr('type') or ''
|
||
if t == 'text' or t == 'tel' or t == 'email':
|
||
if not text_input:
|
||
text_input = inp
|
||
if t == 'password':
|
||
pwd_input = inp
|
||
|
||
if text_input and pwd_input:
|
||
text_input.clear()
|
||
text_input.input(account)
|
||
time.sleep(0.3)
|
||
pwd_input.clear()
|
||
pwd_input.input(password)
|
||
time.sleep(0.3)
|
||
|
||
# 找登录按钮并点击
|
||
btn = page.ele('tag:button') or page.ele('.btn') or page.ele('[type=submit]')
|
||
if btn:
|
||
try:
|
||
page.run_js('arguments[0].click()', btn)
|
||
except:
|
||
btn.click()
|
||
|
||
time.sleep(5)
|
||
|
||
# 检查是否成功
|
||
if 'login' not in page.url.lower() and 'passport' not in page.url.lower():
|
||
print('静默登录成功！')
|
||
save_session(page, session_name)
|
||
log_attempt(url, True, 'headless', 'auto_login')
|
||
return page
|
||
else:
|
||
print('可能需要验证码，切换到显示模式')
|
||
except Exception as e:
|
||
print(f'静默登录失败: {e}')
|
||
finally:
|
||
page.quit()
|
||
|
||
# 第三步：显示窗口让用户手动操作
|
||
print('3. 打开浏览器窗口，请手动登录...')
|
||
page = get_browser(headless=False)
|
||
try:
|
||
page.get(url)
|
||
|
||
# 如果有账号密码，先填入
|
||
if account and password:
|
||
time.sleep(3)
|
||
inputs = page.eles('tag:input')
|
||
for inp in inputs:
|
||
t = inp.attr('type') or ''
|
||
if t in ('text', 'tel', 'email'):
|
||
inp.clear()
|
||
inp.input(account)
|
||
elif t == 'password':
|
||
inp.clear()
|
||
inp.input(password)
|
||
|
||
# 点击登录按钮
|
||
time.sleep(0.5)
|
||
btns = page.eles('tag:button')
|
||
for btn in btns:
|
||
txt = btn.text.lower() if btn.text else ''
|
||
if '登录' in txt or 'login' in txt or '登入' in txt:
|
||
try:
|
||
page.run_js('arguments[0].click()', btn)
|
||
print('已自动点击登录按钮')
|
||
except:
|
||
pass
|
||
break
|
||
|
||
# 等待登录完成
|
||
print('等待登录完成...(检测到跳转后自动保存)')
|
||
original_url = page.url
|
||
for i in range(120): # 最多等2分钟
|
||
time.sleep(1)
|
||
current = page.url
|
||
if 'login' not in current.lower() and 'passport' not in current.lower():
|
||
if current != original_url:
|
||
print('检测到登录成功！')
|
||
time.sleep(2) # 等cookie写入
|
||
break
|
||
if i % 15 == 0 and i > 0:
|
||
print(f'等待中... {i}秒')
|
||
|
||
save_session(page, session_name)
|
||
log_attempt(url, True, 'headed', 'manual_login')
|
||
print('登录完成，session已保存')
|
||
return page
|
||
|
||
except Exception as e:
|
||
print(f'错误: {e}')
|
||
log_attempt(url, False, 'headed', str(e))
|
||
page.quit()
|
||
return None
|
||
|
||
if __name__ == '__main__':
|
||
import sys
|
||
if len(sys.argv) < 3:
|
||
print('Usage: python smart_login.py <url> <session_name> [account] [password]')
|
||
sys.exit(1)
|
||
|
||
url = sys.argv[1]
|
||
name = sys.argv[2]
|
||
account = sys.argv[3] if len(sys.argv) > 3 else None
|
||
password = sys.argv[4] if len(sys.argv) > 4 else None
|
||
|
||
page = smart_login(url, name, account, password)
|
||
if page:
|
||
print(f'当前页面: {page.title}')
|
||
print(f'URL: {page.url}')
|
||
page.quit()
|