HW 06: apply black again

Author: DwarKapex

homework_06/domain/services.py

@@ -17,9 +17,7 @@ def __init__(self, product_repo: ProductRepository, order_repo: OrderRepository)
         self.order_repo = order_repo
 
     def create_product(self, name: str, quantity: int, price: float) -> Product:
-        product = Product(
-            id=generate_next_id(), name=name, quantity=quantity, price=price
-        )
+        product = Product(id=generate_next_id(), name=name, quantity=quantity, price=price)
         self.product_repo.add(product)
         return product
 

homework_06/infrastructure/repositories.py

@@ -1,8 +1,9 @@
 from typing import List
 
+from sqlalchemy.orm import Session
+
 from domain.models import Order, Product
 from domain.repositories import OrderRepository, ProductRepository
-from sqlalchemy.orm import Session
 
 from .orm import OrderORM, ProductORM
 
@@ -12,9 +13,7 @@ def __init__(self, session: Session):
         self.session = session
 
     def add(self, product: Product):
-        product_orm = ProductORM(
-            name=product.name, quantity=product.quantity, price=product.price
-        )
+        product_orm = ProductORM(name=product.name, quantity=product.quantity, price=product.price)
         self.session.add(product_orm)
 
     def get(self, product_id: int) -> Product:
@@ -45,27 +44,18 @@ def __init__(self, session: Session):
 
     def add(self, order: Order):
         order_orm = OrderORM()
-        order_orm.products = [
-            self.session.query(ProductORM).filter_by(id=p.id).one()
-            for p in order.products
-        ]
+        order_orm.products = [self.session.query(ProductORM).filter_by(id=p.id).one() for p in order.products]
         self.session.add(order_orm)
 
     def get(self, order_id: int) -> Order:
         order_orm = self.session.query(OrderORM).filter_by(id=order_id).one()
-        products = [
-            Product(id=p.id, name=p.name, quantity=p.quantity, price=p.price)
-            for p in order_orm.products
-        ]
+        products = [Product(id=p.id, name=p.name, quantity=p.quantity, price=p.price) for p in order_orm.products]
         return Order(id=int(order_orm.id), products=list(products))
 
     def list(self) -> List[Order]:
         orders_orm = self.session.query(OrderORM).all()
         orders = []
         for order_orm in orders_orm:
-            products = [
-                Product(id=p.id, name=p.name, quantity=p.quantity, price=p.price)
-                for p in order_orm.products
-            ]
+            products = [Product(id=p.id, name=p.name, quantity=p.quantity, price=p.price) for p in order_orm.products]
             orders.append(Order(id=int(order_orm.id), products=products))
         return orders

homework_06/main.py

@@ -24,15 +24,11 @@ def main():
 
     warehouse_service = WarehouseService(product_repo, order_repo)
     with uow:
-        new_product = warehouse_service.create_product(
-            name="test1", quantity=1, price=100
-        )
+        new_product = warehouse_service.create_product(name="test1", quantity=1, price=100)
         uow.commit()
         print(f"create product: {new_product}")
 
-        another_product = warehouse_service.create_product(
-            name="test2", quantity=10, price=42
-        )
+        another_product = warehouse_service.create_product(name="test2", quantity=10, price=42)
         uow.commit()
         print(f"create product: {another_product}")
 

homework_06/tests/test_infrastructure/test_repository.py

@@ -1,13 +1,11 @@
 # flake8: noqa
-from domain.models import Order, Product
-from infrastructure.orm import OrderORM, ProductORM
-from infrastructure.repositories import (
-    SqlAlchemyOrderRepository,
-    SqlAlchemyProductRepository,
-)
 from pytest_mock import MockerFixture
 from sqlalchemy.orm import Session
 
+from domain.models import Order, Product
+from infrastructure.orm import OrderORM, ProductORM
+from infrastructure.repositories import SqlAlchemyOrderRepository, SqlAlchemyProductRepository
+
 
 class TestRepository:
     def test_sqlalchemy_product_repository_add(self, mocker: MockerFixture):
@@ -28,9 +26,7 @@ def test_sqlalchemy_product_repository_get(self, mocker: MockerFixture):
         mock_session = mocker.Mock(spec=Session)
         repo = SqlAlchemyProductRepository(mock_session)
         product_orm = ProductORM(id=1, name="Test Product", quantity=10, price=100)
-        mock_session.query.return_value.filter_by.return_value.one.return_value = (
-            product_orm
-        )
+        mock_session.query.return_value.filter_by.return_value.one.return_value = product_orm
 
         product = repo.get(1)
 
@@ -86,9 +82,7 @@ def test_sqlalchemy_order_repository_get(self, mocker: MockerFixture):
         product_orm1 = ProductORM(id=1, name="Product 1", quantity=10, price=100)
         product_orm2 = ProductORM(id=2, name="Product 2", quantity=20, price=200)
         order_orm = OrderORM(id=1, products=[product_orm1, product_orm2])
-        mock_session.query.return_value.filter_by.return_value.one.return_value = (
-            order_orm
-        )
+        mock_session.query.return_value.filter_by.return_value.one.return_value = order_orm
 
         order = repo.get(1)