Files
datie-bom/browser_login/fetch_abnormal_report.py

82 lines
2.9 KiB
Python

import sys
import time
from pathlib import Path
import datetime
import calendar
import json
sys.path.insert(0, str(Path(__file__).parent))
from login import get_page
from config import OUTPUT_DIR
page = get_page(port=9222)
target_tab = page.get_tab(page.latest_tab)
now = datetime.datetime.now()
first_day = datetime.date(now.year, now.month, 1).strftime('%Y-%m-%d')
last_day = datetime.date(now.year, now.month, calendar.monthrange(now.year, now.month)[1]).strftime('%Y-%m-%d')
print(f"Setting dates: {first_day} to {last_day}")
start_label = target_tab.ele('text:下单日期(开始)')
if start_label:
input_ele = start_label.next().ele('tag:input')
if input_ele:
input_ele.run_js(f"this.value = '{first_day}';")
hidden_input = start_label.next().ele('css:input[type="hidden"]')
if hidden_input:
hidden_input.run_js(f"this.value = '{first_day}';")
visible_input = start_label.next().ele('css:input.textbox-text')
if visible_input:
visible_input.run_js(f"this.value = '{first_day}';")
end_label = target_tab.ele('text:下单日期(结束)')
if end_label:
input_ele = end_label.next().ele('tag:input')
if input_ele:
input_ele.run_js(f"this.value = '{last_day}';")
hidden_input = end_label.next().ele('css:input[type="hidden"]')
if hidden_input:
hidden_input.run_js(f"this.value = '{last_day}';")
visible_input = end_label.next().ele('css:input.textbox-text')
if visible_input:
visible_input.run_js(f"this.value = '{last_day}';")
print("Dates set.")
btn = target_tab.ele('#onSearch') or target_tab.ele('text=查询')
if btn:
target_tab.listen.start()
try:
btn.click(by_js=True)
except:
try:
btn.run_js('this.click()')
except:
pass
time.sleep(5)
packets = target_tab.listen.steps()
for p in packets:
if p.method == 'POST' and ('api' in p.url.lower() or 'Report' in p.url or 'Data' in p.url or 'Search' in p.url):
if p.response and p.response.body:
try:
body = p.response.body
data = body if isinstance(body, (dict, list)) else json.loads(body)
if isinstance(data, dict) and 'result' in data:
res = data['result']
print("拦截到数据API:", p.url)
print("总条数:", res.get('totalCount'))
ts = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
path = OUTPUT_DIR / f"report_abnormal_{ts}.json"
with open(path, "w", encoding="utf-8") as f:
json.dump(data, f, ensure_ascii=False, indent=2)
print("数据已保存至:", path)
sys.exit(0)
except Exception as e:
pass
print("没有找到匹配的数据")
else:
print("没有找到查询按钮")