First Gist: Process a CSV File in Python/Django

Process CSV File Python

#Test Gist
import csv

from models import RegistrationCode


REQUIRED_FIELDNAMES = frozenset([
    'Registration code',
])


def process_row(row):

    if len(row) == 1:
        reg_key = list(row.keys())[0]
    else:
        reg_key = 'Kraft Registration code'

    reg_code, _ = RegistrationCode.objects.get_or_create(code=row[reg_key])
    reg_code.save()

    return 'imported %s' % reg_code.code


def process_file(csv_file):

    reader = csv.DictReader(csv_file.read().splitlines())

    csv_headers = frozenset(reader.fieldnames)

    if REQUIRED_FIELDNAMES.issubset(csv_headers) or len(csv_headers) == 1:
        res = map(process_row, reader)
        return '\n\n'.join(res)
    else:
        return '**INVALID DATA FILE**'