Add missing import for validators, improve behaviour when an error occurs.

master
Dimitri Huisman 3 years ago
parent 7d801c560c
commit 55a601de5a

@ -2,6 +2,7 @@ from mailu import models
from mailu.ui import ui, forms, access from mailu.ui import ui, forms, access
from flask import current_app as app from flask import current_app as app
import validators
import flask import flask
import flask_login import flask_login
import wtforms_components import wtforms_components
@ -26,13 +27,13 @@ def domain_create():
flask.flash('Domain %s is already used' % form.name.data, 'error') flask.flash('Domain %s is already used' % form.name.data, 'error')
else: else:
domain = models.Domain() domain = models.Domain()
form.populate_obj(domain) form.populate_obj(domain)
models.db.session.add(domain) models.db.session.add(domain)
models.db.session.commit() models.db.session.commit()
flask.flash('Domain %s created' % domain) flask.flash('Domain %s created' % domain)
return flask.redirect(flask.url_for('.domain_list'))
else: else:
flask.flash('Domain %s is invalid' % form.name.data, 'error') flask.flash('Domain %s is invalid' % form.name.data, 'error')
return flask.redirect(flask.url_for('.domain_list'))
return flask.render_template('domain/create.html', form=form) return flask.render_template('domain/create.html', form=form)

Loading…
Cancel
Save