#303 : enhancement : add a command to delete users

This commit is contained in:
Loïc Le Ninan 2014-06-07 23:45:50 +02:00 committed by Christopher Allan Webber
parent 8db3277cd7
commit 8c7ba963af
2 changed files with 24 additions and 0 deletions

View File

@ -37,6 +37,10 @@ SUBCOMMAND_MAP = {
'setup': 'mediagoblin.gmg_commands.users:changepw_parser_setup', 'setup': 'mediagoblin.gmg_commands.users:changepw_parser_setup',
'func': 'mediagoblin.gmg_commands.users:changepw', 'func': 'mediagoblin.gmg_commands.users:changepw',
'help': 'Changes a user\'s password'}, 'help': 'Changes a user\'s password'},
'deleteuser': {
'setup': 'mediagoblin.gmg_commands.users:deleteuser_parser_setup',
'func': 'mediagoblin.gmg_commands.users:deleteuser',
'help': 'Deletes a user'},
'dbupdate': { 'dbupdate': {
'setup': 'mediagoblin.gmg_commands.dbupdate:dbupdate_parse_setup', 'setup': 'mediagoblin.gmg_commands.dbupdate:dbupdate_parse_setup',
'func': 'mediagoblin.gmg_commands.dbupdate:dbupdate', 'func': 'mediagoblin.gmg_commands.dbupdate:dbupdate',

View File

@ -115,3 +115,23 @@ def changepw(args):
print 'Password successfully changed' print 'Password successfully changed'
else: else:
print 'The user doesn\'t exist' print 'The user doesn\'t exist'
def deleteuser_parser_setup(subparser):
subparser.add_argument(
'username',
help="Username to delete")
def deleteuser(args):
commands_util.setup_app(args)
db = mg_globals.database
user = db.User.query.filter_by(
username=unicode(args.username.lower())).one()
if user:
user.delete()
print 'The user %s has been deleted' % args.username
else:
print 'The user %s doesn\'t exist' % args.username