fastapi-voyager 0.4.2__tar.gz → 0.4.3__tar.gz

{fastapi_voyager-0.4.2 → fastapi_voyager-0.4.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fastapi-voyager
-Version: 0.4.2
+Version: 0.4.3
 Summary: Visualize FastAPI application's routing tree and dependencies
 Project-URL: Homepage, https://github.com/allmonday/fastapi-voyager
 Project-URL: Source, https://github.com/allmonday/fastapi-voyager

{fastapi_voyager-0.4.2 → fastapi_voyager-0.4.3}/src/fastapi_voyager/graph.py

@@ -112,7 +112,6 @@ class Analytics:
             # add response_models and create links from route -> response_model
             for schema in get_core_types(route.response_model):
                 if schema and issubclass(schema, BaseModel):
-                    update_forward_refs(schema)
                     target_name = full_class_name(schema)
                     self.links.append(Link(
                         source=route_id,
@@ -182,6 +181,7 @@ class Analytics:
         3. recursively run walk_schema
         """
         
+        update_forward_refs(schema)
         self.add_to_node_set(schema)
 
         # handle schema inside ensure_subset(schema)

{fastapi_voyager-0.4.2 → fastapi_voyager-0.4.3}/src/fastapi_voyager/type_helper.py

@@ -4,6 +4,7 @@ from pydantic import BaseModel
 from typing import get_origin, get_args, Union, Annotated, Any, Type
 from fastapi_voyager.type import FieldInfo
 from types import UnionType
+import pydantic_resolve.constant as const
 
 # Python <3.12 compatibility: TypeAliasType exists only from 3.12 (PEP 695)
 try:  # pragma: no cover - import guard
@@ -149,8 +150,9 @@ def get_type_name(anno):
 
     return name_of(anno)
 
+
 def is_inheritance_of_pydantic_base(cls):
-        return issubclass(cls, BaseModel) and cls is not BaseModel
+    return issubclass(cls, BaseModel) and cls is not BaseModel
 
 
 def get_bases_fields(schemas: list[type[BaseModel]]) -> set[str]:
@@ -240,15 +242,30 @@ def safe_issubclass(kls, classinfo):
 
 
 def update_forward_refs(kls):
-    def update_pydantic_forward_refs(kls: Type[BaseModel]):
+    def update_pydantic_forward_refs(kls2: Type[BaseModel]):
         """
         recursively update refs.
         """
-        kls.model_rebuild()
-        values = kls.model_fields.values()
+
+        kls2.model_rebuild()
+        setattr(kls2, const.PYDANTIC_FORWARD_REF_UPDATED, True)
+
+        values = kls2.model_fields.values()
         for field in values:
             update_forward_refs(field.annotation)
 
-    for shelled_types in get_core_types(kls):
-        if safe_issubclass(shelled_types, BaseModel):
-            update_pydantic_forward_refs(shelled_types)
+    if safe_issubclass(kls, BaseModel):
+        kls.model_rebuild()
+        setattr(kls, const.PYDANTIC_FORWARD_REF_UPDATED, True)
+        
+    for shelled_type in get_core_types(kls):
+        if getattr(shelled_type, const.PYDANTIC_FORWARD_REF_UPDATED, False):
+            continue
+        if safe_issubclass(shelled_type, BaseModel):
+            update_pydantic_forward_refs(shelled_type)
+
+if __name__ == "__main__":
+    from tests.demo_anno import PageSprint, PageOverall
+
+    update_forward_refs(PageOverall)
+    update_forward_refs(PageSprint)

{fastapi_voyager-0.4.2 → fastapi_voyager-0.4.3}/src/fastapi_voyager/version.py

@@ -1,2 +1,2 @@
 __all__ = ["__version__"]
-__version__ = "0.4.2"
+__version__ = "0.4.3"

{fastapi_voyager-0.4.2 → fastapi_voyager-0.4.3}/tests/demo_anno.py

@@ -12,6 +12,11 @@ app = FastAPI(title="Demo API", description="A demo FastAPI application for rout
 def get_sprint():
     return []
 
+class Tree(BaseModel):
+    id: int
+    name: str
+    children: list[Tree] = []
+
 class PageMember(serv.Member):
     fullname: str = ''
     def post_fullname(self):
@@ -51,6 +56,8 @@ class PageStory(BaseModel):
     owner: Optional[PageMember] = None
     union_tasks: list[TaskUnion] = []
 
+    tree: Optional[Tree] = None
+
 @app.get("/page_overall", tags=['for-page'], response_model=PageOverall)
 async def get_page_info():
     page_overall = PageOverall(sprints=[]) # focus on schema only