round 2

models/image.py

@@ -1,30 +1,26 @@
 from dataclasses import dataclass
-import uuid
+from models.base import BaseModel
 
 @dataclass
-class Image:
+class Image(BaseModel):
     type: int
     extension: str
     permanent: bool = False
-    id: str | None = None
-
-    def __post_init__(self):
-        if self.id is None:
-            self.id = str(uuid.uuid4())
 
     @classmethod
     def from_dict(cls, d: dict):
-        return cls(
-            id=d.get('id'),
+        obj = cls(
             type=d.get('type'),
-            permanent=d.get('permanent', False),
-            extension=d.get('extension')
+            extension=d.get('extension'),
+            permanent=d.get('permanent', False)
         )
+        return cls._apply_base_fields(obj, d)
 
     def to_dict(self):
-        return {
-            'id': self.id,
+        base = super().to_dict()
+        base.update({
             'type': self.type,
             'permanent': self.permanent,
             'extension': self.extension
-        }
+        })
+        return base