st-ten-1/src/lib/db/models/archive.py

35 lines
1.1 KiB
Python
Raw Normal View History

2022-06-01 16:37:19 +00:00
from datetime import datetime
from peewee import AutoField, BooleanField, DateTimeField, ForeignKeyField
from playhouse.sqlite_ext import JSONField
from .base_model import BaseModel, db
from .recipes import Recipes
from .users import Users
class Archive(BaseModel):
id = AutoField(primary_key=True, unique=True, null=False)
time = DateTimeField(unique=True, null=False, default=datetime.now)
user = ForeignKeyField(Users, Users.username, null=False)
recipe = ForeignKeyField(Recipes, null=False)
result = BooleanField(null=False)
overridden = BooleanField(null=False)
test_data = JSONField(null=False)
archived = BooleanField(null=False, default=False)
uploaded = BooleanField(null=False, default=False)
@classmethod
@db.atomic()
def archive(cls, recipe, test_data, result, overridden):
return cls.create(
user=Users.get_session().user,
recipe=recipe,
test_data=test_data,
result=result,
overridden=overridden,
)
class Meta:
table_name = "archive"