Add script to import old member counts from csv

This commit is contained in:
Tanner Collin 2020-04-21 05:46:40 +00:00
parent 8d4a63fbfc
commit 20c93cd5c4
2 changed files with 22 additions and 0 deletions

View File

@ -113,3 +113,4 @@ old_paypal/
missing_paypal/ missing_paypal/
backups/ backups/
secrets.py secrets.py
old_counts.csv

View File

@ -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.')