mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-02-05 10:53:28 -05:00
5f3c72729e
This commit only deals with acceptance by users who are already in system. Changed invite client to return invited user objects instead of dictionaries. Added commented out test. fixed up fixtures to return invited user object for invites
128 lines
3.6 KiB
Python
128 lines
3.6 KiB
Python
from flask import (
|
||
request,
|
||
render_template,
|
||
redirect,
|
||
abort,
|
||
url_for,
|
||
flash
|
||
)
|
||
|
||
from flask_login import (
|
||
login_required,
|
||
current_user
|
||
)
|
||
|
||
from notifications_python_client.errors import HTTPError
|
||
|
||
from app.main import main
|
||
from app.main.forms import InviteUserForm
|
||
from app.main.dao.services_dao import get_service_by_id_or_404
|
||
from app import user_api_client
|
||
from app import invite_api_client
|
||
|
||
fake_users = [
|
||
{
|
||
'name': '',
|
||
'permission_send_messages': True,
|
||
'permission_manage_service': True,
|
||
'permission_manage_api_keys': True,
|
||
'active': True
|
||
}
|
||
]
|
||
|
||
|
||
@main.route("/services/<service_id>/users")
|
||
@login_required
|
||
def manage_users(service_id):
|
||
try:
|
||
users = user_api_client.get_users_for_service(service_id=service_id)
|
||
invited_users = invite_api_client.get_invites_for_service(service_id=service_id)
|
||
return render_template('views/manage-users.html',
|
||
service_id=service_id,
|
||
users=users,
|
||
current_user=current_user,
|
||
invited_users=invited_users)
|
||
except HTTPError as e:
|
||
if e.status_code == 404:
|
||
abort(404)
|
||
else:
|
||
raise e
|
||
|
||
|
||
@main.route("/services/<service_id>/users/invite", methods=['GET', 'POST'])
|
||
@login_required
|
||
def invite_user(service_id):
|
||
|
||
form = InviteUserForm()
|
||
if form.validate_on_submit():
|
||
email_address = form.email_address.data
|
||
permissions = _get_permissions(request.form)
|
||
try:
|
||
invited_user = invite_api_client.create_invite(current_user.id, service_id, email_address, permissions)
|
||
flash('Invite sent to {}'.format(invited_user.email_address), 'default_with_tick')
|
||
return redirect(url_for('.manage_users', service_id=service_id))
|
||
|
||
except HTTPError as e:
|
||
if e.status_code == 404:
|
||
abort(404)
|
||
else:
|
||
raise e
|
||
|
||
return render_template(
|
||
'views/invite-user.html',
|
||
user={},
|
||
service=get_service_by_id_or_404(service_id),
|
||
service_id=service_id,
|
||
form=form
|
||
)
|
||
|
||
|
||
@main.route("/services/<service_id>/users/<user_id>", methods=['GET', 'POST'])
|
||
@login_required
|
||
def edit_user(service_id, user_id):
|
||
|
||
if request.method == 'POST':
|
||
return redirect(url_for('.manage_users', service_id=service_id))
|
||
|
||
return render_template(
|
||
'views/invite-user.html',
|
||
user=fake_users[int(user_id)],
|
||
user_id=user_id,
|
||
service=get_service_by_id_or_404(service_id),
|
||
service_id=service_id
|
||
)
|
||
|
||
|
||
@main.route("/services/<service_id>/users/<user_id>/delete", methods=['GET', 'POST'])
|
||
@login_required
|
||
def delete_user(service_id, user_id):
|
||
|
||
if request.method == 'POST':
|
||
return redirect(url_for('.manage_users', service_id=service_id))
|
||
|
||
user = fake_users[int(user_id)]
|
||
|
||
flash(
|
||
'Are you sure you want to delete {}’s account?'.format(user.get('name') or user['email_localpart']),
|
||
'delete'
|
||
)
|
||
|
||
return render_template(
|
||
'views/invite-user.html',
|
||
user=user,
|
||
user_id=user_id,
|
||
service=get_service_by_id_or_404(service_id),
|
||
service_id=service_id
|
||
)
|
||
|
||
|
||
def _get_permissions(form):
|
||
permissions = []
|
||
if form.get('send_messages') and form['send_messages'] == 'yes':
|
||
permissions.append('send_messages')
|
||
if form.get('manage_service') and form['manage_service'] == 'yes':
|
||
permissions.append('manage_service')
|
||
if form.get('manage_api_keys') and form['manage_api_keys'] == 'yes':
|
||
permissions.append('manage_api_keys')
|
||
return ','.join(permissions)
|