python 在给定一系列日期和文件模板的情况下查找丢失的文件

Posted 2021-05-09

#!/usr/bin/env python

from datetime import date, datetime, timedelta
from os.path import isfile, basename
import sys

from string_utils.strings import replace_vars_in_string

script_name = basename(__file__)

def usage():

    print('{} [START-DATE] [END-DATE] [FILE-TEMPLATE]'.format(script_name))
    print('')
    print('  where:')
    print('')
    print('    START-DATE     start date (YYYYMMDD)')
    print('    END-DATE       end date (YYYYMMDD)')
    print('    FILE-TEMPLATE  file template (use {yyyy}, {mm}, and {dd})')
    print('')
    print('  For example:')
    print('')
    print('    {} 20140101 20141231 /data/{{yyyy}}/{{mm}}/{{dd}}/data-file-{{yyyy}}{{mm}}{{dd}}.txt'.format(script_name))
    print('')

def main():

    try:
        start_date = sys.argv[1]
        end_date = sys.argv[2]
        file_template = sys.argv[3]
    except IndexError:
        usage()
        sys.exit(2)

    start_date_obj = datetime.strptime(start_date, '%Y%m%d')
    end_date_obj = datetime.strptime(end_date, '%Y%m%d')

    date_list = [(start_date_obj + timedelta(days=i)).strftime('%Y%m%d') for i in range((end_date_obj - start_date_obj).days + 1)]

    none_missing = True
    for date in date_list:
        v = {'yyyy': date[0:4], 'mm': date[4:6], 'dd': date[6:8]}
        file = replace_vars_in_string(file_template, **v)
        if not isfile(file):
            none_missing = False
            print('Missing file: {}'.format(file))

    if none_missing:
        print('No missing files...')

if __name__ == '__main__':
    main()