diff --git a/app/core/entities/accessory.py b/app/core/entities/accessory.py index 05e8ca6..5ba46bb 100644 --- a/app/core/entities/accessory.py +++ b/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 diff --git a/app/core/usecases/accessory_service.py b/app/core/usecases/accessory_service.py index cd662aa..612ebd0 100644 --- a/app/core/usecases/accessory_service.py +++ b/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, diff --git a/app/infrastructure/database/models/accessory.py b/app/infrastructure/database/models/accessory.py index 9028af0..5851137 100644 --- a/app/infrastructure/database/models/accessory.py +++ b/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)