shop-java/tools/grab_one.py

# -*- coding: utf-8 -*-
"""抓取 PHP 单接口响应并美化打印：python grab_one.py /store/order.delivery/list "page=1" """
import json
import os
import re
import sys
from pathlib import Path

import requests

PHP_BASE = os.environ.get("YOSHOP_PHP_BASE", "https://shop.qinys.cn/index.php?s=")
SESSION = requests.Session()
SESSION.trust_env = False


def php_token() -> str:
    src = Path(r"F:\import goods\upload_goods.py").read_text(encoding="utf-8")
    m = re.search(r'TOKEN\s*=\s*"([^"]+)"', src)
    return m.group(1)


def main():
    path = sys.argv[1]
    params = {}
    for item in sys.argv[2:]:
        k, _, v = item.partition("=")
        params[k] = v
    r = SESSION.get(PHP_BASE + path, headers={"Access-Token": php_token()}, params=params, timeout=60)
    data = r.json()
    print("HTTP", r.status_code, "status", data.get("status"), data.get("message"))
    print(json.dumps(data.get("data"), ensure_ascii=False, indent=2)[:20000])


if __name__ == "__main__":
    main()