文件预览

handler.py

查看 Skills of ETF data released by ft.tech. 技能包中的文件内容。

文件内容

sub-skills/etf-component/scripts/handler.py

#!/usr/bin/env python3
"""查询单只 ETF 成份(market.ft.tech)"""
import argparse
import json
import sys
import urllib.error
import urllib.parse
import urllib.request

BASE_URL = "https://market.ft.tech"


def main():
    parser = argparse.ArgumentParser(description="查询单只 ETF 成份")
    parser.add_argument(
        "--symbol",
        required=True,
        help="ETF 标的代码,带交易所后缀,如 510300.XSHG、159915.XSHE、510300.SH",
    )
    args = parser.parse_args()

    path = "/data/api/v1/market/data/etf-component?" + urllib.parse.urlencode({"symbol": args.symbol})
    url = BASE_URL + path

    req = urllib.request.Request(url, method="GET")
    req.add_header("Content-Type", "application/json")

    try:
        with urllib.request.urlopen(req) as resp:
            data = json.loads(resp.read().decode())
        print(json.dumps(data, ensure_ascii=False, indent=2))
    except urllib.error.HTTPError as e:
        body = e.read().decode()
        print(body, file=sys.stderr)
        sys.exit(1)


if __name__ == "__main__":
    main()