import peewee as pw
from peewee import DatabaseProxy, Model

db = DatabaseProxy()


class BaseModel(Model):
    id = pw.IntegerField(primary_key=True)

    class Meta:
        database = db


class LKW(BaseModel):
    chassis = pw.TextField(unique=True)
    license = pw.TextField()


class User(BaseModel):
    worker_id = pw.TextField()


class Damage(BaseModel):
    date = pw.DateField()
    time = pw.TimeField()
    path = pw.TextField()  # for images?
    lkw = pw.ForeignKeyField(LKW.id)


class DamageStatus(BaseModel):
    user_id = pw.ForeignKeyField(User.id)
    status = pw.IntegerField()
    datetime = pw.DateTimeField()
    damage_id = pw.ForeignKeyField(Damage.id)