Add any missing days to card scan import script

This commit is contained in:
Tanner Collin 2020-06-04 03:04:06 +00:00
parent 146d324025
commit 0231394cef

13
apiserver/import_card_scans.py Normal file → Executable file
View File

@ -6,12 +6,23 @@ os.environ['DJANGO_SETTINGS_MODULE'] = 'apiserver.settings'
django.setup() django.setup()
import csv import csv
from datetime import datetime from datetime import datetime, timedelta
from apiserver.api import models from apiserver.api import models
from django.utils.timezone import now, pytz from django.utils.timezone import now, pytz
def today_alberta_tz():
return datetime.now(pytz.timezone('America/Edmonton')).date()
days = {} 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: with open('scans.csv', newline='') as csvfile:
reader = csv.DictReader(csvfile) reader = csv.DictReader(csvfile)
for row in reader: for row in reader: