From 0231394cef06cfa4f3194ebe7f60ddb7dfb98ff5 Mon Sep 17 00:00:00 2001 From: Tanner Collin Date: Thu, 4 Jun 2020 03:04:06 +0000 Subject: [PATCH] Add any missing days to card scan import script --- apiserver/import_card_scans.py | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) mode change 100644 => 100755 apiserver/import_card_scans.py diff --git a/apiserver/import_card_scans.py b/apiserver/import_card_scans.py old mode 100644 new mode 100755 index 9774eaf..a4323f7 --- a/apiserver/import_card_scans.py +++ b/apiserver/import_card_scans.py @@ -6,12 +6,23 @@ os.environ['DJANGO_SETTINGS_MODULE'] = 'apiserver.settings' django.setup() import csv -from datetime import datetime +from datetime import datetime, timedelta from apiserver.api import models from django.utils.timezone import now, pytz +def today_alberta_tz(): + return datetime.now(pytz.timezone('America/Edmonton')).date() + days = {} +date = datetime(2020, 3, 7).date() +while date <= today_alberta_tz(): + days[str(date)] = set() + date += timedelta(days=1) + +print('Initialized with:') +print(days) + with open('scans.csv', newline='') as csvfile: reader = csv.DictReader(csvfile) for row in reader: