mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-02-05 10:53:28 -05:00
6ba13a6513
register. On succesful register and verfication they will be added to service and forwarded to dashboard. Nothing is done yet with the permissions requested in the invite to the user.
42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
from flask import (
|
|
render_template,
|
|
redirect,
|
|
session,
|
|
url_for
|
|
)
|
|
|
|
from flask_login import login_required
|
|
|
|
from app.main import main
|
|
from app.main.dao import services_dao, users_dao
|
|
from app.main.forms import AddServiceForm
|
|
from app import user_api_client
|
|
|
|
|
|
@main.route("/add-service", methods=['GET', 'POST'])
|
|
@login_required
|
|
def add_service():
|
|
|
|
invited_user = session.get('invited_user')
|
|
if invited_user:
|
|
# if invited user add to service and redirect to dashboard
|
|
user = users_dao.get_user_by_id(session['user_id'])
|
|
service_id = invited_user['service']
|
|
user_api_client.add_user_to_service(service_id, user.id)
|
|
session.pop('invited_user', None)
|
|
return redirect(url_for('main.service_dashboard', service_id=service_id))
|
|
|
|
form = AddServiceForm(services_dao.find_all_service_names)
|
|
heading = 'Which service do you want to set up notifications for?'
|
|
if form.validate_on_submit():
|
|
session['service_name'] = form.name.data
|
|
user = users_dao.get_user_by_id(session['user_id'])
|
|
service_id = services_dao.insert_new_service(session['service_name'], user.id)
|
|
return redirect(url_for('main.service_dashboard', service_id=service_id))
|
|
else:
|
|
return render_template(
|
|
'views/add-service.html',
|
|
form=form,
|
|
heading=heading
|
|
)
|