def pest():
todays_pest = sanitation.get_todays_tasks(config.pest)
pest_overview = sanitation.get_overview(config.pest)
context = {'todays_pest': todays_pest,
'overview': pest_overview}
form = auth.forms.AlertForm()
if request.method == "POST":
if request.form['buttons'] == 'send':
location = request.form['location']
topic = request.form['topic']
comment = request.form['comment']
msgtosend = comment + " From " + c_u.email
msg = Message(location + ": " + topic,
sender=config.MAIL_USERNAME,
recipients=[config.MAIL_USERNAME])
msg.body = msgtosend
mail.send(msg)
userid = c_u.user_id
if topic == 'Complete':
form.senddb(config.pest, location, comment, userid)
todays_pest = sanitation.get_todays_tasks(config.pest)
overview = sanitation.get_overview(config.pest)
context = {'todays_pest': todays_pest,
'overview': overview}
if request.form['buttons'] == 'save':
location = request.form['location']
days = request.form.getlist('day')
d = []
for day in days:
d.append(int(day))
description = request.form['description']
form.save(config.pest, location, d, description)
todays_pest = sanitation.get_todays_tasks(config.pest)
overview = sanitation.get_overview(config.pest)
context = {'todays_pest': todays_pest,
'overview': overview}
return render_template("pest.html", form=form, **context)
from flask.ext.wtf import Form
from wtforms import TextField, PasswordField, BooleanField,\
validators, fields
from wtforms.validators import DataRequired
from db import login, register, forgotpswd, sanitation
import config
todays_cleaning = sanitation.get_todays_tasks(config.cleaning)
todays_pest = sanitation.get_todays_tasks(config.pest)
class NewUser(Form):
firstname = TextField('Username',
[validators.Length(min=4, max=20)])
lastname = TextField('Username',
[validators.Length(min=4, max=20)])
email = TextField('Email Address',
[DataRequired('Must provide an email address')])
password = PasswordField('New Password',
[validators.Required(),
validators.EqualTo('password2',
message='Passwords must match')])
password2 = PasswordField('Repeat Password')
def validate(self):
if not Form.validate(self):
print 'Invalid!'
return False
print 'Valid!'
user = register.validate(self.firstname.data,
self.lastname.data, self.email.data, self.password.data)
if user:
