code quality run

app/api/v1/stock.py

@@ -1,16 +1,19 @@
+import logging
 from typing import List, Optional
 
-from fastapi import APIRouter, Depends, HTTPException, Query, BackgroundTasks
+from fastapi import APIRouter, BackgroundTasks, Depends, HTTPException, Query
 from sqlalchemy.orm import Session
-from app.core.database import get_db
+
 from app.api.deps import get_current_user
+from app.core.database import get_db
+from app.services.stock_service import stock_service
 from app.tasks.background_tasks import process_marketplace_import
 from middleware.decorators import rate_limit
-from models.api_models import (MarketplaceImportJobResponse, MarketplaceImportRequest, StockResponse,
-                               StockSummaryResponse, StockCreate, StockAdd, StockUpdate)
-from models.database_models import User, MarketplaceImportJob, Shop
-from app.services.stock_service import stock_service
-import logging
+from models.api_models import (MarketplaceImportJobResponse,
+                               MarketplaceImportRequest, StockAdd, StockCreate,
+                               StockResponse, StockSummaryResponse,
+                               StockUpdate)
+from models.database_models import MarketplaceImportJob, Shop, User
 
 router = APIRouter()
 logger = logging.getLogger(__name__)
@@ -18,11 +21,12 @@ logger = logging.getLogger(__name__)
 
 # Stock Management Routes (Protected)
 
+
 @router.post("/stock", response_model=StockResponse)
 def set_stock(
-        stock: StockCreate,
-        db: Session = Depends(get_db),
-        current_user: User = Depends(get_current_user)
+    stock: StockCreate,
+    db: Session = Depends(get_db),
+    current_user: User = Depends(get_current_user),
 ):
     """Set exact stock quantity for a GTIN at a specific location (replaces existing quantity)"""
     try:
@@ -37,9 +41,9 @@ def set_stock(
 
 @router.post("/stock/add", response_model=StockResponse)
 def add_stock(
-        stock: StockAdd,
-        db: Session = Depends(get_db),
-        current_user: User = Depends(get_current_user)
+    stock: StockAdd,
+    db: Session = Depends(get_db),
+    current_user: User = Depends(get_current_user),
 ):
     """Add quantity to existing stock for a GTIN at a specific location (adds to existing quantity)"""
     try:
@@ -54,9 +58,9 @@ def add_stock(
 
 @router.post("/stock/remove", response_model=StockResponse)
 def remove_stock(
-        stock: StockAdd,
-        db: Session = Depends(get_db),
-        current_user: User = Depends(get_current_user)
+    stock: StockAdd,
+    db: Session = Depends(get_db),
+    current_user: User = Depends(get_current_user),
 ):
     """Remove quantity from existing stock for a GTIN at a specific location"""
     try:
@@ -71,9 +75,9 @@ def remove_stock(
 
 @router.get("/stock/{gtin}", response_model=StockSummaryResponse)
 def get_stock_by_gtin(
-        gtin: str,
-        db: Session = Depends(get_db),
-        current_user: User = Depends(get_current_user)
+    gtin: str,
+    db: Session = Depends(get_db),
+    current_user: User = Depends(get_current_user),
 ):
     """Get all stock locations and total quantity for a specific GTIN"""
     try:
@@ -88,9 +92,9 @@ def get_stock_by_gtin(
 
 @router.get("/stock/{gtin}/total")
 def get_total_stock(
-        gtin: str,
-        db: Session = Depends(get_db),
-        current_user: User = Depends(get_current_user)
+    gtin: str,
+    db: Session = Depends(get_db),
+    current_user: User = Depends(get_current_user),
 ):
     """Get total quantity in stock for a specific GTIN"""
     try:
@@ -105,21 +109,17 @@ def get_total_stock(
 
 @router.get("/stock", response_model=List[StockResponse])
 def get_all_stock(
-        skip: int = Query(0, ge=0),
-        limit: int = Query(100, ge=1, le=1000),
-        location: Optional[str] = Query(None, description="Filter by location"),
-        gtin: Optional[str] = Query(None, description="Filter by GTIN"),
-        db: Session = Depends(get_db),
-        current_user: User = Depends(get_current_user)
+    skip: int = Query(0, ge=0),
+    limit: int = Query(100, ge=1, le=1000),
+    location: Optional[str] = Query(None, description="Filter by location"),
+    gtin: Optional[str] = Query(None, description="Filter by GTIN"),
+    db: Session = Depends(get_db),
+    current_user: User = Depends(get_current_user),
 ):
     """Get all stock entries with optional filtering"""
     try:
         result = stock_service.get_all_stock(
-            db=db,
-            skip=skip,
-            limit=limit,
-            location=location,
-            gtin=gtin
+            db=db, skip=skip, limit=limit, location=location, gtin=gtin
         )
         return result
     except Exception as e:
@@ -129,10 +129,10 @@ def get_all_stock(
 
 @router.put("/stock/{stock_id}", response_model=StockResponse)
 def update_stock(
-        stock_id: int,
-        stock_update: StockUpdate,
-        db: Session = Depends(get_db),
-        current_user: User = Depends(get_current_user)
+    stock_id: int,
+    stock_update: StockUpdate,
+    db: Session = Depends(get_db),
+    current_user: User = Depends(get_current_user),
 ):
     """Update stock quantity for a specific stock entry"""
     try:
@@ -147,9 +147,9 @@ def update_stock(
 
 @router.delete("/stock/{stock_id}")
 def delete_stock(
-        stock_id: int,
-        db: Session = Depends(get_db),
-        current_user: User = Depends(get_current_user)
+    stock_id: int,
+    db: Session = Depends(get_db),
+    current_user: User = Depends(get_current_user),
 ):
     """Delete a stock entry"""
     try: