Fix linenos flag

Author: sophstad

snooty/gizaparser/release.py

@@ -31,6 +31,7 @@ def render(self, page: Page, rst_parser: EmbeddedRstParser) -> List[n.Node]:
                     True if self.copyable is None else self.copyable,
                     None,
                     self.code,
+                    False,
                 )
             )
         return children

snooty/gizaparser/steps.py

@@ -47,6 +47,7 @@ def render(self, rst_parser: EmbeddedRstParser) -> List[n.Node]:
                     True if self.copyable is None else self.copyable,
                     None,
                     self.code,
+                    False,
                 )
             )
 

snooty/n.py

@@ -104,12 +104,13 @@ class InlineNode(Node):
 
 @dataclass
 class Code(Node):
-    __slots__ = ("lang", "copyable", "emphasize_lines", "value")
+    __slots__ = ("lang", "copyable", "emphasize_lines", "value", "linenos")
     type = "code"
     lang: Optional[str]
     copyable: bool
     emphasize_lines: Optional[Sequence[Tuple[int, int]]]
     value: str
+    linenos: bool
 
 
 @dataclass

snooty/parser.py

@@ -261,6 +261,7 @@ def dispatch_visit(self, node: docutils.nodes.Node) -> None:
                 node["copyable"],
                 node["emphasize_lines"] if "emphasize_lines" in node else None,
                 node.astext(),
+                node["linenos"],
             )
             top_of_state = self.state[-1]
             assert isinstance(top_of_state, n.Parent)
@@ -512,7 +513,14 @@ def handle_directive(
                 if "language" in options
                 else asset_path.suffix.lstrip(".")
             )
-            code = n.Code((line,), lang, "copyable" in options, [], "")
+            code = n.Code(
+                (line,),
+                lang,
+                "copyable" not in options or options["copyable"] == "true",
+                [],
+                "",
+                "linenos" in options,
+            )
 
             try:
                 static_asset = self.add_static_asset(asset_path, False)

snooty/rstparser.py

@@ -666,6 +666,7 @@ class BaseCodeDirective(docutils.parsers.rst.Directive):
     def run(self) -> List[docutils.nodes.Node]:
         source, line = self.state_machine.get_source_and_line(self.lineno)
         copyable = "copyable" not in self.options or self.options["copyable"] == "true"
+        linenos = "linenos" in self.options
 
         try:
             n_lines = len(self.content)
@@ -683,6 +684,7 @@ def run(self) -> List[docutils.nodes.Node]:
             node["lang"] = self.arguments[0]
         node["copyable"] = copyable
         node["emphasize_lines"] = emphasize_lines
+        node["linenos"] = linenos
         node.document = self.state.document
         node.source, node.line = source, line
         return [node]

snooty/test_parser.py

@@ -87,6 +87,7 @@ def test_codeblock() -> None:
 .. code-block:: sh
    :copyable: false
    :emphasize-lines: 1, 2-3
+   :linenos:
 
    foo
    bar
@@ -97,7 +98,7 @@ def test_codeblock() -> None:
     check_ast_testing_string(
         page.ast,
         """<root>
-        <code lang="sh" emphasize_lines="[(1, 1), (2, 3)]">foo\nbar\nbaz</code>
+        <code lang="sh" emphasize_lines="[(1, 1), (2, 3)]" linenos="True">foo\nbar\nbaz</code>
         </root>""",
     )
 
@@ -147,14 +148,15 @@ def test_literalinclude() -> None:
    :dedent:
    :start-after: Start Example 3
    :end-before: End Example 3
+   :linenos:
 """,
     )
     page.finish(diagnostics)
     assert diagnostics == []
     check_ast_testing_string(
         page.ast,
         """<root>
-<code lang="py">db.inventory.insert_many([
+<code lang="py" copyable="True" linenos="True">db.inventory.insert_many([
     {"item": "journal",
      "qty": 25,
      "tags": ["blank", "red"],