Update.

mywebdav/routers/admin_billing.py

@@ -1,5 +1,4 @@
 from fastapi import APIRouter, Depends, HTTPException
-from typing import List
 from decimal import Decimal
 from pydantic import BaseModel
 
@@ -8,21 +7,25 @@ from ..models import User
 from ..billing.models import PricingConfig, Invoice, SubscriptionPlan
 from ..billing.invoice_generator import InvoiceGenerator
 
+
 def require_superuser(current_user: User = Depends(get_current_user)):
     if not current_user.is_superuser:
         raise HTTPException(status_code=403, detail="Superuser privileges required")
     return current_user
 
+
 router = APIRouter(
     prefix="/api/admin/billing",
     tags=["admin", "billing"],
-    dependencies=[Depends(require_superuser)]
+    dependencies=[Depends(require_superuser)],
 )
 
+
 class PricingConfigUpdate(BaseModel):
     config_key: str
     config_value: float
 
+
 class PlanCreate(BaseModel):
     name: str
     display_name: str
@@ -32,6 +35,7 @@ class PlanCreate(BaseModel):
     price_monthly: float
     price_yearly: float = None
 
+
 @router.get("/pricing")
 async def get_all_pricing(current_user: User = Depends(require_superuser)):
     configs = await PricingConfig.all()
@@ -42,16 +46,17 @@ async def get_all_pricing(current_user: User = Depends(require_superuser)):
             "config_value": float(c.config_value),
             "description": c.description,
             "unit": c.unit,
-            "updated_at": c.updated_at
+            "updated_at": c.updated_at,
         }
         for c in configs
     ]
 
+
 @router.put("/pricing/{config_id}")
 async def update_pricing(
     config_id: int,
     update: PricingConfigUpdate,
-    current_user: User = Depends(require_superuser)
+    current_user: User = Depends(require_superuser),
 ):
     config = await PricingConfig.get_or_none(id=config_id)
     if not config:
@@ -63,11 +68,10 @@ async def update_pricing(
 
     return {"message": "Pricing updated successfully"}
 
+
 @router.post("/generate-invoices/{year}/{month}")
 async def generate_all_invoices(
-    year: int,
-    month: int,
-    current_user: User = Depends(require_superuser)
+    year: int, month: int, current_user: User = Depends(require_superuser)
 ):
     users = await User.filter(is_active=True).all()
     generated = []
@@ -76,35 +80,36 @@ async def generate_all_invoices(
     for user in users:
         invoice = await InvoiceGenerator.generate_monthly_invoice(user, year, month)
         if invoice:
-            generated.append({
-                "user_id": user.id,
-                "invoice_id": invoice.id,
-                "total": float(invoice.total)
-            })
+            generated.append(
+                {
+                    "user_id": user.id,
+                    "invoice_id": invoice.id,
+                    "total": float(invoice.total),
+                }
+            )
         else:
             skipped.append(user.id)
 
-    return {
-        "generated": len(generated),
-        "skipped": len(skipped),
-        "invoices": generated
-    }
+    return {"generated": len(generated), "skipped": len(skipped), "invoices": generated}
+
 
 @router.post("/plans")
 async def create_plan(
-    plan_data: PlanCreate,
-    current_user: User = Depends(require_superuser)
+    plan_data: PlanCreate, current_user: User = Depends(require_superuser)
 ):
     plan = await SubscriptionPlan.create(**plan_data.dict())
     return {"id": plan.id, "message": "Plan created successfully"}
 
+
 @router.get("/stats")
 async def get_billing_stats(current_user: User = Depends(require_superuser)):
-    from tortoise.functions import Sum, Count
+    from tortoise.functions import Sum
 
-    total_revenue = await Invoice.filter(status="paid").annotate(
-        total_sum=Sum("total")
-    ).values("total_sum")
+    total_revenue = (
+        await Invoice.filter(status="paid")
+        .annotate(total_sum=Sum("total"))
+        .values("total_sum")
+    )
 
     invoice_count = await Invoice.all().count()
     pending_invoices = await Invoice.filter(status="open").count()
@@ -112,5 +117,5 @@ async def get_billing_stats(current_user: User = Depends(require_superuser)):
     return {
         "total_revenue": float(total_revenue[0]["total_sum"] or 0),
         "total_invoices": invoice_count,
-        "pending_invoices": pending_invoices
+        "pending_invoices": pending_invoices,
     }