initial import: etf strategy project
This commit is contained in:
35
scripts/tushare_download_daily.py
Normal file
35
scripts/tushare_download_daily.py
Normal file
@@ -0,0 +1,35 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import argparse
|
||||
from pathlib import Path
|
||||
|
||||
from qfr.data.tushare_client import fetch_daily, load_tushare_config
|
||||
|
||||
|
||||
def main() -> None:
|
||||
p = argparse.ArgumentParser()
|
||||
p.add_argument("--env", default=None, help="Path to .env (default: auto-detect)")
|
||||
p.add_argument("--ts-code", default=None, help="e.g. 000001.SZ")
|
||||
p.add_argument("--start", dest="start_date", default=None, help="YYYYMMDD")
|
||||
p.add_argument("--end", dest="end_date", default=None, help="YYYYMMDD")
|
||||
p.add_argument("--trade-date", default=None, help="YYYYMMDD")
|
||||
p.add_argument("--out", default="data/raw/tushare_daily.parquet")
|
||||
args = p.parse_args()
|
||||
|
||||
cfg = load_tushare_config(args.env)
|
||||
df = fetch_daily(
|
||||
cfg,
|
||||
ts_code=args.ts_code,
|
||||
trade_date=args.trade_date,
|
||||
start_date=args.start_date,
|
||||
end_date=args.end_date,
|
||||
)
|
||||
|
||||
out = Path(args.out)
|
||||
out.parent.mkdir(parents=True, exist_ok=True)
|
||||
df.to_parquet(out, index=False)
|
||||
print(f"wrote {len(df)} rows -> {out}")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user