Fix last month of year

This commit is contained in:
Watchful1 2024-01-07 10:45:51 -08:00
parent 82966bf7f6
commit 6114afb53f
3 changed files with 8 additions and 7 deletions

View file

@ -60,7 +60,10 @@ if __name__ == "__main__":
total_objects = 0
total_bytes = 0
minute_iterator = month
end_time = month.replace(month=month.month + 1)
if month == 12:
end_time = month.replace(year=month.year + 1, month=1)
else:
end_time = month.replace(month=month.month + 1)
while minute_iterator < end_time:
minute_file_path = os.path.join(args.input, args.type, minute_iterator.strftime('%y-%m-%d'), f"{prefix}_{minute_iterator.strftime('%y-%m-%d_%H-%M')}.zst")
for obj, line, _ in utils.read_obj_zst_meta(minute_file_path):