31 lines
664 B
Python
31 lines
664 B
Python
# 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() |