2016-01-18 10:47:53 +00:00
|
|
|
from flask import request, render_template, jsonify, redirect, session, url_for, abort
|
2015-12-14 17:12:28 +00:00
|
|
|
from flask_login import login_required
|
|
|
|
|
from app.main import main
|
2015-12-15 09:28:15 +00:00
|
|
|
from app.main.dao import services_dao, users_dao
|
2015-12-14 17:12:28 +00:00
|
|
|
from app.main.forms import AddServiceForm
|
|
|
|
|
|
|
|
|
|
|
2016-01-04 15:31:50 +00:00
|
|
|
@main.route("/add-service", methods=['GET', 'POST'])
|
2016-01-18 10:47:53 +00:00
|
|
|
@main.route("/add-service/<string:first>", methods=['GET', 'POST'])
|
2015-12-14 17:12:28 +00:00
|
|
|
@login_required
|
2016-01-18 10:47:53 +00:00
|
|
|
def add_service(first=False):
|
|
|
|
|
if first:
|
|
|
|
|
if first == 'first':
|
|
|
|
|
heading = 'Set up notifications for your service'
|
|
|
|
|
else:
|
|
|
|
|
abort(404)
|
|
|
|
|
else:
|
|
|
|
|
heading = 'Add a new service'
|
|
|
|
|
|
2016-01-04 15:31:50 +00:00
|
|
|
form = AddServiceForm(services_dao.find_all_service_names())
|
2015-12-14 17:12:28 +00:00
|
|
|
if form.validate_on_submit():
|
2015-12-15 09:28:15 +00:00
|
|
|
user = users_dao.get_user_by_id(session['user_id'])
|
|
|
|
|
services_dao.insert_new_service(form.service_name.data, user)
|
2016-01-13 12:10:29 +00:00
|
|
|
return redirect(url_for('.dashboard', service_id=123))
|
2015-12-14 17:12:28 +00:00
|
|
|
else:
|
2016-01-18 10:47:53 +00:00
|
|
|
return render_template(
|
|
|
|
|
'views/add-service.html',
|
|
|
|
|
form=form,
|
|
|
|
|
heading=heading
|
|
|
|
|
)
|