gear-orders/db/models.py

43 lines
1 KiB
Python
Raw Normal View History

2025-11-14 04:03:21 +00:00
from peewee import *
from settings import SQLITE_DB
database = SqliteDatabase(SQLITE_DB)
class BaseModel(Model):
class Meta:
database = database
class PunishmentStatus(BaseModel):
confirmed_at = DateTimeField(null=True) # TIMESTAMP
created_at = DateTimeField() # TIMESTAMP
mastodon_id = TextField()
text = TextField()
class Meta:
table_name = 'punishment_status'
class OrderStatus(BaseModel):
confirmed_at = DateTimeField(null=True) # TIMESTAMP
created_at = DateTimeField() # TIMESTAMP
due_at = DateTimeField() # TIMESTAMP
mastodon_id = TextField()
punishment = ForeignKeyField(column_name='punishment_id', field='id', model=PunishmentStatus, null=True)
text = TextField()
class Meta:
table_name = 'order_status'
class Repeat(BaseModel):
count = IntegerField(default=0)
orders = TextField()
probability = FloatField()
class Meta:
table_name = 'repeat'
class SkipDay(BaseModel):
date = DateField(unique=True)
class Meta:
table_name = 'skip_day'