DamageManagement/models.py

35 lines
682 B
Python

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)