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


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()
    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)


if __name__ == '__main__':
    main()