ruz-bot/helpers/models.py
2022-10-22 20:45:32 +03:00

31 lines
590 B
Python

import datetime
from dataclasses import dataclass
from typing import Optional
import marshmallow_dataclass
from marshmallow import EXCLUDE
@dataclass
class User:
chat_id: int
name: Optional[str] = None
group: Optional[str] = None
hse_id: Optional[int] = None
state: str = "new"
notify_minutes: Optional[int] = 10
daily_notify_time: Optional[str] = None
next_daily_notify_time: Optional[datetime.datetime] = None
class Meta:
unknown = EXCLUDE
@dataclass
class Lesson:
hse_id: int
UserSchema = marshmallow_dataclass.class_schema(User)