LongArticlesJob/updateAccountV3.py

import time

from datetime import datetime, timedelta
from argparse import ArgumentParser

from tasks.data_tasks.account_position_read_avg_task import (
    AccountPositionReadAvgTask,
    AccountOpenRateAvgTask,
)


def main():
    """
    main job
    :return:
    """
    parser = ArgumentParser()
    parser.add_argument(
        "--run-date",
        help="Run only once for date in format of %Y-%m-%d. \
                                If no specified, run as daily jobs.",
    )
    args = parser.parse_args()
    update_account_read_avg_task = AccountPositionReadAvgTask()
    update_account_open_rate_avg_task = AccountOpenRateAvgTask()
    if args.run_date:
        update_account_read_avg_task.do_task_list(dt=args.run_date)
    else:
        dt_object = datetime.fromtimestamp(int(time.time()))
        one_day = timedelta(days=1)
        yesterday = dt_object - one_day
        yesterday_str = yesterday.strftime("%Y-%m-%d")
        update_account_read_avg_task.do_task_list(dt=yesterday_str)
        update_account_open_rate_avg_task.do_task_list(date_string=yesterday_str)


if __name__ == "__main__":
    main()