add public field for groups

3 files changed, 5 insertions, 2 deletions

diff --git a/forms.py b/forms.py
index ec643ac..a2c36fb 100644
--- a/forms.py
+++ b/forms.py
@@ -1,4 +1,4 @@
-from wtforms import Form, TextField, HiddenField, validators
+from wtforms import Form, BooleanField, TextField, HiddenField, validators
 from utils import Unique
 from models import Group
 
@@ -11,6 +11,7 @@ class CreateGroup(Form):
             message=u'Group name should not start with a special character.'),
         Unique(Group, Group.name,
             message=u'A group with this name already exists.')])
+    public = BooleanField('public')
 
 class DeleteGroup(Form):
     id = HiddenField('group id', [validators.Required()]);
diff --git a/models.py b/models.py
index 51e123a..c6721c1 100644
--- a/models.py
+++ b/models.py
@@ -17,6 +17,7 @@ class User(db.Model):
 class Group(db.Model):
     name = CharField(unique=True)
     api_id = CharField(null=True)
+    public = BooleanField(default=False)
 
     def __unicode__(self):
         return self.name
diff --git a/views.py b/views.py
index 9cb1bd7..5df76f8 100644
--- a/views.py
+++ b/views.py
@@ -17,7 +17,8 @@ def get_group_or_404(*query):
 def index():
     form = CreateGroup(request.form)
     if request.method == 'POST' and form.validate():
-        group = Group.create(name=form.name.data)
+        group = Group.create(name=form.name.data,
+                             public=form.public.data)
         Member.create(user=g.user, group=group, admin=True)
         form = CreateGroup()
     groups = [member.group for member in g.user.groups]