Exemplo n.º 1
0
def import_users(csv_file):
    """ creates Provider object off a CSV filename

    CSV FORMAT:
    FIRST NAME, LAST NAME, EMAIL, PHONE, PHONE2, ROLE CODE, ENTITY CODE
    CSV file MUST include header row. """

    errors = []

    first = True
    f = open(csv_file)
    succ = open('success.csv', 'w')
    for line in f.readlines():
        if first:
            first = False
            continue
        # explode CSV line
        fname, lname, email, \
        phone_number, phone2, role_code, entity_code, area = line.strip().split(',')

        # convert name to unicode for django & .title()
        try:
            fname = unicode(fname.strip(), 'utf-8').title()
        except:
            pass

        try:
            lname = unicode(lname.strip(), 'utf-8').title()
        except:
            pass

        # retrieve parent object if address is provided
        try:
            entity = Entity.objects.get(slug=entity_code.lower())
        except:
            print("BAD ENTITY: %s" % entity_code)
            #raise
            errors.append(line)
            continue

        try:
            role = Role.objects.get(slug=role_code.lower())
        except:
            print("BAD ROLE: %s" % role_code)
            #raise
            errors.append(line)
            continue

        # build an Access based on Role and Entity selected
        # if a national role, force attachment to root entity
        if role.slug in ('antim', 'national'):
            entity = Entity.objects.filter(level=0)[0]
        elif role.slug != entity.type.slug:
            print("ROLE MISMATCH LOCATION TYPE: %s / %s (%s)" %
                  (role_code, entity_code, entity.type))
            errors.append(line)
            continue
        access = Access.find_by(role, entity)

        # forge username
        username = username_from_name(fname, lname)
        # generate password
        password = random_password()

        pcount = Provider.objects.filter(phone_number=phone_number).count()

        if phone_number and pcount > 0:
            print("PHONE NUMBER CONFLICT: %s" % phone_number)
            errors.append(line)
            continue

        # create Provider
        provider = Provider.create_provider(username, \
                                            'xx', access=[access])
        provider.set_password(password)

        # we have a valid provider whatever the case. update details
        provider.first_name = fname
        provider.last_name = lname
        provider.email = email
        # only update if not None to preserve uniqueness

        if phone_number and pcount == 0:
            provider.phone_number = phone_number

        if phone2:
            provider.phone_number_extra = phone2

        provider.save()

        line = u"%(ent)s,%(role)s,%(fname)s,%(lname)s,%(user)s,%(pass)s,%(passenc)s\n" % {
            'ent': entity,
            'role': role,
            'fname': provider.first_name,
            'lname': provider.last_name,
            'user': username,
            'pass': password,
            'passenc': provider.user.password
        }
        succ.write(line.encode('utf-8'))
        print(provider.name_access())
    f.close()
    succ.close()

    if errors:
        error_file = 'errors.csv'
        f = open(error_file, 'w')
        for line in errors:
            f.write(line)
        f.close()
        print("%d errors appended to file %s" % (errors.__len__(), error_file))
Exemplo n.º 2
0
def import_users(csv_file):
    """ creates Provider object off a CSV filename

    CSV FORMAT:
    FIRST NAME, LAST NAME, EMAIL, PHONE, PHONE2, ROLE CODE, ENTITY CODE
    CSV file MUST include header row. """

    errors = []

    first = True
    f = open(csv_file)
    succ = open('success.csv', 'w')
    for line in f.readlines():
        if first:
            first = False
            continue
        # explode CSV line
        fname, lname, email, \
        phone_number, phone2, role_code, entity_code, area = line.strip().split(',')

        # convert name to unicode for django & .title()
        try:
            fname = unicode(fname.strip(), 'utf-8').title()
        except:
            pass

        try:
            lname = unicode(lname.strip(), 'utf-8').title()
        except:
            pass

        # retrieve parent object if address is provided
        try:
            entity = Entity.objects.get(slug=entity_code.lower())
        except:
            print("BAD ENTITY: %s" % entity_code)
            #raise
            errors.append(line)
            continue

        try:
            role = Role.objects.get(slug=role_code.lower())
        except:
            print("BAD ROLE: %s" % role_code)
            #raise
            errors.append(line)
            continue


        # build an Access based on Role and Entity selected
        # if a national role, force attachment to root entity
        if role.slug in ('antim', 'national'):
            entity = Entity.objects.filter(level=0)[0]
        elif role.slug != entity.type.slug:
            print("ROLE MISMATCH LOCATION TYPE: %s / %s (%s)" % (role_code, entity_code, entity.type))
            errors.append(line)
            continue
        access = Access.find_by(role, entity)

        # forge username
        username = username_from_name(fname, lname)
        # generate password
        password = random_password()

        pcount = Provider.objects.filter(phone_number=phone_number).count()

        if phone_number and pcount > 0:
            print("PHONE NUMBER CONFLICT: %s" % phone_number)
            errors.append(line)
            continue

        # create Provider
        provider = Provider.create_provider(username, \
                                            'xx', access=[access])
        provider.set_password(password)

        # we have a valid provider whatever the case. update details
        provider.first_name = fname
        provider.last_name = lname
        provider.email = email
        # only update if not None to preserve uniqueness

        if phone_number and pcount == 0:
            provider.phone_number = phone_number

        if phone2:
            provider.phone_number_extra = phone2

        provider.save()

        line = u"%(ent)s,%(role)s,%(fname)s,%(lname)s,%(user)s,%(pass)s,%(passenc)s\n" % {'ent': entity, 'role': role, 'fname': provider.first_name, 'lname': provider.last_name, 'user': username, 'pass': password, 'passenc': provider.user.password}
        succ.write(line.encode('utf-8'))
        print(provider.name_access())
    f.close()
    succ.close()

    if errors:
        error_file = 'errors.csv'
        f = open(error_file, 'w')
        for line in errors:
            f.write(line)
        f.close()
        print("%d errors appended to file %s" % (errors.__len__(), error_file))