API: Create and Update support

1 files changed, 20 insertions, 4 deletions

diff --git a/budget/forms.py b/budget/forms.py
index 7ac48cc..cfd5788 100644
--- a/budget/forms.py
+++ b/budget/forms.py
@@ -1,6 +1,6 @@
 from flaskext.wtf import *
 from wtforms.widgets import html_params
-from models import Project, Person, Bill
+from models import Project, Person, Bill, db
 from datetime import datetime
 
 
@@ -39,6 +39,15 @@ class ProjectForm(Form):
                 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.id = self.id.data
+        project.password = self.password.data
+        project.contact_email = self.contact_email.data
+
+        return project
+
 
 class AuthenticationForm(Form):
     id = TextField("Project identifier", validators=[Required()])
@@ -76,19 +85,26 @@ class BillForm(Form):
 
 
 class MemberForm(Form):
-    def __init__(self, project, *args, **kwargs):
-        super(MemberForm, self).__init__(*args, **kwargs)
-        self.project = project
 
     name = TextField("Name", validators=[Required()])
     submit = SubmitField("Add a member")
 
+    def __init__(self, project, *args, **kwargs):
+        super(MemberForm, self).__init__(*args, **kwargs)
+        self.project = project
+
     def validate_name(form, field):
         if Person.query.filter(Person.name == field.data)\
                 .filter(Person.project == form.project)\
                 .filter(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
+
+        return person
 
 class InviteForm(Form):
     emails = TextAreaField("People to notify")