diff --git a/apiserver/.gitignore b/apiserver/.gitignore index e1e3561..cd3de99 100644 --- a/apiserver/.gitignore +++ b/apiserver/.gitignore @@ -113,3 +113,4 @@ old_paypal/ missing_paypal/ backups/ secrets.py +old_counts.csv diff --git a/apiserver/import_member_counts.py b/apiserver/import_member_counts.py new file mode 100755 index 0000000..7792168 --- /dev/null +++ b/apiserver/import_member_counts.py @@ -0,0 +1,21 @@ +# Expects a old_counts.csv of the historical counts in format: +# date,member_count,green_count + +import django, sys, os +os.environ['DJANGO_SETTINGS_MODULE'] = 'apiserver.settings' +django.setup() + +import csv +from apiserver.api import models + +with open('old_counts.csv', newline='') as csvfile: + reader = csv.DictReader(csvfile) + for row in reader: + print('Adding', row['date'], row['member_count'], row['green_count']) + + models.StatsMemberCount.objects.update_or_create( + date=row['date'], + defaults=dict(member_count=row['member_count'], green_count=row['green_count']), + ) + +print('Done.')