Merge pull request #36 from mongkok/pydantic-v2

Pydantic v2 support

Author: mongkok

CHANGELOG.md

@@ -1,5 +1,11 @@
 # Changelog
 
+### 0.5.0
+
+* Added Pydantic v2 support
+* Removed Pydantic v1 support
+* Removed `PydanticPanel`
+
 ### 0.4.0
 
 * Fixed middleware `url_path_for`

debug_toolbar/__init__.py

@@ -1 +1 @@
-__version__ = "0.4.0"
+__version__ = "0.5.0"

debug_toolbar/panels/__init__.py

@@ -1,7 +1,6 @@
 import typing as t
 
 from fastapi import Request, Response
-from starlette.datastructures import URL
 from starlette.middleware.base import RequestResponseEndpoint
 
 from debug_toolbar.types import ServerTiming, Stats
@@ -67,11 +66,11 @@ def content(self) -> str:
     def render(self, **context: t.Any) -> str:
         return self.toolbar.render(self.template, **context)
 
-    def url_for(self, name: str, **path_params: t.Any) -> URL:
-        return self.toolbar.request.url_for(name, **path_params)
+    def url_for(self, name: str, **path_params: t.Any) -> str:
+        return str(self.toolbar.request.url_for(name, **path_params))
 
     @property
-    def scripts(self) -> t.List[URL]:
+    def scripts(self) -> t.List[str]:
         return []
 
     async def process_request(self, request: Request) -> Response:

debug_toolbar/panels/pydantic.py

@@ -7,7 +7,7 @@
 import sqlparse
 from fastapi import Request, Response
 from fastapi.encoders import jsonable_encoder
-from pydantic.color import Color
+from pydantic_extra_types.color import Color
 from sqlparse import tokens as T
 
 from debug_toolbar.panels import Panel

debug_toolbar/panels/sql.py

@@ -2,7 +2,6 @@
 from time import perf_counter
 
 from fastapi import Request, Response
-from starlette.datastructures import URL
 
 from debug_toolbar.panels import Panel
 from debug_toolbar.types import ServerTiming, Stats
@@ -46,7 +45,7 @@ def content(self) -> str:
         return self.render(rows=rows)
 
     @property
-    def scripts(self) -> t.List[URL]:
+    def scripts(self) -> t.List[str]:
         scripts = super().scripts
         scripts.append(self.url_for("debug_toolbar.static", path="js/timer.js"))
         return scripts

debug_toolbar/panels/timer.py

@@ -1,7 +1,6 @@
 import typing as t
 
 from fastapi import Request, Response, __version__
-from starlette.datastructures import URL
 
 from debug_toolbar.panels import Panel
 from debug_toolbar.types import Stats
@@ -16,7 +15,7 @@ def nav_subtitle(self) -> str:
         return f"FastAPI {__version__}"
 
     @property
-    def scripts(self) -> t.List[URL]:
+    def scripts(self) -> t.List[str]:
         scripts = super().scripts
         scripts.append(self.url_for("debug_toolbar.static", path="js/versions.js"))
         return scripts

debug_toolbar/panels/versions.py

@@ -2,19 +2,25 @@
 
 from jinja2 import BaseLoader, ChoiceLoader, Environment, PackageLoader
 from jinja2.ext import Extension
-from pydantic import BaseSettings, Field, IPvAnyAddress, root_validator
-from pydantic.color import Color
+from pydantic import Field, IPvAnyAddress, model_validator
+from pydantic_extra_types.color import Color
+from pydantic_settings import BaseSettings, SettingsConfigDict
 
 
 class DebugToolbarSettings(BaseSettings):
+    model_config = SettingsConfigDict(
+        title="Debug Toolbar",
+        env_prefix="DT_",
+        case_sensitive=False,
+    )
+
     DEFAULT_PANELS: t.List[str] = Field(
         [
             "debug_toolbar.panels.versions.VersionsPanel",
             "debug_toolbar.panels.timer.TimerPanel",
             "debug_toolbar.panels.settings.SettingsPanel",
             "debug_toolbar.panels.request.RequestPanel",
             "debug_toolbar.panels.headers.HeadersPanel",
-            "debug_toolbar.panels.pydantic.PydanticPanel",
             "debug_toolbar.panels.routes.RoutesPanel",
             "debug_toolbar.panels.logging.LoggingPanel",
             "debug_toolbar.panels.profiling.ProfilingPanel",
@@ -129,11 +135,6 @@ class DebugToolbarSettings(BaseSettings):
         ),
     )
 
-    class Config:
-        title = "Debug Toolbar"
-        env_prefix = "DT_"
-        case_sensitive = True
-
     def __init__(self, **settings: t.Any) -> None:
         super().__init__(**settings)
         loaders = self.JINJA_LOADERS + [PackageLoader("debug_toolbar", "templates")]
@@ -144,6 +145,6 @@ def __init__(self, **settings: t.Any) -> None:
         for extension in self.JINJA_EXTENSIONS:
             self.JINJA_ENV.add_extension(extension)
 
-    @root_validator(pre=True)
-    def ci(cls, values: t.Dict[str, t.Any]) -> t.Dict[str, t.Any]:
-        return {k.upper(): v for k, v in values.items()}
+    @model_validator(mode="before")
+    def ci(cls, data: dict):
+        return {k.upper(): v for k, v in data.items()}

debug_toolbar/settings.py

@@ -1,5 +1,5 @@
 {% macro pprint_settings(settings, exclude=None) %}
-{% if settings.__config__.title %}<h4>{{ settings.__config__.title }}</h4>{% endif %}
+{% if settings.model_config.title %}<h4>{{ settings.model_config.title }}</h4>{% endif %}
 
 <table>
   <thead>