Segregate the routers into two types

Author: ArtyomVancyan

examples/demonstration/main.py

@@ -12,7 +12,8 @@
 from fastapi_oauth2.middleware import User
 from fastapi_oauth2.router import router as oauth2_router
 from models import User as UserModel
-from router import router as app_router
+from router_api import router_api
+from router_ssr import router_ssr
 
 Base.metadata.create_all(bind=engine)
 
@@ -36,7 +37,8 @@ async def on_auth(auth: Auth, user: User):
 
 
 app = FastAPI()
-app.include_router(app_router)
+app.include_router(router_api)
+app.include_router(router_ssr)
 app.include_router(oauth2_router)
 app.mount("/static", StaticFiles(directory="static"), name="static")
 app.add_middleware(OAuth2Middleware, config=oauth2_config, callback=on_auth)

examples/demonstration/router.py

@@ -0,0 +1,32 @@
+from fastapi import APIRouter
+from fastapi import Request
+from fastapi.templating import Jinja2Templates
+from starlette.responses import RedirectResponse
+
+from fastapi_oauth2.security import OAuth2
+
+oauth2 = OAuth2()
+router_api = APIRouter()
+templates = Jinja2Templates(directory="templates")
+
+
+@router_api.get("/auth")
+def sim_auth(request: Request):
+    access_token = request.auth.jwt_create({
+        "id": 1,
+        "identity": "demo:1",
+        "image": None,
+        "display_name": "John Doe",
+        "email": "[email protected]",
+        "username": "JohnDoe",
+        "exp": 3689609839,
+    })
+    response = RedirectResponse("/")
+    response.set_cookie(
+        "Authorization",
+        value=f"Bearer {access_token}",
+        max_age=request.auth.expires,
+        expires=request.auth.expires,
+        httponly=request.auth.http,
+    )
+    return response

examples/demonstration/router_api.py

@@ -0,0 +1,35 @@
+import json
+
+from fastapi import APIRouter
+from fastapi import Depends
+from fastapi import Request
+from fastapi.responses import HTMLResponse
+from fastapi.templating import Jinja2Templates
+from sqlalchemy.orm import Session
+
+from database import get_db
+from fastapi_oauth2.security import OAuth2
+from models import User
+
+oauth2 = OAuth2()
+router_ssr = APIRouter()
+templates = Jinja2Templates(directory="templates")
+
+
+@router_ssr.get("/", response_class=HTMLResponse)
+async def root(request: Request):
+    return templates.TemplateResponse("index.html", {
+        "json": json,
+        "request": request,
+    })
+
+
+@router_ssr.get("/users", response_class=HTMLResponse)
+async def users(request: Request, db: Session = Depends(get_db), _: str = Depends(oauth2)):
+    return templates.TemplateResponse("users.html", {
+        "json": json,
+        "request": request,
+        "users": [
+            dict([(k, v) for k, v in user.__dict__.items() if not k.startswith("_")]) for user in db.query(User).all()
+        ],
+    })