3 ### Administrative commands
5 ### (c) 2013 Mark Wooding
8 ###----- Licensing notice ---------------------------------------------------
10 ### This file is part of Chopwood: a password-changing service.
12 ### Chopwood is free software; you can redistribute it and/or modify
13 ### it under the terms of the GNU Affero General Public License as
14 ### published by the Free Software Foundation; either version 3 of the
15 ### License, or (at your option) any later version.
17 ### Chopwood is distributed in the hope that it will be useful,
18 ### but WITHOUT ANY WARRANTY; without even the implied warranty of
19 ### MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
20 ### GNU Affero General Public License for more details.
22 ### You should have received a copy of the GNU Affero General Public
23 ### License along with Chopwood; if not, see
24 ### <http://www.gnu.org/licenses/>.
26 from __future__ import with_statement
31 from output import OUT, PRINT
32 import subcommand as SC
36 'listusers', ['admin'], 'List the existing users.',
37 opts = [SC.Opt('service', '-s', '--service',
38 'list users with SERVICE accounts',
39 argname = 'SERVICE')],
40 oparams = [SC.Arg('pat')])
41 def cmd_listuser(service = None, pat = None):
42 CU.format_list(CU.list_users(service, pat),
43 [CU.column('USER', "~={0.user}A", 3),
44 CU.column('EMAIL', "~={0.email}:[---~;~={0.email}A~]")])
47 'adduser', ['admin'], 'Add a user to the master database.',
48 opts = [SC.Opt('email', '-e', '--email',
49 "email address for the new user",
51 params = [SC.Arg('user')])
52 def cmd_adduser(user, email = None):
54 CU.check_user(user, False)
55 D.DB.execute("INSERT INTO users (user, email) VALUES ($user, $email)",
56 user = user, email = email)
57 D.DB.execute("""INSERT INTO services (user, service)
58 VALUES ($user, $service)""",
59 user = user, service = 'master')
62 'deluser', ['admin'], 'Remove USER from the master database.',
63 params = [SC.Arg('user')])
64 def cmd_deluser(user, email = None):
67 D.DB.execute("DELETE FROM users WHERE user = $user", user = user)
70 'edituser', ['admin'], 'Modify a user record.',
71 opts = [SC.Opt('email', '-e', '--email',
72 "change USER's email address",
74 SC.Opt('noemail', '-E', '--no-email',
75 "forget USER's email address"),
76 SC.Opt('rename', '-r', '--rename',
78 argname = 'NEW-NAME')],
79 params = [SC.Arg('user')])
80 def cmd_edituser(user, email = None, noemail = False, rename = None):
83 if rename is not None: check_user(rename, False)
84 CU.edit_records('users', 'user = $user',
85 [('email', email, noemail),
86 ('user', rename, False)],
90 'delsvc', ['admin'], "Remove all records for SERVICE.",
91 params = [SC.Arg('service')])
92 def cmd_delsvc(service):
94 CU.check_service(service, must_config_p = False, must_records_p = True)
95 D.DB.execute("DELETE FROM services WHERE service = $service",
99 'editsvc', ['admin'], "Edit a given SERVICE.",
100 opts = [SC.Opt('rename', '-r', '--rename', "rename the SERVICE",
101 argname = 'NEW-NAME')],
102 params = [SC.Arg('service')])
103 def cmd_editsvc(service, rename = None):
105 if service == 'master':
106 raise U.ExpectedError, (400, "Can't edit the master service")
108 CU.check_service(service, must_config_p = True, must_records_p = True)
110 CU.check_service(service, must_config_p = False, must_records_p = True)
111 CU.check_service(rename, must_config_p = True, must_records_p = False)
112 CU.edit_records('services', 'service = $service',
113 [('service', rename, False)],
117 'addacct', ['admin'], 'Add an account for a user.',
118 opts = [SC.Opt('alias', '-a', '--alias',
119 "alias by which USER is known to SERVICE",
121 params = [SC.Arg('user'), SC.Arg('service')])
122 def cmd_addacct(user, service, alias = None):
125 CU.check_service(service)
126 D.DB.execute("""SELECT 1 FROM services
127 WHERE user = $user AND service = $service""",
128 user = user, service = service)
129 if D.DB.fetchone() is not None:
130 raise U.ExpectedError, (
131 400, "User `%s' already has `%s' account" % (user, service))
132 D.DB.execute("""INSERT INTO services (service, user, alias)
133 VALUES ($service, $user, $alias)""",
134 service = service, user = user, alias = alias)
137 'delacct', ['admin'], "Remove USER's SERVICE account.",
138 params = [SC.Arg('user'), SC.Arg('service')])
139 def cmd_delacct(user, service):
141 CU.resolve_account(service, user)
142 if service == 'master':
143 raise U.ExpectedError, \
144 (400, "Can't delete master accounts: use `deluser'")
145 D.DB.execute("""DELETE FROM services
146 WHERE service = $service AND user = $user""",
147 service = service, user = user)
150 'editacct', ['admin'], "Modify USER's SERVICE account record.",
151 opts = [SC.Opt('alias', '-a', '--alias',
152 "change USER's login name for SERVICE",
154 SC.Opt('noalias', '-A', '--no-alias',
155 "use USER's master login name")],
156 params = [SC.Arg('user'), SC.Arg('service')])
157 def cmd_editacct(user, service, alias = None, noalias = False):
159 CU.resolve_account(service, user)
160 if service == 'master':
161 raise U.ExpectedError, (400, "Can't edit master accounts")
162 CU.edit_records('services', 'user = $user AND service = $service',
163 [('alias', alias, noalias)],
164 user = user, service = service)
167 'source', ['admin', 'userv', 'remote'], """\
168 Write source code (in `.tar.gz' format) to standard output.""")
169 def cmd_source_admin():
172 ###----- That's all, folks --------------------------------------------------