99 lines
3.1 KiB
Python
99 lines
3.1 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
"""
|
||
天翼订购页 1.html 接口：
|
||
- POST /api/submit_phone 传电话号码，完成滑块后返回 hn_userEquitys/getYanZhenMa/v2 响应
|
||
- POST /api/submit_code 传验证码，点击确认订购
|
||
"""
|
||
from __future__ import annotations
|
||
|
||
import threading
|
||
from typing import Any, Optional
|
||
|
||
from fastapi import FastAPI, HTTPException
|
||
from pydantic import BaseModel, Field
|
||
|
||
from tyyp_service import submit_phone as do_submit_phone, submit_code as do_submit_code
|
||
|
||
app = FastAPI(title="天翼订购页 API", description="1.html 手机号与验证码提交流程")
|
||
|
||
# 全局页面实例，用于 submit_phone 与 submit_code 之间复用
|
||
_page: Optional[Any] = None
|
||
_page_lock = threading.Lock()
|
||
|
||
|
||
def _get_or_create_page():
|
||
global _page
|
||
with _page_lock:
|
||
if _page is None:
|
||
from DrissionPage import ChromiumPage
|
||
_page = ChromiumPage()
|
||
return _page
|
||
|
||
|
||
class SubmitPhoneRequest(BaseModel):
|
||
phone: str = Field(..., description="手机号码")
|
||
port: int = Field(0, description="连接已有浏览器端口，0 表示新建")
|
||
url: str = Field("http://yscnb.com/tyyp/1.html", description="目标页面 URL")
|
||
|
||
|
||
class SubmitPhoneResponse(BaseModel):
|
||
success: bool = True
|
||
data: Any = Field(..., description="hn_userEquitys/getYanZhenMa/v2 接口返回的响应体")
|
||
|
||
|
||
class SubmitCodeRequest(BaseModel):
|
||
code: str = Field(..., description="短信验证码")
|
||
|
||
|
||
class SubmitCodeResponse(BaseModel):
|
||
success: bool = True
|
||
message: str = ""
|
||
|
||
|
||
@app.post("/api/submit_phone", response_model=SubmitPhoneResponse)
|
||
def api_submit_phone(req: SubmitPhoneRequest):
|
||
"""
|
||
提交手机号：填写手机号、点击获取验证码、执行滑块验证，
|
||
监听 hn_userEquitys/getYanZhenMa/v2，将其响应体原样返回。
|
||
"""
|
||
global _page
|
||
try:
|
||
from DrissionPage import ChromiumOptions, ChromiumPage
|
||
if req.port:
|
||
co = ChromiumOptions().set_local_port(port=req.port)
|
||
page = ChromiumPage(addr_or_opts=co)
|
||
with _page_lock:
|
||
_page = page # 供 submit_code 复用
|
||
else:
|
||
page = _get_or_create_page()
|
||
|
||
body = do_submit_phone(
|
||
page=page,
|
||
phone=req.phone,
|
||
url=req.url,
|
||
port=req.port,
|
||
)
|
||
return SubmitPhoneResponse(success=True, data=body)
|
||
except Exception as e:
|
||
raise HTTPException(status_code=500, detail=str(e))
|
||
|
||
|
||
@app.post("/api/submit_code", response_model=SubmitCodeResponse)
|
||
def api_submit_code(req: SubmitCodeRequest):
|
||
"""
|
||
提交验证码：在已打开并处于验证码表单的页面上，
|
||
填写验证码并点击确认订购按钮（img.btn-buy）。
|
||
"""
|
||
try:
|
||
page = _get_or_create_page()
|
||
result = do_submit_code(page=page, code=req.code)
|
||
return SubmitCodeResponse(success=True, message=result.get("message", "已点击确认订购"))
|
||
except Exception as e:
|
||
raise HTTPException(status_code=500, detail=str(e))
|
||
|
||
|
||
if __name__ == "__main__":
|
||
import uvicorn
|
||
uvicorn.run(app, host="0.0.0.0", port=8000)
|