106 lines
3.3 KiB
Python
106 lines
3.3 KiB
Python
# GNU MediaGoblin -- federated, autonomous media hosting
|
|
# Copyright (C) 2011, 2012 MediaGoblin contributors. See AUTHORS.
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
from mediagoblin.gmg_commands import util as commands_util
|
|
from mediagoblin import auth
|
|
from mediagoblin import mg_globals
|
|
|
|
def adduser_parser_setup(subparser):
|
|
subparser.add_argument(
|
|
'--username','-u',
|
|
help="Username used to login")
|
|
subparser.add_argument(
|
|
'--password','-p',
|
|
help="Your supersecret word to login, beware of storing it in bash history")
|
|
subparser.add_argument(
|
|
'--email','-e',
|
|
help="Email to receive notifications")
|
|
|
|
|
|
def adduser(args):
|
|
#TODO: Lets trust admins this do not validate Emails :)
|
|
commands_util.setup_app(args[0])
|
|
|
|
args[0].username = commands_util.prompt_if_not_set(args[0].username, "Username:")
|
|
args[0].password = commands_util.prompt_if_not_set(args[0].password, "Password:",True)
|
|
args[0].email = commands_util.prompt_if_not_set(args[0].email, "Email:")
|
|
|
|
db = mg_globals.database
|
|
users_with_username = \
|
|
db.User.query.filter_by(
|
|
username=args[0].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[0].username.lower())
|
|
entry.email = unicode(args[0].email)
|
|
entry.pw_hash = auth.gen_password_hash(args[0].password)
|
|
entry.status = u'active'
|
|
entry.email_verified = True
|
|
entry.save()
|
|
|
|
print "User created (and email marked as verified)"
|
|
|
|
|
|
def makeadmin_parser_setup(subparser):
|
|
subparser.add_argument(
|
|
'username',
|
|
help="Username to give admin level")
|
|
|
|
|
|
def makeadmin(args):
|
|
commands_util.setup_app(args[0])
|
|
|
|
db = mg_globals.database
|
|
|
|
user = db.User.query.filter_by(
|
|
username=unicode(args[0].username.lower())).one()
|
|
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")
|
|
|
|
|
|
def changepw(args):
|
|
commands_util.setup_app(args[0])
|
|
|
|
db = mg_globals.database
|
|
|
|
user = db.User.query.filter_by(
|
|
username=unicode(args[0].username.lower())).one()
|
|
if user:
|
|
user.pw_hash = auth.gen_password_hash(args[0].password)
|
|
user.save()
|
|
print 'Password successfully changed'
|
|
else:
|
|
print 'The user doesn\'t exist'
|