aboutsummaryrefslogtreecommitdiff
path: root/site/service/db.py
blob: 1cb69b8c7a10959e1b9abcb86a7d92dbc5357df5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
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)