script.py

import csv

from school.models import Zipcode


def import_us_population():
    for csvfile in (
        'school_data/us_population_2014.csv',
    ):
        with open(csvfile, 'rU') as csvinput:
            reader = csv.reader(csvinput, delimiter=',', quotechar='"',)
            next(reader)
            for index, row in enumerate(reader):

                if row[3] == '-':
                    continue

                population = row[3].translate(None, '"\+-,')

                try:
                    zipcode = Zipcode.objects.get(
                        five_digit=row[1]
                    )
                except Exception, error:
                    # print index, row[1], str(error)
                    continue

                if zipcode:
                    zipcode.population = population
                    zipcode.save()


def run():
    import_us_population()