mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-05-03 23:51:22 -04:00
b75ab7c3df
When you add a new service, it’s probably the one you want to do stuff with. When you get invited, the service you’ve been invited to is probably the one you want to use. This commit adds the ID of the new service or service you’ve been invited to to the session.
53 lines
1.9 KiB
Python
53 lines
1.9 KiB
Python
from flask import (
|
|
render_template,
|
|
redirect,
|
|
session,
|
|
url_for,
|
|
current_app)
|
|
|
|
from flask_login import login_required
|
|
|
|
from app.main import main
|
|
from app.main.forms import AddServiceForm
|
|
from app.notify_client.models import InvitedUser
|
|
|
|
from app import (
|
|
invite_api_client,
|
|
user_api_client,
|
|
service_api_client
|
|
)
|
|
from app.utils import email_safe
|
|
|
|
|
|
@main.route("/add-service", methods=['GET', 'POST'])
|
|
@login_required
|
|
def add_service():
|
|
invited_user = session.get('invited_user')
|
|
if invited_user:
|
|
invitation = InvitedUser(**invited_user)
|
|
# if invited user add to service and redirect to dashboard
|
|
user = user_api_client.get_user(session['user_id'])
|
|
service_id = invited_user['service']
|
|
user_api_client.add_user_to_service(service_id, user.id, invitation.permissions)
|
|
invite_api_client.accept_invite(service_id, invitation.id)
|
|
return redirect(url_for('main.service_dashboard', service_id=service_id))
|
|
|
|
form = AddServiceForm(service_api_client.find_all_service_email_from)
|
|
heading = 'Which service do you want to set up notifications for?'
|
|
if form.validate_on_submit():
|
|
email_from = email_safe(form.name.data)
|
|
service_id = service_api_client.create_service(service_name=form.name.data,
|
|
active=False,
|
|
message_limit=current_app.config['DEFAULT_SERVICE_LIMIT'],
|
|
restricted=True,
|
|
user_id=session['user_id'],
|
|
email_from=email_from)
|
|
session['service_id'] = service_id
|
|
return redirect(url_for('main.tour', page=1))
|
|
else:
|
|
return render_template(
|
|
'views/add-service.html',
|
|
form=form,
|
|
heading=heading
|
|
)
|