mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-02-05 10:53:28 -05:00
27 lines
970 B
Python
27 lines
970 B
Python
from flask import render_template, redirect, session, url_for
|
|
from flask_login import login_required, current_user
|
|
from app.main import main
|
|
from app.main.dao import services_dao, users_dao
|
|
from app.main.forms import AddServiceForm
|
|
|
|
|
|
@main.route("/add-service", methods=['GET', 'POST'])
|
|
@login_required
|
|
def add_service():
|
|
form = AddServiceForm(services_dao.find_all_service_names)
|
|
services = services_dao.get_services(current_user.id)
|
|
if len(services) > 0:
|
|
heading = 'Set up notifications for your service'
|
|
else:
|
|
heading = 'Add a new service'
|
|
if form.validate_on_submit():
|
|
user = users_dao.get_user_by_id(session['user_id'])
|
|
service_id = services_dao.insert_new_service(form.name.data, 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
|
|
)
|