27 lines
635 B
Python
27 lines
635 B
Python
from dataclasses import dataclass
|
|
from models.base import BaseModel
|
|
|
|
@dataclass
|
|
class Image(BaseModel):
|
|
type: int
|
|
extension: str
|
|
permanent: bool = False
|
|
|
|
@classmethod
|
|
def from_dict(cls, d: dict):
|
|
obj = cls(
|
|
type=d.get('type'),
|
|
extension=d.get('extension'),
|
|
permanent=d.get('permanent', False)
|
|
)
|
|
return cls._apply_base_fields(obj, d)
|
|
|
|
def to_dict(self):
|
|
base = super().to_dict()
|
|
base.update({
|
|
'type': self.type,
|
|
'permanent': self.permanent,
|
|
'extension': self.extension
|
|
})
|
|
return base
|