def run_csv_file(db_conn, filename, optdict):
"""Runs the appropriate actions from a CSV file.
:param db_conn: DB connection object
:param filename: CSV filename
:param optdict: Options dictionary.
:returns int: number of successful user actions.
"""
api = account_mgr.get_api(config)
log = logging.getLogger("run_csv_file")
dict_reader = csv.DictReader(filename)
if 'setpw' in optdict:
user_dicts = assure_keys(dict_reader, SETPW_REQUIRED_KEYS)
emails = (email for email in user_dicts['email_addr'])
pws = (pw for pw in user_dicts['password'])
set_multi_passwords(db_conn, emails, pws)
# All done, so leave the function here.
return len(user_dicts)
success_count = 0
if 'create' in optdict:
# Runs the creation routine for each user.
user_dicts = assure_keys(dict_reader, CREATE_REQUIRED_KEYS)
for user in user_dicts:
api.create_user(
{'name': user['given_name'] + ' ' + user['surname'],
'email': user['email_addr'],
'group_id': user['group_id'],
})
success_count += 1
elif 'set_email' in optdict:
# Sets emails for each user.
user_dicts = assure_keys(dict_reader, SET_EMAIL_REQUIRED_KEYS)
for user in user_dicts:
api.edit_user(user['email_addr'], dict(email=user['new_email']))
success_count += 1
elif 'set_group' in optdict:
# Sets groups for each user.
user_dicts = assure_keys(dict_reader, SET_GROUP_REQUIRED_KEYS)
for user in user_dicts:
api.edit_user(user['email_addr'], dict(group_id=user['group_id']))
success_count += 1
elif 'disable' in optdict:
user_dicts = assure_keys(dict_reader, frozenset(['email_addr']))
for user in user_dicts:
api.edit_user(user['email_addr'], dict(enabled=False))
success_count += 1
elif 'enable' in optdict:
user_dicts = assure_keys(dict_reader, frozenset(['email_addr']))
for user in user_dicts:
api.edit_user(user['email_addr'], dict(enabled=True))
success_count += 1
else:
raise UsersActionError("Got an action that's not accounted for!")
return success_count
def run_csv_file(db_conn, filename, optdict):
"""Runs the appropriate actions from a CSV file.
:param db_conn: DB connection object
:param filename: CSV filename
:param optdict: Options dictionary.
:returns int: number of successful user actions.
"""
log = logging.getLogger("run_csv_file")
dict_reader = csv.DictReader(filename)
if 'setpw' in optdict:
user_dicts = assure_keys(dict_reader, SETPW_REQUIRED_KEYS)
emails = (email for email in user_dicts['email_addr'])
pws = (pw for pw in user_dicts['password'])
set_multi_passwords(db_conn, emails, pws)
# All done, so leave the function here.
return len(user_dicts)
success_count = 0
if 'create' in optdict:
# Runs the creation routine for each user.
user_dicts = assure_keys(dict_reader, CREATE_REQUIRED_KEYS)
for user in user_dicts:
result = api_interface.create_user(
{'firstname': user['given_name'],
'lastname': user['surname'],
'email': user['email_addr'],})
result['group_id'] = user['group_id']
try:
result2 = api_interface.set_user_group(result)
except api_interface.ApiActionFailedError as e:
log.error("User created with no group assignment: %s" %
(user['email_addr'],))
raise e
success_count += 1
elif 'set_email' in optdict:
# Sets emails for each user.
user_dicts = assure_keys(dict_reader, SET_EMAIL_REQUIRED_KEYS)
for user in user_dicts:
result = api_interface.change_email(
{'email': user['new_email'],
'old_email': user['email_addr'],})
success_count += 1
elif 'set_group' in optdict:
# Sets groups for each user.
user_dicts = assure_keys(dict_reader, SET_GROUP_REQUIRED_KEYS)
for user in user_dicts:
result = api_interface.set_user_group(
{'email': user['email_addr'],
'group_id': user['group_id'],})
success_count += 1
elif 'disable' in optdict:
user_dicts = assure_keys(dict_reader, frozenset(['email_addr']))
for user in user_dicts:
result = api_interface.deactivate_user(
{'email': user['email_addr'],})
success_count += 1
elif 'enable' in optdict:
user_dicts = assure_keys(dict_reader, frozenset(['email_addr']))
for user in user_dicts:
result = api_interface.activate_user(
{'email': user['email_addr'],})
success_count += 1
else:
raise UsersActionError("Got an action that's not accounted for!")
return success_count
