Fix pydantic serializer warnings (#31) (#32)

erhosen-libs · May 24, 2024 · 8b7f6f9 · 8b7f6f9
1 parent 8011849
commit 8b7f6f9
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/demo/demo/settings.py b/demo/demo/settings.py
@@ -11,7 +11,7 @@
 """
 
 from pathlib import Path
-from typing import Dict, List
+from typing import Dict, List, Union
 
 from pydantic import Field
 from pydantic.v1.fields import Undefined
@@ -26,7 +26,7 @@
 
 class DatabaseSettings(BaseDBConfig):
     # https://docs.djangoproject.com/en/3.1/ref/settings/#databases
-    default: str = Field(
+    default: Union[str, Dict] = Field(
         default=str(f"sqlite:///{BASE_DIR}/db.sqlite3"),
         validation_alias="DATABASE_URL",
         conn_max_age=0,

diff --git a/tests/test_db_config.py b/tests/test_db_config.py
@@ -1,4 +1,4 @@
-from typing import Dict, Optional
+from typing import Dict, Optional, Union
 
 from pydantic import Field, PostgresDsn
 from pydantic_settings import BaseSettings, SettingsConfigDict
@@ -9,7 +9,7 @@
 
 def test_empty():
     class DatabaseConfig(BaseDBConfig):
-        default: Optional[PostgresDsn] = None
+        default: Union[Optional[PostgresDsn], Dict] = None
         replica: Optional[Dict] = None
 
     db_settings = DatabaseConfig()
@@ -97,7 +97,7 @@ class DatabaseConfig(BaseDBConfig):
 
 def test_dsn_and_exact_config():
     class DatabaseConfig(BaseDBConfig):
-        default: str = Field(default="postgres://user:password@hostname:5432/dbname")
+        default: Union[str, Dict] = Field(default="postgres://user:password@hostname:5432/dbname")
         replica: PostgresDB = PostgresDB()
 
     db_settings = DatabaseConfig()

diff --git a/tests/test_to_django.py b/tests/test_to_django.py
@@ -1,4 +1,4 @@
-from typing import Any, Dict, List
+from typing import Any, Dict, List, Union
 
 from deepdiff import DeepDiff
 from pydantic import Field, PostgresDsn, SecretStr, ValidationInfo, computed_field, field_validator
@@ -9,7 +9,7 @@
 
 def test_to_django_settings():
     class DatabaseConfig(BaseDBConfig):
-        default: PostgresDsn = Field(
+        default: Union[PostgresDsn, Dict] = Field(
             default="postgres://user:password@hostname:5432/dbname",
             validation_alias="DATABASE_URL",
         )