add ability to create new groups

1 files changed, 8 insertions, 1 deletions

diff --git a/models.py b/models.py
index b59ae66..5cc0d4e 100644
--- a/models.py
+++ b/models.py
@@ -1,7 +1,7 @@
 from peewee import CharField, DateTimeField, BooleanField, ForeignKeyField
 from peewee import create_model_tables
 from datetime import datetime
-from app import db
+from app import db, pad
 
 class User(db.Model):
     username = CharField()
@@ -21,6 +21,13 @@ class Group(db.Model):
     def __unicode__(self):
         return self.name
 
+    def save(self, force_insert=False, only=None):
+        if self.id is None or force_insert:
+            self.api_id = pad.createGroup()
+
+        super(Group, self).save(force_insert=force_insert, only=only)
+
+
 class Member(db.Model):
     group = ForeignKeyField(Group, related_name='members')
     user = ForeignKeyField(User, related_name='groups')