diff --git a/main.py b/main.py index e692e43..b19dbc8 100755 --- a/main.py +++ b/main.py @@ -1,9 +1,12 @@ from flask import Flask, render_template, flash, request import requests from requests import HTTPError -from wtforms import Form, StringField, validators +from wtforms import Form, StringField, validators, SubmitField import json +from wtforms.validators import InputRequired +from wtforms.validators import Optional + app = Flask(__name__) app.config['SECRET_KEY'] = '255771f2872f27567d441f2b62425' @@ -65,9 +68,10 @@ def flash_address(results): return class ReusableForm(Form): - postalcode = StringField('Postcode:', validators=[validators.InputRequired(), validators.Length(min=6, max=6)]) - houseno = StringField('Huis nummer:', validators=[validators.InputRequired()]) - houseno_ext = StringField('Toevoeging:', validators=[]) + postalcode = StringField('Postcode', [InputRequired(), validators.Length(min=6, max=6, message='Formaat 1111AA')]) + houseno = StringField('Huisnummer', [InputRequired(),validators.Length(max=5)]) + houseno_ext = StringField('Toevoeging', [Optional(), validators.Length(max=10)]) + submit = SubmitField('Zoek') @app.route('/', methods=['GET', 'POST']) def home(): @@ -81,10 +85,17 @@ class ReusableForm(Form): houseno = request.form['houseno'] houseno_ext = request.form['houseno_ext'] results = search_mp('ELK', postalcode, houseno, houseno_ext) - flash_address(results) - flash_results(results) + if len(results['meteringPoints']) == 0: + flash('Geen ELK aansluiting gevonden bij dit adres', 'info') + else: + flash_address(results) + flash_results(results) + results = search_mp('GAS', postalcode, houseno, houseno_ext) - flash_results(results) + if len(results['meteringPoints']) == 0: + flash('Geen GAS aansluiting gevonden bij dit adres', 'info') + else: + flash_results(results) return render_template('index.html', form=form) diff --git a/templates/index.html b/templates/index.html index e53df5b..920c99c 100755 --- a/templates/index.html +++ b/templates/index.html @@ -6,22 +6,22 @@ {{ form.csrf }}
- -
+ +
{{ form.postalcode(class="form-control") }}
- -
+ +
{{ form.houseno(class="form-control") }}
- -
+ +
{{ form.houseno_ext(class="form-control") }}
- + {{ form.submit(class="input-submit")}}