revamped authentication system

app/routes/vendor_pages.py

@@ -25,7 +25,7 @@ from fastapi import APIRouter, Request, Depends, Path
 from fastapi.responses import HTMLResponse, RedirectResponse
 from fastapi.templating import Jinja2Templates
 
-from app.api.deps import get_current_vendor_user
+from app.api.deps import get_current_vendor_from_cookie_or_header
 from models.database.user import User
 
 router = APIRouter()
@@ -85,7 +85,7 @@ async def vendor_login_page(
 async def vendor_dashboard_page(
         request: Request,
         vendor_code: str = Path(..., description="Vendor code"),
-        current_user: User = Depends(get_current_vendor_user)
+        current_user: User = Depends(get_current_vendor_from_cookie_or_header)
 ):
     """
     Render vendor dashboard.
@@ -114,7 +114,7 @@ async def vendor_dashboard_page(
 async def vendor_products_page(
         request: Request,
         vendor_code: str = Path(..., description="Vendor code"),
-        current_user: User = Depends(get_current_vendor_user)
+        current_user: User = Depends(get_current_vendor_from_cookie_or_header)
 ):
     """
     Render products management page.
@@ -138,7 +138,7 @@ async def vendor_products_page(
 async def vendor_orders_page(
         request: Request,
         vendor_code: str = Path(..., description="Vendor code"),
-        current_user: User = Depends(get_current_vendor_user)
+        current_user: User = Depends(get_current_vendor_from_cookie_or_header)
 ):
     """
     Render orders management page.
@@ -162,7 +162,7 @@ async def vendor_orders_page(
 async def vendor_customers_page(
         request: Request,
         vendor_code: str = Path(..., description="Vendor code"),
-        current_user: User = Depends(get_current_vendor_user)
+        current_user: User = Depends(get_current_vendor_from_cookie_or_header)
 ):
     """
     Render customers management page.
@@ -186,7 +186,7 @@ async def vendor_customers_page(
 async def vendor_inventory_page(
         request: Request,
         vendor_code: str = Path(..., description="Vendor code"),
-        current_user: User = Depends(get_current_vendor_user)
+        current_user: User = Depends(get_current_vendor_from_cookie_or_header)
 ):
     """
     Render inventory management page.
@@ -210,7 +210,7 @@ async def vendor_inventory_page(
 async def vendor_marketplace_page(
         request: Request,
         vendor_code: str = Path(..., description="Vendor code"),
-        current_user: User = Depends(get_current_vendor_user)
+        current_user: User = Depends(get_current_vendor_from_cookie_or_header)
 ):
     """
     Render marketplace import page.
@@ -234,7 +234,7 @@ async def vendor_marketplace_page(
 async def vendor_team_page(
         request: Request,
         vendor_code: str = Path(..., description="Vendor code"),
-        current_user: User = Depends(get_current_vendor_user)
+        current_user: User = Depends(get_current_vendor_from_cookie_or_header)
 ):
     """
     Render team management page.
@@ -258,7 +258,7 @@ async def vendor_team_page(
 async def vendor_settings_page(
         request: Request,
         vendor_code: str = Path(..., description="Vendor code"),
-        current_user: User = Depends(get_current_vendor_user)
+        current_user: User = Depends(get_current_vendor_from_cookie_or_header)
 ):
     """
     Render vendor settings page.