add MarkdownIt.reset_rules

Author: chrisjsewell

markdown_it/main.py

@@ -1,3 +1,4 @@
+from contextlib import contextmanager
 from typing import Any, Callable, Dict, List, Optional, Union
 
 from . import helpers, presets  # noqa F401
@@ -107,10 +108,12 @@ def configure(self, presets: Union[str, AttrDict]):
 
     def get_active_rules(self) -> Dict[str, List[str]]:
         """Return the names of all active rules."""
-        return {
+        rules = {
             chain: self[chain].ruler.get_active_rules()
             for chain in ["core", "block", "inline"]
         }
+        rules["inline2"] = self.inline.ruler2.get_active_rules()
+        return rules
 
     def enable(
         self, names: Union[str, List[str]], ignoreInvalid: bool = False
@@ -168,6 +171,16 @@ def disable(
             raise ValueError(f"MarkdownIt. Failed to disable unknown rule(s): {missed}")
         return self
 
+    @contextmanager
+    def reset_rules(self):
+        """A context manager, that will reset the current enabled rules on exit."""
+        chain_rules = self.get_active_rules()
+        yield
+        for chain, rules in chain_rules.items():
+            if chain != "inline2":
+                self[chain].ruler.enableOnly(rules)
+        self.inline.ruler2.enableOnly(chain_rules["inline2"])
+
     def add_render_rule(self, name: str, function: Callable, fmt="html"):
         """Add a rule for rendering a particular Token type.
 
@@ -243,7 +256,7 @@ def parseInline(self, src: str, env: Optional[AttrDict] = None) -> List[Token]:
             raise TypeError(f"Input data should be an AttrDict, not {type(env)}")
         if not isinstance(src, str):
             raise TypeError(f"Input data should be a string, not {type(src)}")
-        state = self.core.State(src, self, env)
+        state = StateCore(src, self, env)
         state.inlineMode = True
         self.core.process(state)
         return state.tokens

markdown_it/rules_core/block.py

@@ -9,6 +9,6 @@ def block(state: StateCore):
         token.content = state.src
         token.map = [0, 1]
         token.children = []
-        state.tokens.push(token)
+        state.tokens.append(token)
     else:
         state.md.block.parse(state.src, state.md, state.env, state.tokens)

tests/test_api/test_main.py

@@ -7,6 +7,7 @@ def test_load_presets():
         "block": ["paragraph"],
         "core": ["normalize", "block", "inline"],
         "inline": ["text"],
+        "inline2": ["balance_pairs", "text_collapse"],
     }
     md = MarkdownIt("commonmark")
     assert md.get_active_rules() == {
@@ -35,6 +36,7 @@ def test_load_presets():
             "html_inline",
             "entity",
         ],
+        "inline2": ["balance_pairs", "emphasis", "text_collapse"],
     }
 
 
@@ -44,12 +46,14 @@ def test_enable():
         "block": ["heading", "paragraph"],
         "core": ["normalize", "block", "inline"],
         "inline": ["text"],
+        "inline2": ["balance_pairs", "text_collapse"],
     }
     md.enable(["backticks", "autolink"])
     assert md.get_active_rules() == {
         "block": ["heading", "paragraph"],
         "core": ["normalize", "block", "inline"],
         "inline": ["text", "backticks", "autolink"],
+        "inline2": ["balance_pairs", "text_collapse"],
     }
 
 
@@ -59,10 +63,30 @@ def test_disable():
         "block": ["paragraph"],
         "core": ["normalize", "block"],
         "inline": ["text"],
+        "inline2": ["balance_pairs", "text_collapse"],
     }
     md.disable(["text"])
     assert md.get_active_rules() == {
         "block": ["paragraph"],
         "core": ["normalize", "block"],
         "inline": [],
+        "inline2": ["balance_pairs", "text_collapse"],
+    }
+
+
+def test_reset():
+    md = MarkdownIt("zero")
+    with md.reset_rules():
+        md.disable("inline")
+        assert md.get_active_rules() == {
+            "block": ["paragraph"],
+            "core": ["normalize", "block"],
+            "inline": ["text"],
+            "inline2": ["balance_pairs", "text_collapse"],
+        }
+    assert md.get_active_rules() == {
+        "block": ["paragraph"],
+        "core": ["normalize", "block", "inline"],
+        "inline": ["text"],
+        "inline2": ["balance_pairs", "text_collapse"],
     }