如何根据名称处理多个文件？

Question

使用下面的代码，我从多个目录中收集文件。我检查他们的日期，检查他们是否存在于列表missing_dates 中，并检查该特定日期是否存在于每个目录中。之后，我将那些具有相同日期的文件处理为 1 个文件。

简而言之：每个目录中相同日期的 3 个文件被处理为 1 个文件。

这是我的代码：

missing_dates = ['20200907', '20200908', '20200909']

root=Path(r'c:\data\FF\Desktop\new_location\middle_stage_preprocessed_district')

data_per_date = dict()
for missing_date in missing_dates:
    print(f"processing {missing_date}")
    files=[fn for fn in (e for e in root.glob(f"**/*_{missing_date}.txt") if e.is_file())]
    if len(files) != 3:
        # stop processing, check other date
        continue

    for file in files:
        with open(file, 'r') as log_file:
            reader = csv.reader(log_file, delimiter = ',')
            next(reader) # skip header
            for row in reader:
                if filter_row(row):
                    vehicle_loc_dict[(row[9], location_token(row))].append(row)

    data_per_date[missing_date] = vehicle_loc_dict  
                                                                                             
for date in missing_dates:
    file_name = "MM{}-AB.dat".format(date)
    full_path = os.path.join(my_files, 'Directory_X', file_name)
    with open(full_path, 'w+') as output:
        writer = csv.writer(output, delimiter = '\t')
        writer.writerow(headers)
        writer.writerow(data)
        vehicle_loc = data_per_date[date]
        for vehicle_loc_list in vehicle_loc.values():
            for record_group in group_records(vehicle_loc_list):
                writer.writerow(output_record(record_group))

我的文件结构是这样的：

├── dir_1
│   ├── XX_20200907.txt
│   └── XX_20200908.txt
├── dir_2
│   ├── YY_20200907.txt
│   └── YY_20200908.txt
└── dir_3
    ├── ZZ_20200907.txt
    └── ZZ_20200908.txt

我收到以下错误，但我不知道为什么。

Traceback (most recent call last):

  File "C:\data\FF\Desktop\Python\Python\Official_part1.py", line 271, in <module>
    vehicle_loc = data_per_date[date]

KeyError: '20200909'

Answer 1

只是关键字20200909 不在您的字典中。您可以添加一种控制机制来捕获您的错误并通过警告、错误、异常。有很多方法可以解决您的问题。你的问题是：你想做什么？错误退出？继续处理？