initial

1 files changed, 38 insertions, 0 deletions

diff --git a/site/service/db.py b/site/service/db.py
new file mode 100644
index 0000000..1cb69b8
--- /dev/null
+++ b/site/service/db.py
@@ -0,0 +1,38 @@
+from peewee import *
+import datetime
+from playhouse.postgres_ext import *
+import config
+
+_db = PostgresqlExtDatabase(
+    config.dbname,
+    host=config.dbhost,
+    user=config.dbuser,
+    password=config.dbpassword
+)
+
+class BaseModel(Model):
+    class Meta:
+        database = _db
+
+
+class Users(BaseModel):
+    email = TextField()
+    password = TextField()
+    sessionkey = TextField()
+    created = DateTimeField(default=datetime.datetime.now)
+
+class Session(BaseModel):
+    user = ForeignKeyField(Users, null=True)
+    token = TextField(null=True)
+    created = DateTimeField(default=datetime.datetime.now)
+
+class JData(BaseModel):
+    session = ForeignKeyField(Session, null=True)
+    category = TextField(null=True, index=True)
+    data = JSONField()
+    created = DateTimeField(default=datetime.datetime.now)
+
+
+if __name__ == "__main__":
+    _db.create_tables(BaseModel.__subclasses__(), safe=True)
+