._.

app/core/entities/accessory.py

@@ -5,6 +5,7 @@ from typing import Optional
 class AccessoryEntity(BaseModel):
     id: Optional[int] = None
     name: str
+    count: Optional[int] = None
     volume: Optional[float] = None
     weight: Optional[float] = None
     period: Optional[int] = None

app/core/usecases/accessory_service.py

@@ -18,6 +18,7 @@ class AccessoriesService:
                 id=a.id,
                 name=a.name,
                 volume=a.volume,
+                count=a.count,
                 weight=a.weight,
                 period=a.period,
                 city_id=a.city_id,
@@ -32,6 +33,7 @@ class AccessoriesService:
             return AccessoryEntity(
                 id=accessory.id,
                 name=accessory.name,
+                count=accessory.count,
                 volume=accessory.volume,
                 weight=accessory.weight,
                 period=accessory.period,
@@ -42,6 +44,7 @@ class AccessoriesService:
     def create_accessory(self, entity: AccessoryEntity) -> AccessoryEntity:
         accessory_model = Accessory(
             name=entity.name,
+            count=entity.count,
             volume=entity.volume,
             weight=entity.weight,
             period=entity.period,
@@ -51,6 +54,7 @@ class AccessoriesService:
         return AccessoryEntity(
             id=created_accessory.id,
             name=created_accessory.name,
+            count=created_accessory.count,
             volume=created_accessory.volume,
             weight=created_accessory.weight,
             period=created_accessory.period,
@@ -61,6 +65,7 @@ class AccessoriesService:
         accessory_model = self.repository.get_by_id(accessory_id)
         if accessory_model:
             accessory_model.name = entity.name
+            accessory_model.count = entity.count
             accessory_model.volume = entity.volume
             accessory_model.weight = entity.weight
             accessory_model.period = entity.period
@@ -69,6 +74,7 @@ class AccessoriesService:
             return AccessoryEntity(
                 id=accessory_model.id,
                 name=accessory_model.name,
+                count=accessory_model.count,
                 volume=accessory_model.volume,
                 weight=accessory_model.weight,
                 period=accessory_model.period,

app/infrastructure/database/models/accessory.py

@@ -9,6 +9,7 @@ class Accessory(Base):
 
     id = Column(Integer, primary_key=True, autoincrement=True)
     name = Column(VARCHAR(100), nullable=False)
+    count = Column(Integer)
     volume = Column(Float)
     weight = Column(Float)
     period = Column(Integer)