mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-08-19 05:59:44 -04:00
Refactor add-service form such that the dao is not exposed in the forms.
This commit is contained in:
@@ -1,5 +1,7 @@
|
||||
from datetime import datetime
|
||||
|
||||
from sqlalchemy.orm import load_only
|
||||
|
||||
from app import db
|
||||
from app.models import Service
|
||||
|
||||
@@ -39,3 +41,7 @@ def add_service(service):
|
||||
|
||||
def find_service_by_service_name(service_name):
|
||||
return Service.query.filter_by(name=service_name).first()
|
||||
|
||||
|
||||
def find_all_service_names():
|
||||
return [x.name for x in Service.query.options(load_only("name")).all()]
|
||||
|
||||
@@ -2,10 +2,9 @@ from datetime import datetime
|
||||
|
||||
from flask import session
|
||||
from flask_wtf import Form
|
||||
from wtforms import StringField, PasswordField
|
||||
from wtforms import StringField, PasswordField, ValidationError
|
||||
from wtforms.validators import DataRequired, Email, Length, Regexp
|
||||
|
||||
from app.main.dao import verify_codes_dao, services_dao
|
||||
from app.main.dao import verify_codes_dao
|
||||
from app.main.encryption import check_hash
|
||||
from app.main.validators import Blacklist
|
||||
|
||||
@@ -83,14 +82,19 @@ class TextNotReceivedForm(Form):
|
||||
|
||||
|
||||
class AddServiceForm(Form):
|
||||
def __init__(self, service_names, *args, **kwargs):
|
||||
self.service_names = service_names
|
||||
super(AddServiceForm, self).__init__(*args, **kwargs)
|
||||
|
||||
service_name = StringField(validators=[DataRequired(message='Please enter your service name')])
|
||||
|
||||
def validate_service_name(self, a):
|
||||
if services_dao.find_service_by_service_name(self.service_name.data) is not None:
|
||||
self.service_name.errors.append('Duplicate service name')
|
||||
return False
|
||||
else:
|
||||
return True
|
||||
print("a {}".format(a))
|
||||
print("self.service_name {}".format(self.service_name.data))
|
||||
print("service_names {}".format(self.service_names))
|
||||
if self.service_name.data in self.service_names:
|
||||
print('here')
|
||||
raise ValidationError('Service name already exists')
|
||||
|
||||
|
||||
def validate_codes(field, code_type):
|
||||
|
||||
@@ -1,25 +1,20 @@
|
||||
from flask import render_template, jsonify, redirect, session
|
||||
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
|
||||
|
||||
|
||||
@main.route("/add-service", methods=['GET'])
|
||||
@main.route("/add-service", methods=['GET', 'POST'])
|
||||
@login_required
|
||||
def add_service():
|
||||
return render_template('views/add-service.html', form=AddServiceForm())
|
||||
|
||||
|
||||
@main.route("/add-service", methods=['POST'])
|
||||
@login_required
|
||||
def process_add_service():
|
||||
form = AddServiceForm()
|
||||
|
||||
form = AddServiceForm(services_dao.find_all_service_names())
|
||||
print("serivce_names {}".format(services_dao.find_all_service_names()))
|
||||
if form.validate_on_submit():
|
||||
|
||||
user = users_dao.get_user_by_id(session['user_id'])
|
||||
services_dao.insert_new_service(form.service_name.data, user)
|
||||
return redirect('/dashboard')
|
||||
else:
|
||||
return jsonify(form.errors), 400
|
||||
return render_template('views/add-service.html', form=form)
|
||||
§
|
||||
Reference in New Issue
Block a user