diff --git a/apiserver/.gitignore b/apiserver/.gitignore index cd3de99..9ba3c0e 100644 --- a/apiserver/.gitignore +++ b/apiserver/.gitignore @@ -114,3 +114,4 @@ missing_paypal/ backups/ secrets.py old_counts.csv +scans.csv diff --git a/apiserver/apiserver/api/models.py b/apiserver/apiserver/api/models.py index 983c8d9..449b797 100644 --- a/apiserver/apiserver/api/models.py +++ b/apiserver/apiserver/api/models.py @@ -141,6 +141,10 @@ class StatsSignupCount(models.Model): retain_count = models.IntegerField(default=0) vetted_count = models.IntegerField(default=0) +class StatsSpaceActivity(models.Model): + date = models.DateField(default=today_alberta_tz) + card_scans = models.IntegerField() + class HistoryIndex(models.Model): content_type = models.ForeignKey(ContentType, null=True, on_delete=models.SET_NULL) object_id = models.PositiveIntegerField() diff --git a/apiserver/import_card_scans.py b/apiserver/import_card_scans.py new file mode 100644 index 0000000..9774eaf --- /dev/null +++ b/apiserver/import_card_scans.py @@ -0,0 +1,43 @@ +# Expects a scans.csv of the historical scans in format: +# date,card_number + +import django, sys, os +os.environ['DJANGO_SETTINGS_MODULE'] = 'apiserver.settings' +django.setup() + +import csv +from datetime import datetime +from apiserver.api import models +from django.utils.timezone import now, pytz + +days = {} + +with open('scans.csv', newline='') as csvfile: + reader = csv.DictReader(csvfile) + for row in reader: + datetime_obj = datetime.strptime(row['date'], "%Y-%m-%d %H:%M:%S") + datetime_obj_utc = datetime_obj.replace(tzinfo=pytz.timezone('UTC')) + date = datetime_obj_utc.astimezone(pytz.timezone('America/Edmonton')) + + card = row['card_number'] + + print('Processing', date, card) + day = str(date.date()) + + if day not in days: + days[day] = set() + + days[day].add(card) + +print(days) + +for day, cards in days.items(): + print(day, len(cards)) + + models.StatsSpaceActivity.objects.update_or_create( + date=day, + defaults=dict(card_scans=len(cards)), + ) + +print('Done.') +