def create_user(data, field=None):
"""Create users setting the password."""
if field:
return write_models(User, data, field=field)
data = guess_types(data)
for hash_ in data:
is_superuser = hash_.pop('is_superuser', False)
is_staff = None
if is_superuser:
if 'email' not in hash_:
hash_['email'] = '*****@*****.**'
if 'password' not in hash_:
hash_['password'] = '******'
if 'is_staff' in hash_:
del hash_['is_staff']
user = User.objects.create_superuser(**hash_)
else:
is_staff = hash_.pop('is_staff', None)
if 'username' not in hash_:
hash_['username'] = \
hashlib.md5(hash_['email']).hexdigest()[:30]
if 'email' not in hash_:
hash_['email'] = '*****@*****.**' % (hash_['username'])
user = User.objects.create_user(**hash_)
if is_staff is not None:
user.is_staff = is_staff
user.save()
reset_sequence(User)
def write_pandas(data, field):
# It is not necessary to call hashes_data/guess_types, but it might be
# present in old code using the library. Test that it is a no-op
# in that case.
data = guess_types(data)
for hash_ in data:
if 'name' in hash_:
hash_['name'] += ' Panda'
return write_models(Panda, data, field)
def write_with_rego(data, field=None):
for hash_ in data:
hash_['rego'] = hash_['make'][:3].upper() + "001"
write_models(Harvester, data, field=field)
