aboutsummaryrefslogtreecommitdiff
path: root/budget/forms.py
diff options
context:
space:
mode:
authorAlexis Metaireau <alexis@notmyidea.org>2017-07-07 00:06:56 +0200
committerGitHub <noreply@github.com>2017-07-07 00:06:56 +0200
commit3a4282fd75e3b3317b2b08b4aa2e6ac154310e73 (patch)
tree9470c907ba1f884246af87d26d55c3aaac6d6dc5 /budget/forms.py
parent0e374cd5e0ef5a9be67084365f91de2ab84f636c (diff)
downloadihatemoney-mirror-3a4282fd75e3b3317b2b08b4aa2e6ac154310e73.zip
ihatemoney-mirror-3a4282fd75e3b3317b2b08b4aa2e6ac154310e73.tar.gz
ihatemoney-mirror-3a4282fd75e3b3317b2b08b4aa2e6ac154310e73.tar.bz2
Absolute imports & some other improvements (#243)
* Use absolute imports and rename package to ihatemoney * Add a ihatemoney command * Factorize application creation logic * Refactor the tests * Update the wsgi.py module with the new create_app() function * Fix some styling thanks to Flake8. * Automate Flake8 check in the CI.
Diffstat (limited to 'budget/forms.py')
-rw-r--r--budget/forms.py191
1 files changed, 0 insertions, 191 deletions
diff --git a/budget/forms.py b/budget/forms.py
deleted file mode 100644
index bd7fc5b..0000000
--- a/budget/forms.py
+++ /dev/null
@@ -1,191 +0,0 @@
-from flask_wtf.form import FlaskForm
-from wtforms.fields.core import SelectField, SelectMultipleField
-from wtforms.fields.html5 import DateField, DecimalField
-from wtforms.fields.simple import PasswordField, SubmitField, TextAreaField, StringField
-from wtforms.validators import Email, Required, ValidationError
-from flask_babel import lazy_gettext as _
-from flask import request
-
-from wtforms.widgets import html_params
-from datetime import datetime
-from jinja2 import Markup
-
-from .models import Project, Person
-from .utils import slugify
-
-def get_billform_for(project, set_default=True, **kwargs):
- """Return an instance of BillForm configured for a particular project.
-
- :set_default: if set to True, on GET methods (usually when we want to
- display the default form, it will call set_default on it.
-
- """
- form = BillForm(**kwargs)
- form.payed_for.choices = form.payer.choices = [(m.id, m.name)
- for m in project.active_members]
- form.payed_for.default = [m.id for m in project.active_members]
-
- if set_default and request.method == "GET":
- form.set_default()
- return form
-
-
-class CommaDecimalField(DecimalField):
- """A class to deal with comma in Decimal Field"""
- def process_formdata(self, value):
- if value:
- value[0] = str(value[0]).replace(',', '.')
- return super(CommaDecimalField, self).process_formdata(value)
-
-
-class EditProjectForm(FlaskForm):
- name = StringField(_("Project name"), validators=[Required()])
- password = StringField(_("Private code"), validators=[Required()])
- contact_email = StringField(_("Email"), validators=[Required(), Email()])
-
- def save(self):
- """Create a new project with the information given by this form.
-
- Returns the created instance
- """
- project = Project(name=self.name.data, id=self.id.data,
- password=self.password.data,
- contact_email=self.contact_email.data)
- return project
-
- def update(self, project):
- """Update the project with the information from the form"""
- project.name = self.name.data
- project.password = self.password.data
- project.contact_email = self.contact_email.data
-
- return project
-
-
-class ProjectForm(EditProjectForm):
- id = StringField(_("Project identifier"), validators=[Required()])
- password = PasswordField(_("Private code"), validators=[Required()])
- submit = SubmitField(_("Create the project"))
-
- def validate_id(form, field):
- form.id.data = slugify(field.data)
- if (form.id.data == "dashboard") or Project.query.get(form.id.data):
- raise ValidationError(Markup(_("The project identifier is used "
- "to log in and for the URL of the project. "
- "We tried to generate an identifier for you but a project "
- "with this identifier already exists. "
- "Please create a new identifier "
- "that you will be able to remember.")))
-
-
-class AuthenticationForm(FlaskForm):
- id = StringField(_("Project identifier"), validators=[Required()])
- password = PasswordField(_("Private code"), validators=[Required()])
- submit = SubmitField(_("Get in"))
-
-
-class AdminAuthenticationForm(FlaskForm):
- admin_password = PasswordField(_("Admin password"), validators=[Required()])
- submit = SubmitField(_("Get in"))
-
-
-class PasswordReminder(FlaskForm):
- id = StringField(_("Project identifier"), validators=[Required()])
- submit = SubmitField(_("Send me the code by email"))
-
- def validate_id(form, field):
- if not Project.query.get(field.data):
- raise ValidationError(_("This project does not exists"))
-
-
-class BillForm(FlaskForm):
- date = DateField(_("Date"), validators=[Required()], default=datetime.now)
- what = StringField(_("What?"), validators=[Required()])
- payer = SelectField(_("Payer"), validators=[Required()], coerce=int)
- amount = CommaDecimalField(_("Amount paid"), validators=[Required()])
- payed_for = SelectMultipleField(_("For whom?"),
- validators=[Required()], coerce=int)
- submit = SubmitField(_("Submit"))
- submit2 = SubmitField(_("Submit and add a new one"))
-
- def save(self, bill, project):
- bill.payer_id = self.payer.data
- bill.amount = self.amount.data
- bill.what = self.what.data
- bill.date = self.date.data
- bill.owers = [Person.query.get(ower, project)
- for ower in self.payed_for.data]
-
- return bill
-
- def fill(self, bill):
- self.payer.data = bill.payer_id
- self.amount.data = bill.amount
- self.what.data = bill.what
- self.date.data = bill.date
- self.payed_for.data = [int(ower.id) for ower in bill.owers]
-
- def set_default(self):
- self.payed_for.data = self.payed_for.default
-
- def validate_amount(self, field):
- if field.data == 0:
- raise ValidationError(_("Bills can't be null"))
-
-
-class MemberForm(FlaskForm):
-
- name = StringField(_("Name"), validators=[Required()])
- weight = CommaDecimalField(_("Weight"), default=1)
- submit = SubmitField(_("Add"))
-
- def __init__(self, project, edit=False, *args, **kwargs):
- super(MemberForm, self).__init__(*args, **kwargs)
- self.project = project
- self.edit = edit
-
- def validate_name(form, field):
- if field.data == form.name.default:
- raise ValidationError(_("User name incorrect"))
- if (not form.edit and Person.query.filter(
- Person.name == field.data,
- Person.project == form.project,
- Person.activated == True).all()):
- raise ValidationError(_("This project already have this member"))
-
- def save(self, project, person):
- # if the user is already bound to the project, just reactivate him
- person.name = self.name.data
- person.project = project
- person.weight = self.weight.data
-
- return person
-
- def fill(self, member):
- self.name.data = member.name
- self.weight.data = member.weight
-
-
-class InviteForm(FlaskForm):
- emails = TextAreaField(_("People to notify"))
- submit = SubmitField(_("Send invites"))
-
- def validate_emails(form, field):
- validator = Email()
- for email in [email.strip() for email in form.emails.data.split(",")]:
- if not validator.regex.match(email):
- raise ValidationError(_("The email %(email)s is not valid",
- email=email))
-
-
-class ExportForm(FlaskForm):
- export_type = SelectField(_("What do you want to download ?"),
- validators=[Required()],
- coerce=str,
- choices=[("bills", _("bills")), ("transactions", _("transactions"))]
- )
- export_format = SelectField(_("Export file format"),
- validators=[Required()],
- coerce=str,
- choices=[("csv", "csv"), ("json", "json")]
- )