1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
# 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.auth import lib as auth_lib
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)
args.username = commands_util.prompt_if_not_set(args.username, "Username:")
args.password = commands_util.prompt_if_not_set(args.password, "Password:",True)
args.email = commands_util.prompt_if_not_set(args.email, "Email:")
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()
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)
db = mg_globals.database
user = db.User.one({'username': unicode(args.username.lower())})
if user:
user.is_admin = True
user.all_privileges.append(
db.Privilege.one({
'privilege_name':u'admin'})
)
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)
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'
|
