fastapi-voyager 0.6.2__py3-none-any.whl → 0.7.2__py3-none-any.whl

fastapi_voyager/cli.py

@@ -175,6 +175,12 @@ Examples:
         default="127.0.0.1",
         help="Host/IP for the preview server when --server is used (default: 127.0.0.1). Use 0.0.0.0 to listen on all interfaces."
     )
+    parser.add_argument(
+        "--module_prefix",
+        type=str,
+        default=None,
+        help="Prefix routes with module name when rendering brief view (only valid with --server)"
+    )
     
     parser.add_argument(
         "--version", "-v",
@@ -233,6 +239,9 @@ Examples:
             if not any(mc.startswith(key + ":") for mc in existing):
                 args.module_color = (args.module_color or []) + [d]
 
+    if args.module_prefix and not args.server:
+        parser.error("--module_prefix can only be used together with --server")
+
     # Validate required target if not demo
     if not args.demo and not (args.module_name or args.module):
         parser.error("You must provide a module file, -m module name, or use --demo")
@@ -272,7 +281,11 @@ Examples:
             except ImportError:
                 print("Error: uvicorn is required to run the server. Install via 'pip install uvicorn' or 'uv add uvicorn'.")
                 sys.exit(1)
-            app_server = viz_server.create_app_with_fastapi(app, module_color=module_color)
+            app_server = viz_server.create_app_with_fastapi(
+                app,
+                module_color=module_color,
+                module_prefix=args.module_prefix,
+            )
             print(f"Starting preview server at http://{args.host}:{args.port} ... (Ctrl+C to stop)")
             uvicorn.run(app_server, host=args.host, port=args.port)
         else:

fastapi_voyager/filter.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
+from collections import deque
 from typing import Tuple
-from fastapi_voyager.type import Tag, Route, SchemaNode, Link
+from fastapi_voyager.type import Tag, Route, SchemaNode, Link, PK
 
 
 def filter_graph(
@@ -103,3 +104,92 @@ def filter_graph(
     _routes = [r for r in routes if r.id in included_ids]
 
     return _tags, _routes, _nodes, _links
+
+
+def filter_subgraph(
+    *,
+    tags: list[Tag],
+    routes: list[Route],
+    links: list[Link],
+    nodes: list[SchemaNode],
+    module_prefix: str
+) -> Tuple[list[Tag], list[Route], list[SchemaNode], list[Link]]:
+    """Collapse schema graph so routes link directly to nodes whose module matches ``module_prefix``.
+
+    The routine keeps tag→route links untouched, prunes schema nodes whose module does not start
+    with ``module_prefix``, and merges the remaining schema relationships so each route connects
+    directly to the surviving schema nodes. Traversal stops once a qualifying node is reached and
+    guards against cycles in the schema graph.
+    """
+
+    if not module_prefix:
+        # empty prefix keeps existing graph structure, so simply reuse incoming data
+        return tags, routes, nodes, [lk for lk in links if lk.type in ("tag_route", "route_to_schema")]
+
+    route_links = [lk for lk in links if lk.type == "route_to_schema"]
+    schema_links = [lk for lk in links if lk.type in {"schema", "parent", "subset"}]
+    tag_route_links = [lk for lk in links if lk.type == "tag_route"]
+
+    node_lookup: dict[str, SchemaNode] = {node.id: node for node in nodes}
+
+    filtered_nodes = [node for node in nodes if node_lookup[node.id].module.startswith(module_prefix)]
+    filtered_node_ids = {node.id for node in filtered_nodes}
+
+    adjacency: dict[str, list[str]] = {}
+    for link in schema_links:
+        if link.source_origin not in node_lookup or link.target_origin not in node_lookup:
+            continue
+        adjacency.setdefault(link.source_origin, [])
+        if link.target_origin not in adjacency[link.source_origin]:
+            adjacency[link.source_origin].append(link.target_origin)
+
+    merged_links: list[Link] = []
+    seen_pairs: set[tuple[str, str]] = set()
+
+    for link in route_links:
+        route_id = link.source_origin
+        start_node_id = link.target_origin
+        if route_id is None or start_node_id is None:
+            continue
+        if start_node_id not in node_lookup:
+            continue
+
+        visited: set[str] = set()
+        queue: deque[str] = deque([start_node_id])
+
+        while queue:
+            current = queue.popleft()
+            if current in visited:
+                continue
+            visited.add(current)
+
+            if current in filtered_node_ids:
+                key = (route_id, current)
+                if key not in seen_pairs:
+                    seen_pairs.add(key)
+                    merged_links.append(
+                        Link(
+                            source=link.source,
+                            source_origin=route_id,
+                            target=f"{current}::{PK}",
+                            target_origin=current,
+                            type="route_to_schema",
+                        )
+                    )
+                # stop traversing past a qualifying node
+                continue
+
+            for next_node in adjacency.get(current, () ):
+                if next_node not in visited:
+                    queue.append(next_node)
+
+    module_prefix_links = [
+        lk
+        for lk in links
+        if (lk.source_origin or "").startswith(module_prefix)
+        and (lk.target_origin or "").startswith(module_prefix)
+    ]
+
+    filtered_links = tag_route_links + merged_links + module_prefix_links
+
+    return tags, routes, filtered_nodes, filtered_links

fastapi_voyager/server.py

@@ -32,10 +32,12 @@ class Payload(BaseModel):
 	route_name: Optional[str] = None
 	show_fields: str = 'object'
 	show_meta: bool = False
+	brief: bool = False
 
 def create_route(
 	target_app: FastAPI,
 	module_color: dict[str, str] | None = None,
+	module_prefix: Optional[str] = None,
 ):
 	router = APIRouter(tags=['fastapi-voyager'])
 
@@ -73,7 +75,10 @@ def create_route(
 			load_meta=False,
 		)
 		voyager.analysis(target_app)
-		return voyager.render_dot()
+		if payload.brief:
+			return voyager.render_brief_dot(module_prefix=module_prefix)
+		else:
+			return voyager.render_dot()
 
 	@router.post("/dot-core-data", response_model=CoreData)
 	def get_filtered_dot_core_data(payload: Payload) -> str:
@@ -117,8 +122,9 @@ def create_app_with_fastapi(
 	target_app: FastAPI,
 	module_color: dict[str, str] | None = None,
 	gzip_minimum_size: int | None = 500,
+	module_prefix: Optional[str] = None,
 ) -> FastAPI:
-	router = create_route(target_app, module_color=module_color)
+	router = create_route(target_app, module_color=module_color, module_prefix=module_prefix)
 
 	app = FastAPI(title="fastapi-voyager demo server")
 	if gzip_minimum_size is not None and gzip_minimum_size >= 0:

fastapi_voyager/version.py

@@ -1,2 +1,2 @@
 __all__ = ["__version__"]
-__version__ = "0.6.2"
+__version__ = "0.7.2"

fastapi_voyager/voyager.py

@@ -12,7 +12,7 @@ from fastapi_voyager.type_helper import (
 )
 from pydantic import BaseModel
 from fastapi_voyager.type import Route, SchemaNode, Link, Tag, LinkType, FieldType, PK, CoreData
-from fastapi_voyager.filter import filter_graph
+from fastapi_voyager.filter import filter_graph, filter_subgraph
 from fastapi_voyager.render import Renderer
 import pydantic_resolve.constant as const
 
@@ -271,4 +271,16 @@ class Voyager:
             node_set=self.node_set,
         )
         renderer = Renderer(show_fields=self.show_fields, module_color=self.module_color, schema=self.schema)
+        return renderer.render_dot(_tags, _routes, _nodes, _links)
+    
+    def render_brief_dot(self, module_prefix: str | None = None):
+        print(module_prefix)
+        _tags, _routes, _nodes, _links = filter_subgraph(
+            module_prefix=module_prefix,
+            tags=self.tags,
+            routes=self.routes,
+            nodes=self.nodes,
+            links=self.links,
+        )
+        renderer = Renderer(show_fields=self.show_fields, module_color=self.module_color, schema=None)
         return renderer.render_dot(_tags, _routes, _nodes, _links)

fastapi_voyager/web/index.html

@@ -109,6 +109,14 @@
             </div>
           </div>
 
+          <div class="col-auto">
+            <q-checkbox
+              v-model="state.brief"
+              label="Brief"
+              dense
+            />
+          </div>
+          
           <div class="col-auto">
             <q-btn-dropdown
               class="q-ml-md"

fastapi_voyager/web/vue-main.js

@@ -22,6 +22,7 @@ const app = createApp({
         { label: "Object fields", value: "object" },
         { label: "All fields", value: "all" },
       ],
+      brief: false,
       generating: false,
       rawTags: [], // [{ name, routes: [{ id, name }] }]
       rawSchemas: [], // [{ name, fullname }]
@@ -132,6 +133,7 @@ const app = createApp({
           schema_name: state.schemaFullname || null,
           route_name: state.routeId || null,
           show_fields: state.showFields,
+          brief: state.brief,
         };
 
         const res = await fetch("dot", {
@@ -179,6 +181,7 @@ const app = createApp({
           schema_name: state.schemaFullname || null,
           route_name: state.routeId || null,
           show_fields: state.showFields,
+          brief: state.brief,
         };
         const res = await fetch("dot-core-data", {
           method: "POST",
@@ -238,6 +241,7 @@ const app = createApp({
       state.routeId = "";
       state.schemaFullname = null;
       state.showFields = "object";
+      state.brief = false;
       await loadInitial();
     }
 

{fastapi_voyager-0.6.2.dist-info → fastapi_voyager-0.7.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fastapi-voyager
-Version: 0.6.2
+Version: 0.7.2
 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
@@ -146,7 +146,7 @@ or you can open router_viz.dot with vscode extension `graphviz interactive previ
 
 ## Plan before v1.0
 
-features 
+features:
 - [x] group schemas by module hierarchy
 - [x] module-based coloring via Analytics(module_color={...})
 - [x] view in web browser
@@ -162,31 +162,31 @@ features
 - [x] alt+click to show field details
 - [x] display source code of routes (including response_model)
 - [x] handle excluded field 
-- [ ] add tooltips
-- [ ] route
-    - [ ] group routes by module hierarchy
+- [x] add tooltips
+- [x] route
+    - [x] group routes by module hierarchy
     - [ ] add response_model in route
-- [ ] support dataclass
+- [ ] support dataclass (pending)
 - [ ] click field to highlight links
 - [ ] user can generate nodes/edges manually and connect to generated ones
     - [ ] add owner
 - [ ] ui optimization
     - [ ] fixed left/right bar show field information
-- [ ] display standard ER diagram
+- [ ] display standard ER diagram `difficult`
     - [ ] display potential invalid links
 - [ ] integration with pydantic-resolve
     - [ ] show difference between resolve, post fields
     - [x] strikethrough for excluded fields
     - [ ] display loader as edges
+- [x] export voyager core data into json (for better debugging)
+    - [x] add api to rebuild core data from json, and render it
+- [x] fix Generic case  `test_generic.py`
+
+bugs & non feature:
+- [x] fix duplicated link from class and parent class, it also break clicking highlight
 - [ ] add tests
 - [ ] refactor
     - [ ] abstract render module
-- [ ] export voyager core data into json (for better debugging)
-    - [ ] add api to rebuild core data from json, and render it
-- [x] fix Generic case  `test_generic.py`
-
-bugs:
-- [ ] fix duplicated link from class and parent class, it also break clicking highlight
 
 ## Using with pydantic-resolve
 
@@ -203,5 +203,11 @@ TODO: ...
 
 ## Changelog
 
-- 0.6.2: 
-    - fix generic related issue
+- 0.7:
+    - 0.7.2
+        - keep links inside filtered nodes.
+    - 0.7.1
+        - support brief mode, you can use `--module_prefix tests.service` to show links between routes and filtered schemas, to make the graph less complicated.
+- 0.6: 
+    - 0.6.2: 
+        - fix generic related issue

{fastapi_voyager-0.6.2.dist-info → fastapi_voyager-0.7.2.dist-info}/RECORD

@@ -1,20 +1,20 @@
 fastapi_voyager/__init__.py,sha256=E5WTV_sYs2LK8I6jzA7AuvFU5a8_vjnDseC3DMha0iQ,149
-fastapi_voyager/cli.py,sha256=wWI0YEB7DaCfK6PkkEMOh3VxXGPV92NN8x-bB354Njs,10705
-fastapi_voyager/filter.py,sha256=uZrVVMhHG5E7j1wdsiB02RAyoDdF1q8A4J04oCboYAU,4644
+fastapi_voyager/cli.py,sha256=2eixX7mtPsZvukc4vrwQOt6XTPJgHUKIGLBy3IIC2jE,11127
+fastapi_voyager/filter.py,sha256=2Yt37o8mhqSqleafO4YRrumh_ExYUqzXFOxQRPuTbAc,8078
 fastapi_voyager/module.py,sha256=Z2QHNmiLk6ZAJlm2nSmO875Q33TweSg8UxZSzIpU9zY,3499
 fastapi_voyager/render.py,sha256=ctwad-KNbFajhgnA8OI8412s6s67UbV-dvZFXBt_Ssg,7410
-fastapi_voyager/server.py,sha256=6pDFrVd4AZP3KaqukJS7xq_f7ar4gLkb6XhSO5zoW_A,3782
+fastapi_voyager/server.py,sha256=UT-fHggdqicIo5m3uUX86-XFhAVDLXpXBsBQwd1HdIg,4001
 fastapi_voyager/type.py,sha256=nad4WNxTcZFi7Mskw6p2W7v2Gs4f0giVLNoFjZlKmbA,1778
 fastapi_voyager/type_helper.py,sha256=f2Gy5r3Zi6a2wTkbqU9W-AkvcetajEYCfroACzcIcVY,9064
-fastapi_voyager/version.py,sha256=esNfySzRU9jKScInt96qMV6-yrFqc5OFPWfMqNLhEWE,48
-fastapi_voyager/voyager.py,sha256=qMXZrQsmUiBljhOFOUtXV--a5i7THz9l00UNoxDu_0w,10193
+fastapi_voyager/version.py,sha256=PvKHya-0S2FiTp3eTEPk4ccgA86pgtDfAJ-N601Tvss,48
+fastapi_voyager/voyager.py,sha256=uOQEzrs3o6UUUswvHGRKELNWYUH1ix0Z7SbzMHwm320,10709
 fastapi_voyager/web/graph-ui.js,sha256=eEjDnJVMvk35LdRoxcqX_fZxLFS9_bUrGAZL6K2O5C0,4176
 fastapi_voyager/web/graphviz.svg.css,sha256=zDCjjpT0Idufu5YOiZI76PL70-avP3vTyzGPh9M85Do,1563
 fastapi_voyager/web/graphviz.svg.js,sha256=lvAdbjHc-lMSk4GQp-iqYA2PCFX4RKnW7dFaoe0LUHs,16005
-fastapi_voyager/web/index.html,sha256=hRbsdraGUZYB1dP_FIifyhp6ZqT0VyoNE-U3S4lCRaQ,10611
+fastapi_voyager/web/index.html,sha256=B_MaewnyHmEqozJHmPHYBfF6n5WOvrhoU3tDryS6ngY,10795
 fastapi_voyager/web/quasar.min.css,sha256=F5jQe7X2XT54VlvAaa2V3GsBFdVD-vxDZeaPLf6U9CU,203145
 fastapi_voyager/web/quasar.min.js,sha256=h0ftyPMW_CRiyzeVfQqiup0vrVt4_QWojpqmpnpn07E,502974
-fastapi_voyager/web/vue-main.js,sha256=twafDl_vDIRjlY0Oltslg2_UAO4L1fezKxuKszvnbjU,8799
+fastapi_voyager/web/vue-main.js,sha256=ZIxPa0mWOKK7xntfVWbAKzKSx_Tak-TLU-3rkr13wjI,8906
 fastapi_voyager/web/component/render-graph.js,sha256=e8Xgh2Kl-nYU0P1gstEmAepCgFnk2J6UdxW8TlMafGs,2322
 fastapi_voyager/web/component/route-code-display.js,sha256=NECC1OGcPCdDfbghtRJEnmFM6HmH5J3win2ibapWPeA,2649
 fastapi_voyager/web/component/schema-code-display.js,sha256=oOusgTvCaWGnoKb-NBwu0SXqJJf2PTUtp3lUczokTBM,5515
@@ -26,8 +26,8 @@ fastapi_voyager/web/icon/favicon-16x16.png,sha256=JC07jEzfIYxBIoQn_FHXvyHuxESdhW
 fastapi_voyager/web/icon/favicon-32x32.png,sha256=C7v1h58cfWOsiLp9yOIZtlx-dLasBcq3NqpHVGRmpt4,1859
 fastapi_voyager/web/icon/favicon.ico,sha256=tZolYIXkkBcFiYl1A8ksaXN2VjGamzcSdes838dLvNc,15406
 fastapi_voyager/web/icon/site.webmanifest,sha256=ep4Hzh9zhmiZF2At3Fp1dQrYQuYF_3ZPZxc1KcGBvwQ,263
-fastapi_voyager-0.6.2.dist-info/METADATA,sha256=LGxHZaMAoK5qMgAJiWZTM4ukoDrrndj4R3NUKsygzko,6866
-fastapi_voyager-0.6.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-fastapi_voyager-0.6.2.dist-info/entry_points.txt,sha256=pEIKoUnIDXEtdMBq8EmXm70m16vELIu1VPz9-TBUFWM,53
-fastapi_voyager-0.6.2.dist-info/licenses/LICENSE,sha256=lNVRR3y_bFVoFKuK2JM8N4sFaj3m-7j29kvL3olFi5Y,1067
-fastapi_voyager-0.6.2.dist-info/RECORD,,
+fastapi_voyager-0.7.2.dist-info/METADATA,sha256=4BI77g1dUQy0LM7Vkl8JlRgPBjFL07de6QHPFV5Dego,7154
+fastapi_voyager-0.7.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+fastapi_voyager-0.7.2.dist-info/entry_points.txt,sha256=pEIKoUnIDXEtdMBq8EmXm70m16vELIu1VPz9-TBUFWM,53
+fastapi_voyager-0.7.2.dist-info/licenses/LICENSE,sha256=lNVRR3y_bFVoFKuK2JM8N4sFaj3m-7j29kvL3olFi5Y,1067
+fastapi_voyager-0.7.2.dist-info/RECORD,,