1 # GNU MediaGoblin -- federated, autonomous media hosting
2 # Copyright (C) 2011, 2012 MediaGoblin contributors. See AUTHORS.
4 # This program is free software: you can redistribute it and/or modify
5 # it under the terms of the GNU Affero General Public License as published by
6 # the Free Software Foundation, either version 3 of the License, or
7 # (at your option) any later version.
9 # This program is distributed in the hope that it will be useful,
10 # but WITHOUT ANY WARRANTY; without even the implied warranty of
11 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 # GNU Affero General Public License for more details.
14 # You should have received a copy of the GNU Affero General Public License
15 # along with this program. If not, see <http://www.gnu.org/licenses/>.
17 from __future__
import print_function
23 from mediagoblin
.db
.models
import LocalUser
24 from mediagoblin
.gmg_commands
import util
as commands_util
25 from mediagoblin
import auth
26 from mediagoblin
import mg_globals
28 def adduser_parser_setup(subparser
):
29 subparser
.add_argument(
31 help="Username used to login")
32 subparser
.add_argument(
34 help="Your supersecret word to login, beware of storing it in bash history")
35 subparser
.add_argument(
37 help="Email to receive notifications")
41 #TODO: Lets trust admins this do not validate Emails :)
42 commands_util
.setup_app(args
)
44 args
.username
= six
.text_type(commands_util
.prompt_if_not_set(args
.username
, "Username:"))
45 args
.password
= commands_util
.prompt_if_not_set(args
.password
, "Password:",True)
46 args
.email
= commands_util
.prompt_if_not_set(args
.email
, "Email:")
48 db
= mg_globals
.database
49 users_with_username
= \
50 db
.LocalUser
.query
.filter(
51 LocalUser
.username
==args
.username
.lower()
54 if users_with_username
:
55 print(u
'Sorry, a user with that name already exists.')
60 entry
= db
.LocalUser()
61 entry
.username
= six
.text_type(args
.username
.lower())
62 entry
.email
= six
.text_type(args
.email
)
63 entry
.pw_hash
= auth
.gen_password_hash(args
.password
)
64 default_privileges
= [
65 db
.Privilege
.query
.filter(
66 db
.Privilege
.privilege_name
==u
'commenter').one(),
67 db
.Privilege
.query
.filter(
68 db
.Privilege
.privilege_name
==u
'uploader').one(),
69 db
.Privilege
.query
.filter(
70 db
.Privilege
.privilege_name
==u
'reporter').one(),
71 db
.Privilege
.query
.filter(
72 db
.Privilege
.privilege_name
==u
'active').one()
74 entry
.all_privileges
= default_privileges
77 print(u
"User created (and email marked as verified)")
80 def makeadmin_parser_setup(subparser
):
81 subparser
.add_argument(
83 help="Username to give admin level")
87 commands_util
.setup_app(args
)
89 db
= mg_globals
.database
91 user
= db
.LocalUser
.query
.filter(
92 LocalUser
.username
==six
.text_type(args
.username
.lower())
95 user
.all_privileges
.append(
96 db
.Privilege
.query
.filter(
97 db
.Privilege
.privilege_name
==u
'admin').one()
100 print(u
'The user is now Admin')
102 print(u
'The user doesn\'t exist')
106 def changepw_parser_setup(subparser
):
107 subparser
.add_argument(
109 help="Username used to login")
110 subparser
.add_argument(
112 help="Your NEW supersecret word to login")
116 commands_util
.setup_app(args
)
118 db
= mg_globals
.database
120 user
= db
.LocalUser
.query
.filter(
121 LocalUser
.username
==six
.text_type(args
.username
.lower())
124 user
.pw_hash
= auth
.gen_password_hash(args
.password
)
126 print(u
'Password successfully changed')
128 print(u
'The user doesn\'t exist')
132 def deleteuser_parser_setup(subparser
):
133 subparser
.add_argument(
135 help="Username to delete",
139 def deleteuser(args
):
140 commands_util
.setup_app(args
)
142 db
= mg_globals
.database
144 user
= db
.LocalUser
.query
.filter(
145 LocalUser
.username
==args
.username
.lower()
149 print('The user %s has been deleted' % args
.username
)
151 print('The user %s doesn\'t exist' % args
.username
)