44 lines
1.2 KiB
Python
44 lines
1.2 KiB
Python
import typing
|
|
|
|
import tortoise
|
|
import tortoise.validators
|
|
|
|
from app.models.base import AbstractModel
|
|
|
|
|
|
class Task(AbstractModel):
|
|
list_name = tortoise.fields.CharField(max_length=128, db_index=True, null=False)
|
|
name = tortoise.fields.TextField(null=False)
|
|
reset_cron = tortoise.fields.CharField(
|
|
max_length=64,
|
|
null=True,
|
|
)
|
|
check_date = tortoise.fields.DatetimeField(null=True)
|
|
previous_check_date = tortoise.fields.DatetimeField(null=True)
|
|
|
|
@classmethod
|
|
def validate_create(cls, data: dict[str, typing.Any]) -> dict | None:
|
|
out_data = {}
|
|
for field in [
|
|
"list_name",
|
|
"name",
|
|
"reset_cron",
|
|
"check_date",
|
|
"previous_check_date",
|
|
]:
|
|
out_data[field] = data.get(field)
|
|
return out_data
|
|
|
|
@classmethod
|
|
def validate_update(cls, data: dict[str, typing.Any]) -> dict | None:
|
|
out_data = {}
|
|
for field in [
|
|
"name",
|
|
"reset_cron",
|
|
"check_date",
|
|
"previous_check_date",
|
|
]:
|
|
if field in data:
|
|
out_data[field] = data[field]
|
|
return out_data
|