Lint

danielkjellid · Jun 19, 2024 · 54fb474 · 54fb474
1 parent ef197ca
commit 54fb474
Show file tree

Hide file tree

Showing 5 changed files with 7 additions and 12 deletions.
diff --git a/nest/recipes/ingredients/services.py b/nest/recipes/ingredients/services.py
@@ -1,7 +1,7 @@
 import functools
 
 import structlog
-from django.db import models, transaction
+from django.db import transaction
 from django.http import HttpRequest
 from pydantic import BaseModel, Field
 
@@ -59,7 +59,7 @@ class IngredientGroupItem(BaseModel):
 
 def _get_ingredient_item_group_id(
     group_item: IngredientGroupItem,
-    recipe_groups: models.QuerySet[RecipeIngredientItemGroup],
+    recipe_groups: list[RecipeIngredientItemGroup],
 ) -> int:
     return next(
         g.id

diff --git a/nest/recipes/steps/services.py b/nest/recipes/steps/services.py
@@ -5,7 +5,7 @@
 from decimal import Decimal
 
 import structlog
-from django.db import models, transaction
+from django.db import transaction
 from pydantic import BaseModel
 
 from nest.core.exceptions import ApplicationError
@@ -89,7 +89,7 @@ def create_or_update_recipe_steps(recipe_id: int, steps: list[Step]) -> None:
 
 def _find_ingredient_item_id_for_step_item(
     item: IngredientItem,
-    recipe_ingredient_items: models.QuerySet[RecipeIngredientItem],
+    recipe_ingredient_items: list[RecipeIngredientItem],
 ) -> int:
     if item.id is not None:
         return item.id
@@ -106,9 +106,7 @@ def _find_ingredient_item_id_for_step_item(
     return item_id
 
 
-def _find_step_id_for_step_item(
-    step: Step, recipe_steps: models.QuerySet[RecipeStep]
-) -> int:
+def _find_step_id_for_step_item(step: Step, recipe_steps: list[RecipeStep]) -> int:
     if step.id is not None:
         return step.id
 

diff --git a/tests/recipes/fixtures.py b/tests/recipes/fixtures.py
@@ -16,7 +16,6 @@
 from nest.recipes.steps.enums import RecipeStepType
 from nest.recipes.steps.models import RecipeStep, RecipeStepIngredientItem
 
-
 ##########
 # Recipe #
 ##########

diff --git a/tests/recipes/test_ingredients_services.py b/tests/recipes/test_ingredients_services.py
@@ -21,8 +21,6 @@
     create_recipe_ingredient,
     delete_recipe_ingredient,
 )
-from nest.recipes.steps.enums import RecipeStepType
-from nest.recipes.steps.services import Step
 
 
 @pytest.mark.product

diff --git a/tests/recipes/test_steps_services.py b/tests/recipes/test_steps_services.py
@@ -8,10 +8,10 @@
 from nest.recipes.steps.models import RecipeStep, RecipeStepIngredientItem
 from nest.recipes.steps.services import (
     Step,
-    create_or_update_recipe_steps,
-    _find_step_id_for_step_item,
     _find_ingredient_item_id_for_step_item,
+    _find_step_id_for_step_item,
     create_or_update_recipe_step_ingredient_items,
+    create_or_update_recipe_steps,
 )