95 lines
2.6 KiB
Python
95 lines
2.6 KiB
Python
from mediagoblin.gmg_commands import util as commands_util
|
|
from mediagoblin.auth import lib as auth_lib
|
|
from mediagoblin import mg_globals
|
|
|
|
|
|
def adduser_parser_setup(subparser):
|
|
subparser.add_argument(
|
|
'username',
|
|
help="Username used to login")
|
|
subparser.add_argument(
|
|
'password',
|
|
help="Your supersecret word to login")
|
|
subparser.add_argument(
|
|
'email',
|
|
help="Email to recieve notifications")
|
|
subparser.add_argument(
|
|
'-cf', '--conf_file', default='mediagoblin.ini',
|
|
help="Config file used to set up environment")
|
|
|
|
|
|
def adduser(args):
|
|
#TODO: Lets trust admins this do not validate Emails :)
|
|
commands_util.setup_app(args)
|
|
|
|
db = mg_globals.database
|
|
users_with_username = \
|
|
db.User.find({
|
|
'username': args.username.lower()
|
|
}).count()
|
|
|
|
if users_with_username:
|
|
print u'Sorry, a user with that name already exists.'
|
|
|
|
else:
|
|
# Create the user
|
|
entry = db.User()
|
|
entry['username'] = unicode(args.username.lower())
|
|
entry['email'] = unicode(args.email)
|
|
entry['pw_hash'] = auth_lib.bcrypt_gen_password_hash(args.password)
|
|
entry['status'] = u'active'
|
|
entry['email_verified'] = True
|
|
entry.save(validate=True)
|
|
|
|
print "User created (and email marked as verified)"
|
|
|
|
|
|
def makeadmin_parser_setup(subparser):
|
|
subparser.add_argument(
|
|
'username',
|
|
help="Username to give admin level")
|
|
subparser.add_argument(
|
|
'-cf', '--conf_file', default='mediagoblin.ini',
|
|
help="Config file used to set up environment")
|
|
|
|
|
|
def makeadmin(args):
|
|
commands_util.setup_app(args)
|
|
|
|
db = mg_globals.database
|
|
|
|
user = db.User.one({'username':unicode(args.username.lower())})
|
|
if user:
|
|
user['is_admin'] = True
|
|
user.save()
|
|
print 'The user is now Admin'
|
|
else:
|
|
print 'The user doesn\'t exist'
|
|
|
|
|
|
def changepw_parser_setup(subparser):
|
|
subparser.add_argument(
|
|
'username',
|
|
help="Username used to login")
|
|
subparser.add_argument(
|
|
'password',
|
|
help="Your NEW supersecret word to login")
|
|
subparser.add_argument(
|
|
'-cf', '--conf_file', default='mediagoblin.ini',
|
|
help="Config file used to set up environment")
|
|
|
|
|
|
def changepw(args):
|
|
commands_util.setup_app(args)
|
|
|
|
db = mg_globals.database
|
|
|
|
user = db.User.one({'username':unicode(args.username.lower())})
|
|
if user:
|
|
user['pw_hash'] = auth_lib.bcrypt_gen_password_hash(args.password)
|
|
user.save()
|
|
print 'Password successfully changed'
|
|
else:
|
|
print 'The user doesn\'t exist'
|
|
|