add ability to create new groups

1 files changed, 10 insertions, 3 deletions

diff --git a/views.py b/views.py
index e918d3c..c6f4a8a 100644
--- a/views.py
+++ b/views.py
@@ -1,11 +1,18 @@
 from app import app
 from auth import auth
-from flask import g
+from flask import g, request
+from models import Group, Member
+from forms import CreateGroup
 from utils import templated
 
-@app.route('/')
+@app.route('/', methods=['GET', 'POST'])
 @templated('index.html')
 @auth.login_required
 def index():
+    form = CreateGroup(request.form)
+    if request.method == 'POST' and form.validate():
+        group = Group.create(name=form.name.data)
+        Member.create(user=g.user, group=group, admin=True)
+        form = CreateGroup()
     groups = [member.group for member in g.user.groups]
-    return {'groups': groups}
+    return {'groups': groups, 'create_form': form}