207 lines
7.4 KiB
Python
207 lines
7.4 KiB
Python
import socket
|
|
import sys
|
|
from typing import Dict
|
|
|
|
import click
|
|
from zope import component
|
|
|
|
from .utils import http_post, http_get, http_patch, http_delete, \
|
|
handle_stream_response, handle_sync_response, print_colon_kv, \
|
|
get_failed_operations
|
|
from ceo_common.interfaces import IConfig
|
|
from ceo_common.utils import get_current_term, add_term, get_max_term
|
|
from ceod.transactions.members import (
|
|
AddMemberTransaction,
|
|
DeleteMemberTransaction,
|
|
)
|
|
|
|
|
|
@click.group()
|
|
def members():
|
|
pass
|
|
|
|
|
|
@members.command(short_help='Add a new member or club rep')
|
|
@click.argument('username')
|
|
@click.option('--cn', help='Full name', prompt='Full name')
|
|
@click.option('--program', required=False, help='Academic program')
|
|
@click.option('--terms', 'num_terms', type=click.IntRange(1, 100),
|
|
help='Number of terms to add', prompt='Number of terms')
|
|
@click.option('--clubrep', is_flag=True, default=False,
|
|
help='Add non-member terms instead of member terms')
|
|
@click.option('--forwarding-address', required=False,
|
|
help=('Forwarding address to set in ~/.forward. '
|
|
'Default is UW address. '
|
|
'Set to the empty string to disable forwarding.'))
|
|
def add(username, cn, program, num_terms, clubrep, forwarding_address):
|
|
cfg = component.getUtility(IConfig)
|
|
uw_domain = cfg.get('uw_domain')
|
|
|
|
current_term = get_current_term()
|
|
terms = [current_term]
|
|
for _ in range(1, num_terms):
|
|
term = add_term(terms[-1])
|
|
terms.append(term)
|
|
if forwarding_address is None:
|
|
forwarding_address = username + '@' + uw_domain
|
|
|
|
click.echo("The following user will be created:")
|
|
lines = [
|
|
('uid', username),
|
|
('cn', cn),
|
|
]
|
|
if program is not None:
|
|
lines.append(('program', program))
|
|
if clubrep:
|
|
lines.append(('non-member terms', ','.join(terms)))
|
|
else:
|
|
lines.append(('member terms', ','.join(terms)))
|
|
if forwarding_address != '':
|
|
lines.append(('forwarding address', forwarding_address))
|
|
print_colon_kv(lines)
|
|
|
|
click.confirm('Do you want to continue?', abort=True)
|
|
|
|
body = {
|
|
'uid': username,
|
|
'cn': cn,
|
|
}
|
|
if program is not None:
|
|
body['program'] = program
|
|
if clubrep:
|
|
body['terms'] = terms
|
|
else:
|
|
body['non_member_terms'] = terms
|
|
if forwarding_address != '':
|
|
body['forwarding_addresses'] = [forwarding_address]
|
|
resp = http_post('/api/members', json=body)
|
|
data = handle_stream_response(resp, AddMemberTransaction.operations)
|
|
result = data[-1]['result']
|
|
print_user_lines(result)
|
|
|
|
failed_operations = get_failed_operations(data)
|
|
if 'send_welcome_message' in failed_operations:
|
|
click.echo(click.style(
|
|
'Warning: welcome message was not sent. You now need to manually '
|
|
'send the user their password.', fg='yellow'))
|
|
|
|
|
|
def print_user_lines(result: Dict):
|
|
"""Pretty-print a user JSON response."""
|
|
lines = [
|
|
('uid', result['uid']),
|
|
('cn', result['cn']),
|
|
('program', result.get('program', 'Unknown')),
|
|
('UID number', result['uid_number']),
|
|
('GID number', result['gid_number']),
|
|
('login shell', result['login_shell']),
|
|
('home directory', result['home_directory']),
|
|
('is a club', result['is_club']),
|
|
]
|
|
if 'forwarding_addresses' in result:
|
|
lines.append(('forwarding addresses', ','.join(result['forwarding_addresses'])))
|
|
if 'terms' in result:
|
|
lines.append(('terms', ','.join(result['terms'])))
|
|
if 'non_member_terms' in result:
|
|
lines.append(('non-member terms', ','.join(result['non_member_terms'])))
|
|
if 'password' in result:
|
|
lines.append(('password', result['password']))
|
|
print_colon_kv(lines)
|
|
|
|
|
|
@members.command(short_help='Get info about a user')
|
|
@click.argument('username')
|
|
def get(username):
|
|
resp = http_get('/api/members/' + username)
|
|
result = handle_sync_response(resp)
|
|
print_user_lines(result)
|
|
|
|
|
|
@members.command(short_help="Replace a user's login shell or forwarding addresses")
|
|
@click.argument('username')
|
|
@click.option('--login-shell', required=False, help='Login shell')
|
|
@click.option('--forwarding-addresses', required=False,
|
|
help='Comma-separated list of forwarding addresses')
|
|
def modify(username, login_shell, forwarding_addresses):
|
|
if login_shell is None and forwarding_addresses is None:
|
|
click.echo('Nothing to do.')
|
|
sys.exit()
|
|
operations = []
|
|
body = {}
|
|
if login_shell is not None:
|
|
body['login_shell'] = login_shell
|
|
operations.append('replace_login_shell')
|
|
click.echo('Login shell will be set to: ' + login_shell)
|
|
if forwarding_addresses is not None:
|
|
forwarding_addresses = forwarding_addresses.split(',')
|
|
body['forwarding_addresses'] = forwarding_addresses
|
|
operations.append('replace_forwarding_addresses')
|
|
prefix = '~/.forward will be set to: '
|
|
click.echo(prefix + forwarding_addresses[0])
|
|
for address in forwarding_addresses[1:]:
|
|
click.echo((' ' * len(prefix)) + address)
|
|
|
|
click.confirm('Do you want to continue?', abort=True)
|
|
|
|
resp = http_patch('/api/members/' + username, json=body)
|
|
handle_stream_response(resp, operations)
|
|
|
|
|
|
@members.command(short_help="Renew a member or club rep's membership")
|
|
@click.argument('username')
|
|
@click.option('--terms', 'num_terms', type=click.IntRange(1, 100),
|
|
help='Number of terms to add', prompt='Number of terms')
|
|
@click.option('--clubrep', is_flag=True, default=False,
|
|
help='Add non-member terms instead of member terms')
|
|
def renew(username, num_terms, clubrep):
|
|
resp = http_get('/api/members/' + username)
|
|
result = handle_sync_response(resp)
|
|
max_term = None
|
|
if clubrep and 'non_member_terms' in result:
|
|
max_term = get_max_term(result['non_member_terms'])
|
|
elif not clubrep and 'terms' in result:
|
|
max_term = get_max_term(result['terms'])
|
|
if max_term is not None:
|
|
max_term = get_max_term([max_term, get_current_term()])
|
|
else:
|
|
max_term = get_current_term()
|
|
|
|
terms = [add_term(max_term)]
|
|
for _ in range(1, num_terms):
|
|
term = add_term(terms[-1])
|
|
terms.append(term)
|
|
|
|
if clubrep:
|
|
body = {'non_member_terms': terms}
|
|
click.echo('The following non-member terms will be added: ' + ','.join(terms))
|
|
else:
|
|
body = {'terms': terms}
|
|
click.echo('The following member terms will be added: ' + ','.join(terms))
|
|
|
|
click.confirm('Do you want to continue?', abort=True)
|
|
|
|
resp = http_post(f'/api/members/{username}/renew', json=body)
|
|
handle_sync_response(resp)
|
|
click.echo('Done.')
|
|
|
|
|
|
@members.command(short_help="Reset a user's password")
|
|
@click.argument('username')
|
|
def pwreset(username):
|
|
click.confirm(f"Are you sure you want to reset {username}'s password?", abort=True)
|
|
resp = http_post(f'/api/members/{username}/pwreset')
|
|
result = handle_sync_response(resp)
|
|
click.echo('New password: ' + result['password'])
|
|
|
|
|
|
@members.command(short_help="Delete a user")
|
|
@click.argument('username')
|
|
def delete(username):
|
|
# a hack to determine if we're in the dev environment
|
|
if not socket.getfqdn().endswith('.csclub.internal'):
|
|
click.echo('This command may only be called during development.')
|
|
sys.exit(1)
|
|
click.confirm(f"Are you sure you want to delete {username}?", abort=True)
|
|
resp = http_delete(f'/api/members/{username}')
|
|
handle_stream_response(resp, DeleteMemberTransaction.operations)
|