fix: pre commit issues

cisco-open · Apr 25, 2024 · fcced9a · fcced9a
1 parent 6c9c546
commit fcced9a
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 8 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -33,7 +33,7 @@ repos:
   # Code analysis (only checks staged files)
 
   - repo: https://github.com/pre-commit/mirrors-mypy
-    rev: v1.2.0
+    rev: v1.9.0
     hooks:
       - id: mypy
         name: mypy

diff --git a/endpoints-md.py b/endpoints-md.py
@@ -12,6 +12,7 @@
 from urllib.request import pathname2url
 
 from packaging.specifiers import SpecifierSet  # type: ignore
+from typing_extensions import get_original_bases
 
 from catalystwan import __package__
 from catalystwan.endpoints import BASE_PATH, APIEndpointRequestMeta, TypeSpecifier, request, versions, view
@@ -110,13 +111,20 @@ def from_type_specifier(typespec: TypeSpecifier) -> CompositeTypeLink:
                 if payloadtype.__module__ == "builtins":
                     return CompositeTypeLink.text_only(payloadtype.__name__)
                 elif sourcefile := getsourcefile(payloadtype):
-                    print(payloadtype)
-                    return CompositeTypeLink(
-                        link_text=payloadtype.__name__,
-                        sourcefile=create_sourcefile_link(sourcefile),
-                        lineno=getsourcelines(payloadtype)[1],
-                        origin=generate_origin_string(typespec),
-                    )
+                    try:
+                        lineno = getsourcelines(payloadtype)[1]
+                        link_text = payloadtype.__name__
+                    except OSError:
+                        base, *_ = get_original_bases(payloadtype)
+                        lineno = getsourcelines(base)[1]
+                        link_text = base.__name__
+                    finally:
+                        return CompositeTypeLink(
+                            link_text=link_text,
+                            sourcefile=create_sourcefile_link(sourcefile),
+                            lineno=lineno,
+                            origin=generate_origin_string(typespec),
+                        )
         return CompositeTypeLink.text_only("")
 
     def md(self) -> str: