50 lines
1.7 KiB
Python
50 lines
1.7 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
from __future__ import annotations
|
||
|
||
import argparse
|
||
|
||
from DrissionPage import ChromiumOptions, ChromiumPage
|
||
|
||
from tyyp_app.api import app
|
||
from tyyp_app.automation import input_code, submit_phone
|
||
from tyyp_app.config import DEFAULT_PHONE, DEFAULT_TARGET_URL
|
||
|
||
|
||
def build_parser() -> argparse.ArgumentParser:
|
||
parser = argparse.ArgumentParser(description="天翼云盘订购页自动化")
|
||
parser.add_argument("--api", action="store_true", help="以 FastAPI 服务模式启动")
|
||
parser.add_argument("--host", default="0.0.0.0", help="API 监听地址（--api 模式）")
|
||
parser.add_argument("--api-port", type=int, default=8000, help="API 监听端口（--api 模式）")
|
||
parser.add_argument("--phone", default=DEFAULT_PHONE, help="手机号码（默认用代码中的 DEFAULT_PHONE）")
|
||
parser.add_argument("--url", default=DEFAULT_TARGET_URL, help="目标页面 URL")
|
||
parser.add_argument("--port", type=int, default=0, help="连接已有浏览器端口，0 表示新建")
|
||
return parser
|
||
|
||
|
||
def run_once(phone: str, url: str, port: int = 0) -> None:
|
||
if port:
|
||
co = ChromiumOptions().set_local_port(port=port)
|
||
page = ChromiumPage(addr_or_opts=co)
|
||
else:
|
||
page = ChromiumPage()
|
||
|
||
print(f"打开页面: {url}，手机号: {phone}")
|
||
body = submit_phone(page=page, phone=phone, url=url)
|
||
print("hn_userEquitys/getYanZhenMa/v2 响应:")
|
||
print(body)
|
||
|
||
input_code(page=page, code=123123)
|
||
|
||
|
||
def main() -> None:
|
||
args = build_parser().parse_args()
|
||
|
||
if args.api:
|
||
import uvicorn
|
||
|
||
uvicorn.run(app, host=args.host, port=args.api_port)
|
||
return
|
||
|
||
run_once(phone=args.phone, url=args.url, port=args.port)
|