94 lines
3.3 KiB
Python
94 lines
3.3 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""
|
|||
|
|
咪咕 miguSM 命令行工具（TgeBrowser + DrissionPage）：
|
|||
|
|
1. 传入手机号 → 新建浏览器、输入手机号、点击发送验证码
|
|||
|
|
2. 传入验证码 → 填写并提交
|
|||
|
|
|
|||
|
|
用法:
|
|||
|
|
set TGEBROWSER_API_KEY=你的API密钥
|
|||
|
|
python run_migu_cli.py --phone 13800138000
|
|||
|
|
python run_migu_cli.py --session-id <上一步返回的session_id> --code 123456
|
|||
|
|
"""
|
|||
|
|
from __future__ import annotations
|
|||
|
|
|
|||
|
|
import argparse
|
|||
|
|
import os
|
|||
|
|
import sys
|
|||
|
|
|
|||
|
|
# 确保 TgeBrowser 客户端和 migu 模块可导入
|
|||
|
|
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
|
|||
|
|
|
|||
|
|
from migu_miguSM_dp import (
|
|||
|
|
MIGU_HOME_URL,
|
|||
|
|
connect_dp_to_tgebrowser,
|
|||
|
|
connect_dp_to_ws,
|
|||
|
|
input_code_and_submit,
|
|||
|
|
input_phone_and_send_code,
|
|||
|
|
)
|
|||
|
|
from tgebrowser_client import TgeBrowserClient
|
|||
|
|
|
|||
|
|
|
|||
|
|
def cmd_submit_phone(phone: str, url: str) -> str:
|
|||
|
|
"""步骤一：输入手机号并发送验证码，返回 session_id（此处为 env_id 的字符串形式，用于后续填写验证码）。"""
|
|||
|
|
client = TgeBrowserClient()
|
|||
|
|
start_data = client.create_and_start(
|
|||
|
|
browser_name=f"miguSM_{phone[-4:]}",
|
|||
|
|
start_page_url=url,
|
|||
|
|
)
|
|||
|
|
port = start_data.get("port")
|
|||
|
|
ws = start_data.get("ws")
|
|||
|
|
env_id = start_data.get("envId")
|
|||
|
|
|
|||
|
|
if port:
|
|||
|
|
page = connect_dp_to_tgebrowser(port)
|
|||
|
|
elif ws:
|
|||
|
|
page = connect_dp_to_ws(ws)
|
|||
|
|
else:
|
|||
|
|
raise RuntimeError("TgeBrowser 未返回 port 或 ws")
|
|||
|
|
|
|||
|
|
input_phone_and_send_code(page, phone, url=url)
|
|||
|
|
print("输入电话号码成功，请查收短信验证码")
|
|||
|
|
# 保存到全局，供 submit_code 使用（CLI 模式只支持单次流程）
|
|||
|
|
cmd_submit_phone._page = page # type: ignore
|
|||
|
|
cmd_submit_phone._client = client # type: ignore
|
|||
|
|
cmd_submit_phone._env_id = env_id # type: ignore
|
|||
|
|
return str(env_id or "ok")
|
|||
|
|
|
|||
|
|
|
|||
|
|
def cmd_submit_code(code: str) -> None:
|
|||
|
|
"""步骤二：填写验证码并提交。"""
|
|||
|
|
page = getattr(cmd_submit_phone, "_page", None)
|
|||
|
|
if not page:
|
|||
|
|
raise RuntimeError("请先执行步骤一：python run_migu_cli.py --phone 13800138000")
|
|||
|
|
result = input_code_and_submit(page, code)
|
|||
|
|
print(result.get("message", "验证码已填写"))
|
|||
|
|
|
|||
|
|
|
|||
|
|
def main():
|
|||
|
|
parser = argparse.ArgumentParser(description="咪咕 miguSM 自动化 CLI")
|
|||
|
|
parser.add_argument("--phone", help="手机号码（步骤一）")
|
|||
|
|
parser.add_argument("--code", help="短信验证码（步骤二）")
|
|||
|
|
parser.add_argument("--interactive", "-i", action="store_true", help="步骤一后等待输入验证码（同一进程完成两步）")
|
|||
|
|
parser.add_argument("--url", default=MIGU_HOME_URL, help=f"目标页面，默认 {MIGU_HOME_URL}")
|
|||
|
|
args = parser.parse_args()
|
|||
|
|
|
|||
|
|
if args.phone:
|
|||
|
|
cmd_submit_phone(args.phone, args.url)
|
|||
|
|
if args.interactive:
|
|||
|
|
code = input("请输入短信验证码: ").strip()
|
|||
|
|
if code:
|
|||
|
|
cmd_submit_code(code)
|
|||
|
|
elif args.code:
|
|||
|
|
cmd_submit_code(args.code)
|
|||
|
|
else:
|
|||
|
|
parser.print_help()
|
|||
|
|
print("\n示例:")
|
|||
|
|
print(" python run_migu_cli.py --phone 13800138000 -i # 输入手机号后交互输入验证码")
|
|||
|
|
print(" python run_migu_cli.py --phone 13800138000 # 仅步骤一")
|
|||
|
|
print(" python run_migu_cli.py --code 123456 # 仅步骤二（需在 API 模式或同一进程的步骤一之后）")
|
|||
|
|
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
main()
|