Example #1
0
 def handle_noargs(self, **options):
     
     def clean_int(value):
         if value.strip()=='':
             value=None
         else:
             value=int(value.replace('$', '').replace(',', ''))
         return value
         
     for year in range(2001, 2015):
         data = HeadStartEnrollment.objects.filter(year=year)
         count = data.count()
         if count > 0:
             print str(year) + ': skipping. ' + str(count) + ' records already loaded'
         else:
             source_file = '%s/health/head_start_enrollment/head_start_enrollment_%s.csv' % (settings.LOCAL_DATA_ROOT, year)
             try:
                 data_reader = csv.reader(open(source_file))
                 
                 insert_count = 0
                 for i, row in enumerate(data_reader):
                     if i == 0:
                         header_row = row
                     elif i > 0:
                         if row[0].strip() <> '':
                             record = HeadStartEnrollment()
                             record.year = year
                             record.state = row[0]
                             record.funding = clean_int(row[1])
                             record.enrollment = clean_int(row[2])
                             record.save()
                             insert_count = insert_count + 1
                 print str(year) + ': ' + str(insert_count) + ' records inserted' 
             except:
                 print 'data file not found for year ' + str(year)
 def handle_noargs(self, **options):
     
     def clean_int(value):
         if value.strip()=='':
             value=None
         else:
             value=int(value.replace('$', '').replace(',', ''))
         return value
         
     for year in range(2001, 2008):
         source_file = '%s/health/head_start_enrollment/head_start_enrollment_%s.csv' % (settings.LOCAL_DATA_ROOT, year)
         data_reader = csv.reader(open(source_file))
     
         for i, row in enumerate(data_reader):
             if i == 0:
                 header_row = row
             elif i > 0:
                 if row[0].strip() <> '':
                     record = HeadStartEnrollment()
                     record.year = year
                     record.state = row[0]
                     record.funding = clean_int(row[1])
                     record.enrollment = clean_int(row[2])
                     record.save()