Add script to import historical number of members older than six months
This commit is contained in:
		
							
								
								
									
										21
									
								
								apiserver/import_six_month_plus_count.py
									
									
									
									
									
										Executable file
									
								
							
							
						
						
									
										21
									
								
								apiserver/import_six_month_plus_count.py
									
									
									
									
									
										Executable file
									
								
							@@ -0,0 +1,21 @@
 | 
			
		||||
# Expects a old_counts.csv of the historical counts in format:
 | 
			
		||||
# date,six_month_plus_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['six_month_plus_count'])
 | 
			
		||||
 | 
			
		||||
        models.StatsMemberCount.objects.update_or_create(
 | 
			
		||||
            date=row['date'],
 | 
			
		||||
            defaults=dict(six_month_plus_count=row['six_month_plus_count']),
 | 
			
		||||
        )
 | 
			
		||||
 | 
			
		||||
print('Done.')
 | 
			
		||||
		Reference in New Issue
	
	Block a user