HW 06 (#4)

* HW 06

* HW 06: update toml project

* HW 06: apply black again

Author: DwarKapex

.flake8

@@ -1,2 +1,4 @@
 [flake8]
 extend-ignore = E203,E501
+exclude =
+    tests

.github/workflows/main.yaml

@@ -91,4 +91,33 @@ jobs:
       - name: Run pytest
         run: |
           cd homework_05
+          poetry run pytest tests
+
+  hw_06:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v2
+
+      - uses: actions/setup-python@v2
+        with:
+          python-version: 3.12
+
+      - name: Run image # install poetry
+        uses: abatilo/[email protected]
+        with:
+          poetry-version: 1.8.2
+
+      - name: Install dependencies # install all dependencies
+        run: |
+          cd homework_06
+          poetry install --no-root
+
+      - name: Run static analyzis
+        run: |
+          cd homework_06
+          poetry run pre-commit run --files *
+
+      - name: Run pytest
+        run: |
+          cd homework_06
           poetry run pytest tests

.pre-commit-config.yaml

@@ -38,6 +38,7 @@ repos:
     -   id: mypy
         exclude: 'tests'
 
+
 # run local pylint in venv
 -   repo: local
     hooks:
@@ -51,7 +52,7 @@ repos:
         - --max-line-length=120
         - --ignore-imports=yes
         - -d duplicate-code
-        - -d C0111,W0621,R0913,R1705,W0201,W0613
+        - -d C0111,W0621,R0913,R1705,W0201,R0903,W0613
 
 
 -   repo: https://github.com/asottile/pyupgrade

homework_06/domain/__init__.py

@@ -0,0 +1,19 @@
+from dataclasses import dataclass, field
+from typing import List
+
+
+@dataclass
+class Product:
+    id: int
+    name: str
+    quantity: int
+    price: float
+
+
+@dataclass
+class Order:
+    id: int
+    products: List[Product] = field(default_factory=list)
+
+    def add_product(self, product: Product):
+        self.products.append(product)

homework_06/domain/exceptions.py

@@ -0,0 +1,32 @@
+from abc import ABC, abstractmethod
+from typing import List
+
+from .models import Order, Product
+
+
+class ProductRepository(ABC):
+    @abstractmethod
+    def add(self, product: Product):
+        pass
+
+    @abstractmethod
+    def get(self, product_id: int) -> Product:
+        pass
+
+    @abstractmethod
+    def list(self) -> List[Product]:
+        pass
+
+
+class OrderRepository(ABC):
+    @abstractmethod
+    def add(self, order: Order):
+        pass
+
+    @abstractmethod
+    def get(self, order_id: int) -> Order:
+        pass
+
+    @abstractmethod
+    def list(self) -> List[Order]:
+        pass

homework_06/domain/models.py

@@ -0,0 +1,27 @@
+from typing import List
+
+from .models import Order, Product
+from .repositories import OrderRepository, ProductRepository
+
+
+def generate_next_id():
+    num = 1
+    while True:
+        yield num
+        num += 1
+
+
+class WarehouseService:
+    def __init__(self, product_repo: ProductRepository, order_repo: OrderRepository):
+        self.product_repo = product_repo
+        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)
+        self.product_repo.add(product)
+        return product
+
+    def create_order(self, products: List[Product]) -> Order:
+        order = Order(id=generate_next_id(), products=products)
+        self.order_repo.add(order)
+        return order

homework_06/domain/repositories.py

@@ -0,0 +1,19 @@
+from abc import ABC, abstractmethod
+
+
+class UnitOfWork(ABC):
+    @abstractmethod
+    def __enter__(self):
+        pass
+
+    @abstractmethod
+    def __exit__(self, exc_type, exc_val, exc_tb):
+        pass
+
+    @abstractmethod
+    def commit(self):
+        pass
+
+    @abstractmethod
+    def rollback(self):
+        pass