# app/runner.py
import argparse
from datetime import datetime

from . import etl_busy, etl_idle


def main():
    parser = argparse.ArgumentParser(description="Feiqiu ETL Runner")
    parser.add_argument(
        "--mode",
        choices=["busy", "idle"],
        required=True,
        help="ETL mode: busy or idle",
    )

    args = parser.parse_args()
    now = datetime.now().strftime("%Y-%m-%d %H:%M:%S")

    print(f"[{now}] Start ETL mode={args.mode}")

    if args.mode == "busy":
        etl_busy.run()
    else:
        etl_idle.run()

    print(f"[{datetime.now().strftime('%Y-%m-%d %H:%M:%S')}] ETL finished.")


if __name__ == "__main__":
    main()