Extend typing in docstub

Author: lagru

src/docstub/_analysis.py

@@ -410,6 +410,8 @@ class TypesDatabase:
     ----------
     current_source : Path | None
     source_pkgs : list[Path]
+    known_imports: dict[str, KnownImport]
+    stats : dict[str, Any]
 
     Examples
     --------

src/docstub/_cli.py

@@ -144,6 +144,14 @@ def report_execution_time():
 @click.help_option("-h", "--help")
 @report_execution_time()
 def main(source_dir, out_dir, config_path, verbose):
+    """
+    Parameters
+    ----------
+    source_dir : Path
+    out_dir : Path
+    config_path : Path
+    verbose : str
+    """
     _setup_logging(verbose=verbose)
 
     source_dir = Path(source_dir)

src/docstub/_docstrings.py

@@ -150,7 +150,10 @@ class DoctypeTransformer(lark.visitors.Transformer):
 
     Attributes
     ----------
-    blacklisted_qualnames : frozenset[str]
+    types_db : ~.TypesDatabase
+    replace_doctypes : dict[str, str]
+    stats : dict[str, Any]
+    blacklisted_qualnames : ClassVar[frozenset[str]]
         All Python keywords [1]_ are blacklisted from use in qualnames except for ``True``
         ``False`` and ``None``.
 

src/docstub/_stubs.py

@@ -291,7 +291,9 @@ class Py2StubTransformer(cst.CSTTransformer):
 
     Attributes
     ----------
-    types_db : ~.TypesDatabase
+    types_db : ~.TypesDatabase | None
+    replace_doctypes : dict[str, str] | None
+    transformer : ~.DoctypeTransformer
 
     References
     ----------

src/docstub/_utils.py

@@ -156,9 +156,11 @@ class ContextFormatter:
         More details
     """
 
-    path: Path = None
-    line: int = None
-    column: int = None
+    # docstub: off
+    path: Path | None = None
+    line: int | None = None
+    column: int | None = None
+    # docstub: on
 
     def with_line(self, line=None, *, offset=0):
         """Return a new copy with a modified line.