Merge pull request #148 from python-ellar/pydanticV2_support

Pydantic v2 migration

Author: eadwinCode

.github/workflows/test_full.yml

@@ -37,8 +37,6 @@ jobs:
         run: pip install flit
       - name: Install Dependencies
         run: flit install --symlink
-      - name: Black
-        run: black --check ellar tests examples
       - name: Ruff Linting Check
         run: ruff check ellar tests examples
       - name: mypy

Makefile

@@ -19,12 +19,11 @@ install-full: ## Install dependencies
 	pre-commit install -f
 
 lint:fmt ## Run code linters
-	black --check ellar tests examples
 	ruff check ellar tests examples
 	mypy ellar
 
 fmt format:clean ## Run code formatters
-	black ellar tests examples
+	ruff format ellar tests examples
 	ruff check --fix ellar tests examples
 
 test:clean ## Run tests

ellar/app/context.py

@@ -4,6 +4,7 @@
 from types import TracebackType
 
 from ellar.common.constants import ELLAR_CONFIG_MODULE
+from ellar.common.logger import logger
 from ellar.common.utils.functional import SimpleLazyObject, empty
 from ellar.core import Config
 from ellar.di import EllarInjector
@@ -95,10 +96,10 @@ def _get_application_config() -> Config:
     if app_context is empty:
         config_module = os.environ.get(ELLAR_CONFIG_MODULE)
         if not config_module:
-            raise RuntimeError(
+            logger.warning(
                 "You are trying to access app config outside app context "
                 "and %s is not specified. This may cause differences in config "
-                "values when the app." % (ELLAR_CONFIG_MODULE,)
+                "values with the app" % (ELLAR_CONFIG_MODULE,)
             )
         return Config(config_module=config_module)
 

ellar/auth/handlers/schemes/base.py

@@ -103,8 +103,7 @@ def openapi_security_scheme(
     ) -> t.Dict:
         assert cls.openapi_in, "openapi_in is required"
         return {
-            cls.openapi_name
-            or cls.__name__: {
+            cls.openapi_name or cls.__name__: {
                 "type": "apiKey",
                 "description": cls.openapi_description,
                 "in": cls.openapi_in,
@@ -153,8 +152,7 @@ def openapi_security_scheme(
     ) -> t.Dict:
         assert cls.scheme, "openapi_scheme is required"
         return {
-            cls.openapi_name
-            or cls.__name__: {
+            cls.openapi_name or cls.__name__: {
                 "type": "http",
                 "description": cls.openapi_description,
                 "scheme": cls.scheme,

ellar/auth/session/options.py

@@ -1,7 +1,7 @@
 import sys
 import typing as t
 
-from pydantic import BaseModel
+from ellar.pydantic import BaseModel
 
 if sys.version_info >= (3, 8):  # pragma: no cover
     from typing import Literal

ellar/cache/schema.py

@@ -1,28 +1,25 @@
 import typing as t
 
 from ellar.cache import BaseCacheBackend
-from pydantic import BaseModel, validator
+from ellar.pydantic import BaseModel, as_pydantic_validator, field_validator
 
 
+@as_pydantic_validator("__validate_input__", schema={"type": "object"})
 class TBaseCacheBackend:
     @classmethod
-    def __get_validators__(
-        cls: t.Type["TBaseCacheBackend"],
-    ) -> t.Iterable[t.Callable[..., t.Any]]:
-        yield cls.validate
+    def __validate_input__(
+        cls: t.Type["TBaseCacheBackend"], __input: t.Any, _: t.Any
+    ) -> t.Any:
+        if isinstance(__input, BaseCacheBackend):
+            return __input
 
-    @classmethod
-    def validate(cls: t.Type["TBaseCacheBackend"], v: t.Any) -> t.Any:
-        if isinstance(v, BaseCacheBackend):
-            return v
-
-        raise ValueError(f"Expected BaseCacheBackend, received: {type(v)}")
+        raise ValueError(f"Expected BaseCacheBackend, received: {type(__input)}")
 
 
 class CacheModuleSchemaSetup(BaseModel):
     CACHES: t.Dict[str, TBaseCacheBackend] = {}
 
-    @validator("CACHES", pre=True)
+    @field_validator("CACHES", mode="before")
     def pre_cache_validate(cls, value: t.Dict) -> t.Any:
         if value and not value.get("default"):
             raise ValueError("CACHES configuration must have a 'default' key")

ellar/common/__init__.py

@@ -96,7 +96,7 @@
     trace,
     ws_route,
 )
-from .serializer import DataclassSerializer, Serializer, serialize_object
+from .serializer import Serializer, serialize_object
 from .templating import TemplateResponse, render_template, render_template_string
 
 __all__ = [
@@ -108,7 +108,6 @@
     "EllarTyper",
     "GlobalGuard",
     "Serializer",
-    "DataclassSerializer",
     "WebSocketException",
     "APIException",
     "AuthenticationFailed",

ellar/common/compatible/__init__.py

@@ -1,10 +1,8 @@
 from .cache_properties import cached_property
 from .dict import AttributeDict, AttributeDictAccessMixin
-from .emails import EmailStr
 
 __all__ = [
     "cached_property",
-    "EmailStr",
     "AttributeDictAccessMixin",
     "AttributeDict",
 ]

ellar/common/constants.py

@@ -5,13 +5,6 @@
 
 from ellar.common.types import TMessage
 from ellar.di import AnnotationToValue
-from pydantic.fields import (
-    SHAPE_LIST,
-    SHAPE_SEQUENCE,
-    SHAPE_SET,
-    SHAPE_TUPLE,
-    SHAPE_TUPLE_ELLIPSIS,
-)
 
 POST = "POST"
 PUT = "PUT"
@@ -81,21 +74,8 @@ class CONTROLLER_METADATA(metaclass=AnnotationToValue):
     INCLUDE_IN_SCHEMA: str
 
 
-sequence_shapes = {
-    SHAPE_LIST,
-    SHAPE_SET,
-    SHAPE_TUPLE,
-    SHAPE_SEQUENCE,
-    SHAPE_TUPLE_ELLIPSIS,
-}
 sequence_types = (list, set, tuple)
-sequence_shape_to_type = {
-    SHAPE_LIST: list,
-    SHAPE_SET: set,
-    SHAPE_TUPLE: tuple,
-    SHAPE_SEQUENCE: list,
-    SHAPE_TUPLE_ELLIPSIS: list,
-}
+
 primitive_types = (int, float, bool, str)
 METHODS_WITH_BODY = {"GET", "HEAD", "POST", "PUT", "DELETE", "PATCH"}
 STATUS_CODES_WITH_NO_BODY = {100, 101, 102, 103, 204, 304}

ellar/common/converters.py

@@ -1,7 +1,7 @@
 import typing as t
 from abc import ABC, abstractmethod
 
-from pydantic.typing import get_args, get_origin
+from typing_extensions import get_args, get_origin
 
 _origin_maps = {
     list: t.List,