52 lines
1.7 KiB
Python
52 lines
1.7 KiB
Python
import typing
|
|
|
|
import tortoise
|
|
|
|
from app.models.base import AbstractModel
|
|
|
|
# https://stackoverflow.com/a/57639657
|
|
CRONTAB_REGEX = r"/(@(annually|yearly|monthly|weekly|daily|hourly|reboot))|(@every (\d+(ns|us|µs|ms|s|m|h))+)|((((\d+,)+\d+|(\d+(\/|-)\d+)|\d+|\*) ?){5,7})/"
|
|
|
|
|
|
class Task(AbstractModel):
|
|
list = tortoise.fields.CharField(max_length=32, 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 = {}
|
|
try:
|
|
for field in [
|
|
"list",
|
|
"name",
|
|
"reset_cron",
|
|
"check_date",
|
|
"previous_check_date",
|
|
]:
|
|
getattr(cls, field).validate(data.get(field))
|
|
out_data[field] = data.get(field)
|
|
except tortoise.exceptions.ValidationError:
|
|
return None
|
|
return out_data
|
|
|
|
@classmethod
|
|
def validate_update(cls, data: dict[str, typing.Any]) -> dict | None:
|
|
out_data = {}
|
|
try:
|
|
for field in [
|
|
"list",
|
|
"name",
|
|
"reset_cron",
|
|
"check_date",
|
|
"previous_check_date",
|
|
]:
|
|
if field in data:
|
|
getattr(cls, field).validate(data["field"])
|
|
out_data[field] = data["field"]
|
|
except tortoise.exceptions.ValidationError:
|
|
return None
|
|
return out_data
|