diff --git a/models.py b/models.py new file mode 100644 index 0000000..422e81b --- /dev/null +++ b/models.py @@ -0,0 +1,34 @@ +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)