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)