From 24f17616320dca515864aa4701b737ba8b486ff7 Mon Sep 17 00:00:00 2001
From: phil <phil.dev@philome.mooo.com>
Date: Thu, 2 Jan 2025 10:46:02 +0100
Subject: [PATCH] Cosmetic

---
 src/oidc-test/main.py | 19 +++++++++----------
 1 file changed, 9 insertions(+), 10 deletions(-)

diff --git a/src/oidc-test/main.py b/src/oidc-test/main.py
index a53083a..06dac5f 100644
--- a/src/oidc-test/main.py
+++ b/src/oidc-test/main.py
@@ -1,27 +1,21 @@
 from typing import Annotated, Type, Tuple
 import string
 import random
-from json import dumps
 
 from httpx import HTTPError
 from fastapi import Depends, FastAPI, HTTPException, Request, status
 from fastapi.responses import HTMLResponse, RedirectResponse, PlainTextResponse
 from fastapi.templating import Jinja2Templates
-from fastapi.security import (
-    OAuth2PasswordBearer,
-    OpenIdConnect,
-    OAuth2AuthorizationCodeBearer,
-)
+from fastapi.security import OpenIdConnect
 from starlette.middleware.sessions import SessionMiddleware
 from authlib.integrations.starlette_client.apps import StarletteOAuth2App
 from authlib.integrations.starlette_client import OAuth, OAuthError
-from pydantic import BaseModel
+from pydantic import BaseModel, computed_field
 from pydantic_settings import (
     BaseSettings,
     YamlConfigSettingsSource,
     PydanticBaseSettingsSource,
 )
-from jose import jwt
 
 from .models import User
 
@@ -35,10 +29,12 @@ class OIDCProvider(BaseModel):
     client_secret: str = ""
     is_swagger: bool = False
 
+    @computed_field
     @property
     def provider_url(self):
         return self.url + "/.well-known/openid-configuration"
 
+    @computed_field
     @property
     def token_url(self):
         return "auth/" + self.name
@@ -95,6 +91,8 @@ if swagger_provider is not None:
         "appName": "Test Application",
         # "usePkceWithAuthorizationCodeGrant": True,
     }
+else:
+    swagger_ui_init_oauth = None
 
 app = FastAPI(
     title="OIDC auth test",
@@ -124,7 +122,6 @@ oidc_providers = dict(
     for provider in settings.oidc.providers
 )
 
-oauth2_scheme = OAuth2PasswordBearer(tokenUrl=swagger_provider.token_url)
 oidc_scheme = oidc_providers[swagger_provider.name]
 
 
@@ -179,6 +176,7 @@ async def login(request: Request, provider: str) -> RedirectResponse:
     except AttributeError:
         raise HTTPException(500, "")
     try:
+        breakpoint()
         return await provider_.authorize_redirect(request, redirect_uri)
     except HTTPError:
         raise HTTPException(status.HTTP_401_UNAUTHORIZED, "Cannot reach provider")
@@ -191,6 +189,7 @@ async def auth(request: Request, provider: str) -> RedirectResponse:
     except AttributeError:
         raise HTTPException(500, "")
     try:
+        breakpoint()
         token = await provider_.authorize_access_token(request)
     except OAuthError as error:
         raise HTTPException(status_code=401, detail=error.error)
@@ -215,7 +214,7 @@ async def home(
     return templates.TemplateResponse(
         request=request,
         context={
-            "settings": settings.dict(),
+            "settings": settings.model_dump(),
             "user": user,
             "auth_data": request.session.get("user"),
         },