You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
59 lines
2.0 KiB
Python
59 lines
2.0 KiB
Python
"""Build continuation snapshots from an existing event-day snapshot."""
|
|
from __future__ import annotations
|
|
|
|
import argparse
|
|
import asyncio
|
|
import json
|
|
|
|
from libs.common.config import get_settings
|
|
from libs.common.logging import configure_logging
|
|
from libs.export.continuation_snapshot import export_continuation_snapshot_from_base
|
|
|
|
|
|
async def run_continuation_snapshot(
|
|
*,
|
|
base_snapshot_dir: str,
|
|
output_dir: str,
|
|
snapshot_id: str,
|
|
lookback_days: int,
|
|
) -> dict:
|
|
return await export_continuation_snapshot_from_base(
|
|
base_snapshot_dir=base_snapshot_dir,
|
|
output_dir=output_dir,
|
|
snapshot_id=snapshot_id,
|
|
lookback_days=lookback_days,
|
|
)
|
|
|
|
|
|
def main() -> None:
|
|
parser = argparse.ArgumentParser(description="Build continuation snapshot from base snapshot")
|
|
parser.add_argument("--base-snapshot-dir", required=True, help="Base snapshot directory path")
|
|
parser.add_argument("--output-dir", default="./data/datasets/snapshots", help="Snapshot output root")
|
|
parser.add_argument("--snapshot-id", required=True, help="New snapshot id")
|
|
parser.add_argument("--lookback-days", type=int, default=3, help="Continuation signal lookback in trading days")
|
|
parser.add_argument("--json", action="store_true", help="Print manifest JSON")
|
|
args = parser.parse_args()
|
|
|
|
configure_logging(get_settings().log_level)
|
|
manifest = asyncio.run(
|
|
run_continuation_snapshot(
|
|
base_snapshot_dir=args.base_snapshot_dir,
|
|
output_dir=args.output_dir,
|
|
snapshot_id=args.snapshot_id,
|
|
lookback_days=args.lookback_days,
|
|
)
|
|
)
|
|
if args.json:
|
|
print(json.dumps(manifest, indent=2))
|
|
else:
|
|
print(f"Snapshot exported: {manifest['snapshot_id']}")
|
|
print(f" Output: {manifest['output_dir']}")
|
|
print(f" Total rows: {manifest['total_rows']}")
|
|
for split, count in manifest["row_counts"].items():
|
|
print(f" {split}: {count} rows")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|
|
|