diff mbox series

[35/57] docs/qapidoc: Fix static typing on qapidoc.py

Message ID 20250305034610.960147-36-jsnow@redhat.com (mailing list archive)
State New
Headers show
Series docs: Add new QAPI transmogrifier | expand

Commit Message

John Snow March 5, 2025, 3:45 a.m. UTC
Now that the legacy code is factored out, fix up the typing on the
remaining code in qapidoc.py. Add a type ignore to qapi_legacy.py to
prevent the errors there from bleeding out into qapidoc.py.

Signed-off-by: John Snow <jsnow@redhat.com>
---
 docs/sphinx/qapidoc.py        | 40 ++++++++++++++++++++++-------------
 docs/sphinx/qapidoc_legacy.py |  1 +
 2 files changed, 26 insertions(+), 15 deletions(-)

Comments

Markus Armbruster March 7, 2025, 11:59 a.m. UTC | #1
John Snow <jsnow@redhat.com> writes:

> Now that the legacy code is factored out, fix up the typing on the
> remaining code in qapidoc.py. Add a type ignore to qapi_legacy.py to
> prevent the errors there from bleeding out into qapidoc.py.
>
> Signed-off-by: John Snow <jsnow@redhat.com>
> ---
>  docs/sphinx/qapidoc.py        | 40 ++++++++++++++++++++++-------------
>  docs/sphinx/qapidoc_legacy.py |  1 +
>  2 files changed, 26 insertions(+), 15 deletions(-)
>
> diff --git a/docs/sphinx/qapidoc.py b/docs/sphinx/qapidoc.py
> index f4abf42e7bf..5246832b68c 100644
> --- a/docs/sphinx/qapidoc.py
> +++ b/docs/sphinx/qapidoc.py
> @@ -24,17 +24,18 @@
>  https://www.sphinx-doc.org/en/master/development/index.html
>  """
>  
> +from __future__ import annotations
> +
>  import os
>  import sys
> -from typing import List
> +from typing import TYPE_CHECKING
>  
>  from docutils import nodes
>  from docutils.parsers.rst import Directive, directives
>  from qapi.error import QAPIError
> -from qapi.gen import QAPISchemaVisitor
> -from qapi.schema import QAPISchema
> +from qapi.schema import QAPISchema, QAPISchemaVisitor
>  
> -from qapidoc_legacy import QAPISchemaGenRSTVisitor
> +from qapidoc_legacy import QAPISchemaGenRSTVisitor  # type: ignore
>  from sphinx import addnodes
>  from sphinx.directives.code import CodeBlock
>  from sphinx.errors import ExtensionError
> @@ -42,6 +43,15 @@
>  from sphinx.util.nodes import nested_parse_with_titles
>  
>  
> +if TYPE_CHECKING:
> +    from typing import Any, List, Sequence
> +
> +    from docutils.statemachine import StringList
> +
> +    from sphinx.application import Sphinx
> +    from sphinx.util.typing import ExtensionMetadata

Can you briefly explain why this needs to be conditional?

> +
> +
>  __version__ = "1.0"
>  
>  
> @@ -53,11 +63,11 @@ class QAPISchemaGenDepVisitor(QAPISchemaVisitor):
>      schema file associated with each module in the QAPI input.
>      """
>  
> -    def __init__(self, env, qapidir):
> +    def __init__(self, env: Any, qapidir: str) -> None:

@env is QAPIDocDirective.state.document.settings.env, i.e. something
deep in Sphinx.  I assume Any is the best Sphinx lets you do here.

>          self._env = env
>          self._qapidir = qapidir
>  
> -    def visit_module(self, name):
> +    def visit_module(self, name: str) -> None:
>          if name != "./builtin":
>              qapifile = self._qapidir + "/" + name
>              self._env.note_dependency(os.path.abspath(qapifile))
> @@ -65,10 +75,10 @@ def visit_module(self, name):
>  
>  
>  class NestedDirective(Directive):
> -    def run(self):
> +    def run(self) -> Sequence[nodes.Node]:
>          raise NotImplementedError
>  
> -    def do_parse(self, rstlist, node):
> +    def do_parse(self, rstlist: StringList, node: nodes.Node) -> None:
>          """
>          Parse rST source lines and add them to the specified node
>  
> @@ -93,15 +103,15 @@ class QAPIDocDirective(NestedDirective):
>      }
>      has_content = False
>  
> -    def new_serialno(self):
> +    def new_serialno(self) -> str:
>          """Return a unique new ID string suitable for use as a node's ID"""
>          env = self.state.document.settings.env
>          return "qapidoc-%d" % env.new_serialno("qapidoc")
>  
> -    def transmogrify(self, schema) -> nodes.Element:
> +    def transmogrify(self, schema: QAPISchema) -> nodes.Element:
>          raise NotImplementedError
>  
> -    def legacy(self, schema) -> nodes.Element:
> +    def legacy(self, schema: QAPISchema) -> nodes.Element:
>          vis = QAPISchemaGenRSTVisitor(self)
>          vis.visit_begin(schema)
>          for doc in schema.docs:
> @@ -109,9 +119,9 @@ def legacy(self, schema) -> nodes.Element:
>                  vis.symbol(doc, schema.lookup_entity(doc.symbol))
>              else:
>                  vis.freeform(doc)
> -        return vis.get_document_node()
> +        return vis.get_document_node()  # type: ignore

This is where you call qapidoc_legacy.py, which remains untyped.  Okay.

>  
> -    def run(self):
> +    def run(self) -> Sequence[nodes.Node]:
>          env = self.state.document.settings.env
>          qapifile = env.config.qapidoc_srctree + "/" + self.arguments[0]
>          qapidir = os.path.dirname(qapifile)
> @@ -185,7 +195,7 @@ def _highlightlang(self) -> addnodes.highlightlang:
>          )
>          return node
>  
> -    def admonition_wrap(self, *content) -> List[nodes.Node]:
> +    def admonition_wrap(self, *content: nodes.Node) -> List[nodes.Node]:
>          title = "Example:"
>          if "title" in self.options:
>              title = f"{title} {self.options['title']}"
> @@ -231,7 +241,7 @@ def run(self) -> List[nodes.Node]:
>          return self.admonition_wrap(*content_nodes)
>  
>  
> -def setup(app):
> +def setup(app: Sphinx) -> ExtensionMetadata:
>      """Register qapi-doc directive with Sphinx"""
>      app.add_config_value("qapidoc_srctree", None, "env")
>      app.add_directive("qapi-doc", QAPIDocDirective)
> diff --git a/docs/sphinx/qapidoc_legacy.py b/docs/sphinx/qapidoc_legacy.py
> index 679f38356b1..13520f4c26b 100644
> --- a/docs/sphinx/qapidoc_legacy.py
> +++ b/docs/sphinx/qapidoc_legacy.py
> @@ -1,4 +1,5 @@
>  # coding=utf-8
> +# type: ignore
>  #
>  # QEMU qapidoc QAPI file parsing extension
>  #

Types look good to me.
John Snow March 8, 2025, 6:32 p.m. UTC | #2
On Fri, Mar 7, 2025 at 7:00 AM Markus Armbruster <armbru@redhat.com> wrote:

> John Snow <jsnow@redhat.com> writes:
>
> > Now that the legacy code is factored out, fix up the typing on the
> > remaining code in qapidoc.py. Add a type ignore to qapi_legacy.py to
> > prevent the errors there from bleeding out into qapidoc.py.
> >
> > Signed-off-by: John Snow <jsnow@redhat.com>
> > ---
> >  docs/sphinx/qapidoc.py        | 40 ++++++++++++++++++++++-------------
> >  docs/sphinx/qapidoc_legacy.py |  1 +
> >  2 files changed, 26 insertions(+), 15 deletions(-)
> >
> > diff --git a/docs/sphinx/qapidoc.py b/docs/sphinx/qapidoc.py
> > index f4abf42e7bf..5246832b68c 100644
> > --- a/docs/sphinx/qapidoc.py
> > +++ b/docs/sphinx/qapidoc.py
> > @@ -24,17 +24,18 @@
> >  https://www.sphinx-doc.org/en/master/development/index.html
> >  """
> >
> > +from __future__ import annotations
> > +
> >  import os
> >  import sys
> > -from typing import List
> > +from typing import TYPE_CHECKING
> >
> >  from docutils import nodes
> >  from docutils.parsers.rst import Directive, directives
> >  from qapi.error import QAPIError
> > -from qapi.gen import QAPISchemaVisitor
> > -from qapi.schema import QAPISchema
> > +from qapi.schema import QAPISchema, QAPISchemaVisitor
> >
> > -from qapidoc_legacy import QAPISchemaGenRSTVisitor
> > +from qapidoc_legacy import QAPISchemaGenRSTVisitor  # type: ignore
> >  from sphinx import addnodes
> >  from sphinx.directives.code import CodeBlock
> >  from sphinx.errors import ExtensionError
> > @@ -42,6 +43,15 @@
> >  from sphinx.util.nodes import nested_parse_with_titles
> >
> >
> > +if TYPE_CHECKING:
> > +    from typing import Any, List, Sequence
> > +
> > +    from docutils.statemachine import StringList
> > +
> > +    from sphinx.application import Sphinx
> > +    from sphinx.util.typing import ExtensionMetadata
>
> Can you briefly explain why this needs to be conditional?
>

No requisite, but if they aren't used outside of type hints, they don't
actually need to be imported at runtime (when we use from __future__ import
annotations). Improves startup speed slightly and potentially makes the
plugin less porcelain at runtime.


>
> > +
> > +
> >  __version__ = "1.0"
> >
> >
> > @@ -53,11 +63,11 @@ class QAPISchemaGenDepVisitor(QAPISchemaVisitor):
> >      schema file associated with each module in the QAPI input.
> >      """
> >
> > -    def __init__(self, env, qapidir):
> > +    def __init__(self, env: Any, qapidir: str) -> None:
>
> @env is QAPIDocDirective.state.document.settings.env, i.e. something
> deep in Sphinx.  I assume Any is the best Sphinx lets you do here.
>

Will double-check, I did this a while ago.


>
> >          self._env = env
> >          self._qapidir = qapidir
> >
> > -    def visit_module(self, name):
> > +    def visit_module(self, name: str) -> None:
> >          if name != "./builtin":
> >              qapifile = self._qapidir + "/" + name
> >              self._env.note_dependency(os.path.abspath(qapifile))
> > @@ -65,10 +75,10 @@ def visit_module(self, name):
> >
> >
> >  class NestedDirective(Directive):
> > -    def run(self):
> > +    def run(self) -> Sequence[nodes.Node]:
> >          raise NotImplementedError
> >
> > -    def do_parse(self, rstlist, node):
> > +    def do_parse(self, rstlist: StringList, node: nodes.Node) -> None:
> >          """
> >          Parse rST source lines and add them to the specified node
> >
> > @@ -93,15 +103,15 @@ class QAPIDocDirective(NestedDirective):
> >      }
> >      has_content = False
> >
> > -    def new_serialno(self):
> > +    def new_serialno(self) -> str:
> >          """Return a unique new ID string suitable for use as a node's
> ID"""
> >          env = self.state.document.settings.env
> >          return "qapidoc-%d" % env.new_serialno("qapidoc")
> >
> > -    def transmogrify(self, schema) -> nodes.Element:
> > +    def transmogrify(self, schema: QAPISchema) -> nodes.Element:
> >          raise NotImplementedError
> >
> > -    def legacy(self, schema) -> nodes.Element:
> > +    def legacy(self, schema: QAPISchema) -> nodes.Element:
> >          vis = QAPISchemaGenRSTVisitor(self)
> >          vis.visit_begin(schema)
> >          for doc in schema.docs:
> > @@ -109,9 +119,9 @@ def legacy(self, schema) -> nodes.Element:
> >                  vis.symbol(doc, schema.lookup_entity(doc.symbol))
> >              else:
> >                  vis.freeform(doc)
> > -        return vis.get_document_node()
> > +        return vis.get_document_node()  # type: ignore
>
> This is where you call qapidoc_legacy.py, which remains untyped.  Okay.
>
> >
> > -    def run(self):
> > +    def run(self) -> Sequence[nodes.Node]:
> >          env = self.state.document.settings.env
> >          qapifile = env.config.qapidoc_srctree + "/" + self.arguments[0]
> >          qapidir = os.path.dirname(qapifile)
> > @@ -185,7 +195,7 @@ def _highlightlang(self) -> addnodes.highlightlang:
> >          )
> >          return node
> >
> > -    def admonition_wrap(self, *content) -> List[nodes.Node]:
> > +    def admonition_wrap(self, *content: nodes.Node) -> List[nodes.Node]:
> >          title = "Example:"
> >          if "title" in self.options:
> >              title = f"{title} {self.options['title']}"
> > @@ -231,7 +241,7 @@ def run(self) -> List[nodes.Node]:
> >          return self.admonition_wrap(*content_nodes)
> >
> >
> > -def setup(app):
> > +def setup(app: Sphinx) -> ExtensionMetadata:
> >      """Register qapi-doc directive with Sphinx"""
> >      app.add_config_value("qapidoc_srctree", None, "env")
> >      app.add_directive("qapi-doc", QAPIDocDirective)
> > diff --git a/docs/sphinx/qapidoc_legacy.py
> b/docs/sphinx/qapidoc_legacy.py
> > index 679f38356b1..13520f4c26b 100644
> > --- a/docs/sphinx/qapidoc_legacy.py
> > +++ b/docs/sphinx/qapidoc_legacy.py
> > @@ -1,4 +1,5 @@
> >  # coding=utf-8
> > +# type: ignore
> >  #
> >  # QEMU qapidoc QAPI file parsing extension
> >  #
>
> Types look good to me.
>
>
Markus Armbruster March 9, 2025, 5:37 a.m. UTC | #3
John Snow <jsnow@redhat.com> writes:

> On Fri, Mar 7, 2025 at 7:00 AM Markus Armbruster <armbru@redhat.com> wrote:
>
>> John Snow <jsnow@redhat.com> writes:
>>
>> > Now that the legacy code is factored out, fix up the typing on the
>> > remaining code in qapidoc.py. Add a type ignore to qapi_legacy.py to
>> > prevent the errors there from bleeding out into qapidoc.py.
>> >
>> > Signed-off-by: John Snow <jsnow@redhat.com>
>> > ---
>> >  docs/sphinx/qapidoc.py        | 40 ++++++++++++++++++++++-------------
>> >  docs/sphinx/qapidoc_legacy.py |  1 +
>> >  2 files changed, 26 insertions(+), 15 deletions(-)
>> >
>> > diff --git a/docs/sphinx/qapidoc.py b/docs/sphinx/qapidoc.py
>> > index f4abf42e7bf..5246832b68c 100644
>> > --- a/docs/sphinx/qapidoc.py
>> > +++ b/docs/sphinx/qapidoc.py
>> > @@ -24,17 +24,18 @@
>> >  https://www.sphinx-doc.org/en/master/development/index.html
>> >  """
>> >
>> > +from __future__ import annotations
>> > +
>> >  import os
>> >  import sys
>> > -from typing import List
>> > +from typing import TYPE_CHECKING
>> >
>> >  from docutils import nodes
>> >  from docutils.parsers.rst import Directive, directives
>> >  from qapi.error import QAPIError
>> > -from qapi.gen import QAPISchemaVisitor
>> > -from qapi.schema import QAPISchema
>> > +from qapi.schema import QAPISchema, QAPISchemaVisitor
>> >
>> > -from qapidoc_legacy import QAPISchemaGenRSTVisitor
>> > +from qapidoc_legacy import QAPISchemaGenRSTVisitor  # type: ignore
>> >  from sphinx import addnodes
>> >  from sphinx.directives.code import CodeBlock
>> >  from sphinx.errors import ExtensionError
>> > @@ -42,6 +43,15 @@
>> >  from sphinx.util.nodes import nested_parse_with_titles
>> >
>> >
>> > +if TYPE_CHECKING:
>> > +    from typing import Any, List, Sequence
>> > +
>> > +    from docutils.statemachine import StringList
>> > +
>> > +    from sphinx.application import Sphinx
>> > +    from sphinx.util.typing import ExtensionMetadata
>>
>> Can you briefly explain why this needs to be conditional?
>>
>
> No requisite, but if they aren't used outside of type hints, they don't
> actually need to be imported at runtime (when we use from __future__ import
> annotations). Improves startup speed slightly and potentially makes the
> plugin less porcelain at runtime.

Should we do that for all typing-only imports everywhere?

[...]
John Snow March 9, 2025, 6:46 a.m. UTC | #4
On Sun, Mar 9, 2025 at 12:38 AM Markus Armbruster <armbru@redhat.com> wrote:

> John Snow <jsnow@redhat.com> writes:
>
> > On Fri, Mar 7, 2025 at 7:00 AM Markus Armbruster <armbru@redhat.com>
> wrote:
> >
> >> John Snow <jsnow@redhat.com> writes:
> >>
> >> > Now that the legacy code is factored out, fix up the typing on the
> >> > remaining code in qapidoc.py. Add a type ignore to qapi_legacy.py to
> >> > prevent the errors there from bleeding out into qapidoc.py.
> >> >
> >> > Signed-off-by: John Snow <jsnow@redhat.com>
> >> > ---
> >> >  docs/sphinx/qapidoc.py        | 40
> ++++++++++++++++++++++-------------
> >> >  docs/sphinx/qapidoc_legacy.py |  1 +
> >> >  2 files changed, 26 insertions(+), 15 deletions(-)
> >> >
> >> > diff --git a/docs/sphinx/qapidoc.py b/docs/sphinx/qapidoc.py
> >> > index f4abf42e7bf..5246832b68c 100644
> >> > --- a/docs/sphinx/qapidoc.py
> >> > +++ b/docs/sphinx/qapidoc.py
> >> > @@ -24,17 +24,18 @@
> >> >  https://www.sphinx-doc.org/en/master/development/index.html
> >> >  """
> >> >
> >> > +from __future__ import annotations
> >> > +
> >> >  import os
> >> >  import sys
> >> > -from typing import List
> >> > +from typing import TYPE_CHECKING
> >> >
> >> >  from docutils import nodes
> >> >  from docutils.parsers.rst import Directive, directives
> >> >  from qapi.error import QAPIError
> >> > -from qapi.gen import QAPISchemaVisitor
> >> > -from qapi.schema import QAPISchema
> >> > +from qapi.schema import QAPISchema, QAPISchemaVisitor
> >> >
> >> > -from qapidoc_legacy import QAPISchemaGenRSTVisitor
> >> > +from qapidoc_legacy import QAPISchemaGenRSTVisitor  # type: ignore
> >> >  from sphinx import addnodes
> >> >  from sphinx.directives.code import CodeBlock
> >> >  from sphinx.errors import ExtensionError
> >> > @@ -42,6 +43,15 @@
> >> >  from sphinx.util.nodes import nested_parse_with_titles
> >> >
> >> >
> >> > +if TYPE_CHECKING:
> >> > +    from typing import Any, List, Sequence
> >> > +
> >> > +    from docutils.statemachine import StringList
> >> > +
> >> > +    from sphinx.application import Sphinx
> >> > +    from sphinx.util.typing import ExtensionMetadata
> >>
> >> Can you briefly explain why this needs to be conditional?
> >>
> >
> > No requisite, but if they aren't used outside of type hints, they don't
> > actually need to be imported at runtime (when we use from __future__
> import
> > annotations). Improves startup speed slightly and potentially makes the
> > plugin less porcelain at runtime.
>
> Should we do that for all typing-only imports everywhere?
>

Maybe! It's probably not too important, I just noticed Sphinx internals
doing it a lot and started following along because it seemed like a good
idea.
diff mbox series

Patch

diff --git a/docs/sphinx/qapidoc.py b/docs/sphinx/qapidoc.py
index f4abf42e7bf..5246832b68c 100644
--- a/docs/sphinx/qapidoc.py
+++ b/docs/sphinx/qapidoc.py
@@ -24,17 +24,18 @@ 
 https://www.sphinx-doc.org/en/master/development/index.html
 """
 
+from __future__ import annotations
+
 import os
 import sys
-from typing import List
+from typing import TYPE_CHECKING
 
 from docutils import nodes
 from docutils.parsers.rst import Directive, directives
 from qapi.error import QAPIError
-from qapi.gen import QAPISchemaVisitor
-from qapi.schema import QAPISchema
+from qapi.schema import QAPISchema, QAPISchemaVisitor
 
-from qapidoc_legacy import QAPISchemaGenRSTVisitor
+from qapidoc_legacy import QAPISchemaGenRSTVisitor  # type: ignore
 from sphinx import addnodes
 from sphinx.directives.code import CodeBlock
 from sphinx.errors import ExtensionError
@@ -42,6 +43,15 @@ 
 from sphinx.util.nodes import nested_parse_with_titles
 
 
+if TYPE_CHECKING:
+    from typing import Any, List, Sequence
+
+    from docutils.statemachine import StringList
+
+    from sphinx.application import Sphinx
+    from sphinx.util.typing import ExtensionMetadata
+
+
 __version__ = "1.0"
 
 
@@ -53,11 +63,11 @@  class QAPISchemaGenDepVisitor(QAPISchemaVisitor):
     schema file associated with each module in the QAPI input.
     """
 
-    def __init__(self, env, qapidir):
+    def __init__(self, env: Any, qapidir: str) -> None:
         self._env = env
         self._qapidir = qapidir
 
-    def visit_module(self, name):
+    def visit_module(self, name: str) -> None:
         if name != "./builtin":
             qapifile = self._qapidir + "/" + name
             self._env.note_dependency(os.path.abspath(qapifile))
@@ -65,10 +75,10 @@  def visit_module(self, name):
 
 
 class NestedDirective(Directive):
-    def run(self):
+    def run(self) -> Sequence[nodes.Node]:
         raise NotImplementedError
 
-    def do_parse(self, rstlist, node):
+    def do_parse(self, rstlist: StringList, node: nodes.Node) -> None:
         """
         Parse rST source lines and add them to the specified node
 
@@ -93,15 +103,15 @@  class QAPIDocDirective(NestedDirective):
     }
     has_content = False
 
-    def new_serialno(self):
+    def new_serialno(self) -> str:
         """Return a unique new ID string suitable for use as a node's ID"""
         env = self.state.document.settings.env
         return "qapidoc-%d" % env.new_serialno("qapidoc")
 
-    def transmogrify(self, schema) -> nodes.Element:
+    def transmogrify(self, schema: QAPISchema) -> nodes.Element:
         raise NotImplementedError
 
-    def legacy(self, schema) -> nodes.Element:
+    def legacy(self, schema: QAPISchema) -> nodes.Element:
         vis = QAPISchemaGenRSTVisitor(self)
         vis.visit_begin(schema)
         for doc in schema.docs:
@@ -109,9 +119,9 @@  def legacy(self, schema) -> nodes.Element:
                 vis.symbol(doc, schema.lookup_entity(doc.symbol))
             else:
                 vis.freeform(doc)
-        return vis.get_document_node()
+        return vis.get_document_node()  # type: ignore
 
-    def run(self):
+    def run(self) -> Sequence[nodes.Node]:
         env = self.state.document.settings.env
         qapifile = env.config.qapidoc_srctree + "/" + self.arguments[0]
         qapidir = os.path.dirname(qapifile)
@@ -185,7 +195,7 @@  def _highlightlang(self) -> addnodes.highlightlang:
         )
         return node
 
-    def admonition_wrap(self, *content) -> List[nodes.Node]:
+    def admonition_wrap(self, *content: nodes.Node) -> List[nodes.Node]:
         title = "Example:"
         if "title" in self.options:
             title = f"{title} {self.options['title']}"
@@ -231,7 +241,7 @@  def run(self) -> List[nodes.Node]:
         return self.admonition_wrap(*content_nodes)
 
 
-def setup(app):
+def setup(app: Sphinx) -> ExtensionMetadata:
     """Register qapi-doc directive with Sphinx"""
     app.add_config_value("qapidoc_srctree", None, "env")
     app.add_directive("qapi-doc", QAPIDocDirective)
diff --git a/docs/sphinx/qapidoc_legacy.py b/docs/sphinx/qapidoc_legacy.py
index 679f38356b1..13520f4c26b 100644
--- a/docs/sphinx/qapidoc_legacy.py
+++ b/docs/sphinx/qapidoc_legacy.py
@@ -1,4 +1,5 @@ 
 # coding=utf-8
+# type: ignore
 #
 # QEMU qapidoc QAPI file parsing extension
 #