Fix definition of SqlalchemyDsn for pydantic 2

DIRACGrid · Jun 8, 2024 · 3347de0 · 3347de0
1 parent 4bfc9c2
commit 3347de0
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/diracx-core/src/diracx/core/settings.py b/diracx-core/src/diracx/core/settings.py
@@ -17,9 +17,7 @@
 
 T = TypeVar("T")
 
-
-class SqlalchemyDsn(AnyUrl):
-    allowed_schemes = {"sqlite+aiosqlite", "mysql+aiomysql"}
+SqlalchemyDsn = Annotated[AnyUrl, UrlConstraints(schemes={"sqlite+aiosqlite", "mysql+aiomysql"})]
 
 
 class _TokenSigningKey(SecretStr):