test.py

import argparse
from batch_runner import BatchRunner
from local_test_settings import (
    CLIENT_FTP_HOST,
    CLIENT_FTP_PASSWORD,
    CLIENT_FTP_PORT,
    CLIENT_FTP_ROOT_PATH,
    CLIENT_FTP_USERNAME,
    FTP_PASSIVE_MODE,
    LOCAL_DB_PATH,
    LOCAL_WORK_DIR,
    RUPI_SCALE_PERCENT,
    SERVER_FTP_HOST,
    SERVER_FTP_PASSWORD,
    SERVER_FTP_PORT,
    SERVER_FTP_ROOT_PATH,
    SERVER_FTP_USERNAME,
)


def build_parser():
    parser = argparse.ArgumentParser(description="날짜와 RUBI/RUPI만 받아 FTP 파일을 처리합니다")
    parser.add_argument("--input-date", required=True, help="YYYY-MM-DD 형식")
    parser.add_argument("--parser", required=True, help="RUBI 또는 RUPI")
    return parser


def main():
    args = build_parser().parse_args()
    runner = BatchRunner(
        input_date=args.input_date,
        parser_name=args.parser,
        client_host=CLIENT_FTP_HOST,
        client_port=CLIENT_FTP_PORT,
        client_username=CLIENT_FTP_USERNAME,
        client_password=CLIENT_FTP_PASSWORD,
        client_root_path=CLIENT_FTP_ROOT_PATH,
        server_host=SERVER_FTP_HOST,
        server_port=SERVER_FTP_PORT,
        server_username=SERVER_FTP_USERNAME,
        server_password=SERVER_FTP_PASSWORD,
        server_root_path=SERVER_FTP_ROOT_PATH,
        db_path=LOCAL_DB_PATH,
        work_dir=LOCAL_WORK_DIR,
        scale_percent=RUPI_SCALE_PERCENT,
        passive=FTP_PASSIVE_MODE,
    )
    runner.run()
    return 0


if __name__ == "__main__":
    raise SystemExit(main())