fastapi-voyager 0.11.9__tar.gz → 0.11.10__tar.gz

{fastapi_voyager-0.11.9 → fastapi_voyager-0.11.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fastapi-voyager
-Version: 0.11.9
+Version: 0.11.10
 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
@@ -277,6 +277,8 @@ or you can open router_viz.dot with vscode extension `graphviz interactive previ
     - [x] fix swagger link in another way
 - 0.11.9
     - [x] replace issubclass with safe_issubclass to prevent exception.
+- 0.11.10
+    - [x] fix bug during updating forward refs
 
 #### 0.12
 - [ ] add tests

{fastapi_voyager-0.11.9 → fastapi_voyager-0.11.10}/README.md

@@ -248,6 +248,8 @@ or you can open router_viz.dot with vscode extension `graphviz interactive previ
     - [x] fix swagger link in another way
 - 0.11.9
     - [x] replace issubclass with safe_issubclass to prevent exception.
+- 0.11.10
+    - [x] fix bug during updating forward refs
 
 #### 0.12
 - [ ] add tests

{fastapi_voyager-0.11.9 → fastapi_voyager-0.11.10}/src/fastapi_voyager/type_helper.py

@@ -232,27 +232,24 @@ def safe_issubclass(kls, classinfo):
     try:
         return issubclass(kls, classinfo)
     except TypeError:
-        print(kls.__name__, 'is not a class')
+        # may raise error for corner case such as ForwardRef
+        print(str(kls), 'is not a target class')  
         return False
 
 
 def update_forward_refs(kls):
     # TODO: refactor
-    def update_pydantic_forward_refs(kls2: Type[BaseModel]):
+    def update_pydantic_forward_refs(pydantic_kls: Type[BaseModel]):
         """
         recursively update refs.
         """
 
-        kls2.model_rebuild()
-        setattr(kls2, const.PYDANTIC_FORWARD_REF_UPDATED, True)
+        pydantic_kls.model_rebuild()
+        setattr(pydantic_kls, const.PYDANTIC_FORWARD_REF_UPDATED, True)
 
-        values = kls2.model_fields.values()
+        values = pydantic_kls.model_fields.values()
         for field in values:
             update_forward_refs(field.annotation)
-
-    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):

{fastapi_voyager-0.11.9 → fastapi_voyager-0.11.10}/src/fastapi_voyager/version.py

@@ -1,2 +1,2 @@
 __all__ = ["__version__"]
-__version__ = "0.11.9"
+__version__ = "0.11.10"

{fastapi_voyager-0.11.9 → fastapi_voyager-0.11.10}/tests/demo.py

@@ -4,6 +4,7 @@ from typing import Optional, Generic, TypeVar
 from pydantic_resolve import ensure_subset, Resolver
 from tests.service.schema import Story, Task, A
 import tests.service.schema as serv
+from dataclasses import dataclass
 
 app = FastAPI(title="Demo API", description="A demo FastAPI application for router visualization")
 
@@ -15,6 +16,11 @@ class PageMember(serv.Member):
     fullname: str = ''
     def post_fullname(self):
         return self.first_name + ' ' + self.last_name
+    sh: 'Something'  # forward reference
+
+@dataclass
+class Something:
+    id: int
 
 class TaskA(Task):
     task_type: str = 'A'