code quality run

app/api/v1/stats.py

@@ -1,18 +1,21 @@
+import logging
+from datetime import datetime
 from typing import List, Optional
 
-from fastapi import APIRouter, Depends, HTTPException, Query, BackgroundTasks
+from fastapi import APIRouter, BackgroundTasks, Depends, HTTPException, Query
 from sqlalchemy import func
 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.stats_service import stats_service
 from app.tasks.background_tasks import process_marketplace_import
 from middleware.decorators import rate_limit
-from models.api_models import MarketplaceImportJobResponse, MarketplaceImportRequest, StatsResponse, \
-    MarketplaceStatsResponse
-from models.database_models import User, MarketplaceImportJob, Shop, Product, Stock
-from datetime import datetime
-import logging
+from models.api_models import (MarketplaceImportJobResponse,
+                               MarketplaceImportRequest,
+                               MarketplaceStatsResponse, StatsResponse)
+from models.database_models import (MarketplaceImportJob, Product, Shop, Stock,
+                                    User)
 
 router = APIRouter()
 logger = logging.getLogger(__name__)
@@ -20,7 +23,9 @@ logger = logging.getLogger(__name__)
 
 # Enhanced Statistics with Marketplace Support
 @router.get("/stats", response_model=StatsResponse)
-def get_stats(db: Session = Depends(get_db), current_user: User = Depends(get_current_user)):
+def get_stats(
+    db: Session = Depends(get_db), current_user: User = Depends(get_current_user)
+):
     """Get comprehensive statistics with marketplace data (Protected)"""
     try:
         stats_data = stats_service.get_comprehensive_stats(db=db)
@@ -32,7 +37,7 @@ def get_stats(db: Session = Depends(get_db), current_user: User = Depends(get_cu
             unique_marketplaces=stats_data["unique_marketplaces"],
             unique_shops=stats_data["unique_shops"],
             total_stock_entries=stats_data["total_stock_entries"],
-            total_inventory_quantity=stats_data["total_inventory_quantity"]
+            total_inventory_quantity=stats_data["total_inventory_quantity"],
         )
     except Exception as e:
         logger.error(f"Error getting comprehensive stats: {str(e)}")
@@ -40,7 +45,9 @@ def get_stats(db: Session = Depends(get_db), current_user: User = Depends(get_cu
 
 
 @router.get("/stats/marketplace", response_model=List[MarketplaceStatsResponse])
-def get_marketplace_stats(db: Session = Depends(get_db), current_user: User = Depends(get_current_user)):
+def get_marketplace_stats(
+    db: Session = Depends(get_db), current_user: User = Depends(get_current_user)
+):
     """Get statistics broken down by marketplace (Protected)"""
     try:
         marketplace_stats = stats_service.get_marketplace_breakdown_stats(db=db)
@@ -50,8 +57,9 @@ def get_marketplace_stats(db: Session = Depends(get_db), current_user: User = De
                 marketplace=stat["marketplace"],
                 total_products=stat["total_products"],
                 unique_shops=stat["unique_shops"],
-                unique_brands=stat["unique_brands"]
-            ) for stat in marketplace_stats
+                unique_brands=stat["unique_brands"],
+            )
+            for stat in marketplace_stats
         ]
     except Exception as e:
         logger.error(f"Error getting marketplace stats: {str(e)}")