added schema storage

Author: CosmoV

fastapi_jsonapi/api.py

@@ -75,7 +75,6 @@ def __init__(
         pagination_default_offset: Optional[int] = None,
         pagination_default_limit: Optional[int] = None,
         methods: Iterable[str] = (),
-        max_cache_size: int = 0,
         ending_slash: bool = True,
     ) -> None:
         """
@@ -127,7 +126,7 @@ def __init__(
         self.pagination_default_number: Optional[int] = pagination_default_number
         self.pagination_default_offset: Optional[int] = pagination_default_offset
         self.pagination_default_limit: Optional[int] = pagination_default_limit
-        self.schema_builder = SchemaBuilder(resource_type=resource_type, max_cache_size=max_cache_size)
+        self.schema_builder = SchemaBuilder(resource_type=resource_type)
 
         dto = self.schema_builder.create_schemas(
             schema=schema,

fastapi_jsonapi/atomic/atomic_handler.py

@@ -8,7 +8,7 @@
 
 from fastapi import HTTPException, status
 from fastapi.requests import Request
-from pydantic import BaseModel, ValidationError
+from pydantic import ValidationError
 
 from fastapi_jsonapi import RoutersJSONAPI
 from fastapi_jsonapi.atomic.prepared_atomic_operation import LocalIdsType, OperationBase
@@ -150,14 +150,14 @@ async def handle(self) -> Union[AtomicResponseDict, AtomicResultResponse, None]:
                 results.append({})
                 continue
             only_empty_responses = False
+
+            data = response["data"]
             results.append(
-                {
-                    "data": response.data.model_dump() if isinstance(response.data, BaseModel) else response.data,
-                },
+                {"data": data},
             )
 
-            if operation.data.lid and response.data:
-                self.local_ids_cache[operation.data.type][operation.data.lid] = response.data.id
+            if operation.data.lid and data:
+                self.local_ids_cache[operation.data.type][operation.data.lid] = data["id"]
 
             # reset context var
             current_atomic_operation.reset(ctx_var_token)

fastapi_jsonapi/schema.py

@@ -6,9 +6,10 @@
 
 from __future__ import annotations
 
+from dataclasses import dataclass
 from inspect import isclass
 from types import GenericAlias
-from typing import TYPE_CHECKING, Optional, Sequence, Type, Union, get_args
+from typing import TYPE_CHECKING, Any, Callable, Optional, Sequence, Type, Union, get_args
 
 from fastapi import FastAPI
 from pydantic import BaseModel, ConfigDict, Field
@@ -20,6 +21,7 @@
 from pydantic.fields import FieldInfo
 
 from fastapi_jsonapi.common import search_relationship_info
+from fastapi_jsonapi.types_metadata import RelationshipInfo
 
 if TYPE_CHECKING:
     from fastapi_jsonapi.data_typing import TypeSchema
@@ -137,6 +139,50 @@ class JSONAPISchemaIntrospectionError(Exception):
     pass
 
 
+# todo: when 3.9 support is dropped, return back `slots=True to JSONAPIObjectSchemas dataclass`
+
+
+@dataclass(frozen=True)
+class JSONAPIObjectSchemas:
+    attributes_schema: Type[BaseModel]
+    relationships_schema: Type[BaseModel]
+    object_jsonapi_schema: Type[JSONAPIObjectSchema]
+    can_be_included_schemas: dict[str, Type[JSONAPIObjectSchema]]
+
+    @property
+    def included_schemas_list(self) -> list[Type[JSONAPIObjectSchema]]:
+        return list(self.can_be_included_schemas.values())
+
+
+@dataclass(frozen=True)
+class BuiltSchemasDTO:
+    schema_in_post: Type[BaseJSONAPIDataInSchema]
+    schema_in_post_data: Type[BaseJSONAPIItemInSchema]
+    schema_in_patch: Type[BaseJSONAPIDataInSchema]
+    schema_in_patch_data: Type[BaseJSONAPIItemInSchema]
+    detail_response_schema: Type[JSONAPIResultDetailSchema]
+    list_response_schema: Type[JSONAPIResultListSchema]
+
+
+FieldValidators = dict[str, Callable]
+
+
+@dataclass(frozen=True)
+class SchemasInfoDTO:
+    # id field
+    resource_id_field: tuple[Type, FieldInfo, Callable, FieldValidators]
+    # pre-built attributes
+    attributes_schema: Type[BaseModel]
+    # relationships
+    relationships_schema: Type[BaseModel]
+    # has any required relationship
+    has_required_relationship: bool
+    # anything that can be included
+    included_schemas: list[tuple[str, BaseModel, str]]
+
+    relationships_info: dict[str, tuple[RelationshipInfo, Any]]
+
+
 def get_model_field(schema: Type["TypeSchema"], field: str) -> str:
     """
     Get the model field of a schema field.