from collections import defaultdict from flask import * from flaskext.mail import Mail, Message # local modules from models import db, Project, Person, Bill from forms import (ProjectForm, AuthenticationForm, BillForm, MemberForm, InviteForm, CreateArchiveForm) from utils import get_billform_for, Redirect303 # create the application, initialize stuff app = Flask(__name__) app.config.from_object("default_settings") mail = Mail() # db db.init_app(app) db.app = app db.create_all() # mail mail.init_app(app) @app.url_defaults def add_project_id(endpoint, values): """Add the project id to the url calls if it is expected. This is to not carry it everywhere in the templates. """ if 'project_id' in values or not hasattr(g, 'project'): return if app.url_map.is_endpoint_expecting(endpoint, 'project_id'): values['project_id'] = g.project.id @app.url_value_preprocessor def pull_project(endpoint, values): """When a request contains a project_id value, transform it directly into a project by checking the credentials are stored in session. If not, redirect the user to an authentication form """ if endpoint == "authenticate": return if not values: values = {} project_id = values.pop('project_id', None) if project_id: project = Project.query.get(project_id) if not project: raise Redirect303(url_for("create_project", project_id=project_id)) if project.id in session and session[project.id] == project.password: # add project into kwargs and call the original function g.project = project else: # redirect to authentication page raise Redirect303( url_for("authenticate", project_id=project_id)) @app.route("/authenticate", methods=["GET", "POST"]) def authenticate(project_id=None): """Authentication form""" form = AuthenticationForm() if not form.id.data and request.args['project_id']: form.id.data = request.args['project_id'] project_id = form.id.data project = Project.query.get(project_id) create_project = False # We don't want to create the project by default if not project: # But if the user try to connect to an unexisting project, we will # propose him a link to the creation form. create_project = project_id else: # if credentials are already in session, redirect if project_id in session and project.password == session[project_id]: setattr(g, 'project', project) return redirect(url_for("list_bills")) # else process the form if request.method == "POST": if form.validate(): if not form.password.data == project.password: form.errors['password'] = ["The password is not the right one"] else: # maintain a list of visited projects if "projects" not in session: session["projects"] = [] # add the project on the top of the list session["projects"].insert(0, (project_id, project.name)) session[project_id] = form.password.data session.update() setattr(g, 'project', project) return redirect(url_for("list_bills")) return render_template("authenticate.html", form=form, create_project=create_project) @app.route("/") def home(): project_form = ProjectForm() auth_form = AuthenticationForm() return render_template("home.html", project_form=project_form, auth_form=auth_form, session=session) @app.route("/create", methods=["GET", "POST"]) def create_project(): form = ProjectForm() if request.method == "GET" and 'project_id' in request.values: form.name.data = request.values['project_id'] if request.method == "POST": if form.validate(): # save the object in the db project = form.save() db.session.add(project) db.session.commit() # create the session object (authenticate) session[project.id] = project.password session.update() # redirect the user to the next step (invite) return redirect(url_for("invite", project_id=project.id)) return render_template("create_project.html", form=form) @app.route("/exit") def exit(): # delete the session session.clear() return redirect(url_for("home")) @app.route("/demo") def demo(): """ Authenticate the user for the demonstration project and redirect him to the bills list for this project. Create a demo project if it doesnt exists yet (or has been deleted) """ project = Project.query.get("demo") if not project: project = Project(id="demo", name=u"demonstration", password="demo", contact_email="demo@notmyidea.org") db.session.add(project) db.session.commit() session[project.id] = project.password return redirect(url_for("list_bills", project_id=project.id)) @app.route("//invite", methods=["GET", "POST"]) def invite(): """Send invitations for this particular project""" form = InviteForm() if request.method == "POST": if form.validate(): # send the email message_body = render_template("invitation_mail") message_title = "You have been invited to share your"\ + " expenses for %s" % g.project.name msg = Message(message_title, body=message_body, recipients=[email.strip() for email in form.emails.data.split(",")]) mail.send(msg) flash("You invitations have been sent") return redirect(url_for("list_bills")) return render_template("send_invites.html", form=form) @app.route("//") def list_bills(): bills = g.project.get_bills() return render_template("list_bills.html", bills=bills, member_form=MemberForm(g.project), bill_form=get_billform_for(g.project) ) @app.route("//members/add", methods=["GET", "POST"]) def add_member(): # FIXME manage form errors on the list_bills page form = MemberForm(g.project) if request.method == "POST": if form.validate(): # if the user is already bound to the project, just reactivate him person = Person.query.filter(Person.name == form.name.data)\ .filter(Project.id == g.project.id).all() if person: person[0].activated = True db.session.commit() flash("%s is part of this project again" % person[0].name) return redirect(url_for("list_bills")) db.session.add(Person(name=form.name.data, project=g.project)) db.session.commit() return redirect(url_for("list_bills")) return render_template("add_member.html", form=form) @app.route("//members//reactivate", methods=["GET",]) def reactivate(member_id): person = Person.query.filter(Person.id == member_id)\ .filter(Project.id == g.project.id).all() if person: person[0].activated = True db.session.commit() flash("%s is part of this project again" % person[0].name) return redirect(url_for("list_bills")) @app.route("//members//delete", methods=["GET", "POST"]) def remove_member(member_id): member = g.project.remove_member(member_id) if member.activated == False: flash("User '%s' has been desactivated" % member.name) else: flash("User '%s' has been removed" % member.name) return redirect(url_for("list_bills")) @app.route("//add", methods=["GET", "POST"]) def add_bill(): form = get_billform_for(g.project) if request.method == 'POST': if form.validate(): bill = Bill() db.session.add(form.save(bill)) db.session.commit() flash("The bill has been added") return redirect(url_for('list_bills')) return render_template("add_bill.html", form=form) @app.route("//delete/") def delete_bill(bill_id): bill = Bill.query.get_or_404(bill_id) db.session.delete(bill) db.session.commit() flash("The bill has been deleted") return redirect(url_for('list_bills')) @app.route("//edit/", methods=["GET", "POST"]) def edit_bill(bill_id): bill = Bill.query.get_or_404(bill_id) form = get_billform_for(g.project, set_default=False) if request.method == 'POST' and form.validate(): form.save(bill) db.session.commit() flash("The bill has been modified") return redirect(url_for('list_bills')) form.fill(bill) return render_template("add_bill.html", form=form, edit=True) @app.route("//compute") def compute_bills(): """Compute the sum each one have to pay to each other and display it""" return render_template("compute_bills.html") @app.route("//archives/create") def create_archive(): form = CreateArchiveForm() if request.method == "POST": if form.validate(): pass flash("The data from XX to XX has been archived") return render_template("create_archive.html", form=form) def main(): app.run(host="0.0.0.0", debug=True) if __name__ == '__main__': main()