self/ruz-bot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

preload

Browse Source
This commit is contained in:
Administrator 2022-10-22 20:59:06 +03:00
parent 4c4fc87647
commit e6391e9fe6
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
helpers/models.py
View File

@ -3,7 +3,7 @@ from dataclasses import dataclass
from typing import Optional
import marshmallow_dataclass
from marshmallow import EXCLUDE, post_load
from marshmallow import EXCLUDE, pre_load
@dataclass
@ -26,14 +26,13 @@ class Lesson:
hse_id: int
USchema = marshmallow_dataclass.class_schema(User)
class UserSchema(USchema):
@post_load
def stringify_time(self, in_data, **kwargs):
if 'next_daily_notify_time' in in_data:
in_data['next_daily_notify_time'] = str(in_data['next_daily_notify_time'])
return in_data
@pre_load
def stringify_time(self, data, many, **kwargs):
if 'next_daily_notify_time' in data:
data['next_daily_notify_time'] = str(data['next_daily_notify_time'])
return data