models: add simple user model

2 files changed, 20 insertions, 0 deletions

diff --git a/main.py b/main.py
index cf23305..d4dcc7f 100755
--- a/main.py
+++ b/main.py
@@ -1,7 +1,9 @@
 #!/usr/bin/env python
 
 from app import app, db
+from models import create_tables
 
 if __name__ == '__main__':
     db.connect_db()
+    create_tables()
     app.run(host = '::')
diff --git a/models.py b/models.py
new file mode 100644
index 0000000..6ed3b48
--- /dev/null
+++ b/models.py
@@ -0,0 +1,18 @@
+from peewee import CharField, DateTimeField, BooleanField
+from peewee import create_model_tables
+from datetime import datetime
+from app import db
+
+class User(db.Model):
+    username = CharField()
+    api_id = CharField(null=True)
+    email = CharField()
+    last_login = DateTimeField(default=datetime.now)
+    active = BooleanField(default=True)
+    admin = BooleanField(default=False)
+
+    def __unicode__(self):
+        return self.username
+
+def create_tables():
+    create_model_tables([User], fail_silently = True)