Reverted fixes and removed previous ones due to CP 9.0.0 fixing the issue on re module level

Author: michalpokusa

adafruit_templateengine.py

@@ -156,43 +156,38 @@ def safe_markdown(value: Any) -> str:
 _PRECOMPILED_BLOCK_PATTERN = re.compile(r"{% block \w+? %}")
 _PRECOMPILED_INCLUDE_PATTERN = re.compile(r"{% include '.+?' %}|{% include \".+?\" %}")
 _PRECOMPILED_HASH_COMMENT_PATTERN = re.compile(r"{# .+? #}")
-# Workaround for bug in re module https://github.com/adafruit/circuitpython/issues/8525
 _PRECOMPILED_BLOCK_COMMENT_PATTERN = re.compile(
-    # TODO: Use r"{% comment ('.*?' |\".*?\" )?%}[\s\S]*?{% endcomment %}" without flags when fixed
-    r"{% comment ('.*?' |\".*?\" )?%}.*?{% endcomment %}",
-    16,  # re.DOTALL flag
+    r"{% comment ('.*?' |\".*?\" )?%}[\s\S]*?{% endcomment %}"
 )
 _PRECOMPILED_TOKEN_PATTERN = re.compile(r"{{ .+? }}|{% .+? %}")
 
 
-def _find_next_extends(template: bytes) -> "re.Match[bytes]":
+def _find_next_extends(template: str):
     return _PRECOMPILED_EXTENDS_PATTERN.search(template)
 
 
-def _find_next_block(template: bytes) -> "re.Match[bytes]":
+def _find_next_block(template: str):
     return _PRECOMPILED_BLOCK_PATTERN.search(template)
 
 
-def _find_next_include(template: bytes) -> "re.Match[bytes]":
+def _find_next_include(template: str):
     return _PRECOMPILED_INCLUDE_PATTERN.search(template)
 
 
-def _find_named_endblock(template: bytes, name: bytes) -> "re.Match[bytes]":
-    return re.search(
-        r"{% endblock ".encode("utf-8") + name + r" %}".encode("utf-8"), template
-    )
+def _find_named_endblock(template: str, name: str):
+    return re.search(r"{% endblock " + name + r" %}", template)
 
 
-def _exists_and_is_file(path: str) -> bool:
+def _exists_and_is_file(path: str):
     try:
         return (os.stat(path)[0] & 0b_11110000_00000000) == 0b_10000000_00000000
     except OSError:
         return False
 
 
-def _resolve_includes(template: bytes) -> bytes:
+def _resolve_includes(template: str):
     while (include_match := _find_next_include(template)) is not None:
-        template_path = include_match.group(0)[12:-4].decode("utf-8")
+        template_path = include_match.group(0)[12:-4]
 
         # TODO: Restrict include to specific directory
 
@@ -203,19 +198,19 @@ def _resolve_includes(template: bytes) -> bytes:
         with open(template_path, "rt", encoding="utf-8") as template_file:
             template = (
                 template[: include_match.start()]
-                + template_file.read().encode("utf-8")
+                + template_file.read()
                 + template[include_match.end() :]
             )
     return template
 
 
-def _check_for_unsupported_nested_blocks(template: bytes):
+def _check_for_unsupported_nested_blocks(template: str):
     if _find_next_block(template) is not None:
         raise ValueError("Nested blocks are not supported")
 
 
-def _resolve_includes_blocks_and_extends(template: bytes) -> bytes:
-    block_replacements: "dict[bytes, bytes]" = {}
+def _resolve_includes_blocks_and_extends(template: str):
+    block_replacements: "dict[str, str]" = {}
 
     # Processing nested child templates
     while (extends_match := _find_next_extends(template)) is not None:
@@ -225,7 +220,7 @@ def _resolve_includes_blocks_and_extends(template: bytes) -> bytes:
         with open(
             extended_template_name, "rt", encoding="utf-8"
         ) as extended_template_file:
-            extended_template = extended_template_file.read().encode("utf-8")
+            extended_template = extended_template_file.read()
 
         # Removed the extend tag
         template = template[extends_match.end() :]
@@ -248,7 +243,7 @@ def _resolve_includes_blocks_and_extends(template: bytes) -> bytes:
 
             if block_name in block_replacements:
                 block_replacements[block_name] = block_replacements[block_name].replace(
-                    r"{{ block.super }}".encode("utf-8"), block_content
+                    r"{{ block.super }}", block_content
                 )
             else:
                 block_replacements.setdefault(block_name, block_content)
@@ -265,16 +260,14 @@ def _resolve_includes_blocks_and_extends(template: bytes) -> bytes:
     return _replace_blocks_with_replacements(template, block_replacements)
 
 
-def _replace_blocks_with_replacements(
-    template: bytes, replacements: "dict[bytes, bytes]"
-) -> bytes:
+def _replace_blocks_with_replacements(template: str, replacements: "dict[str, str]"):
     # Replace blocks in top-level template
     while (block_match := _find_next_block(template)) is not None:
         block_name = block_match.group(0)[9:-3]
 
         # Self-closing block tag without default content
         if (endblock_match := _find_named_endblock(template, block_name)) is None:
-            replacement = replacements.get(block_name, "".encode("utf-8"))
+            replacement = replacements.get(block_name, "")
 
             template = (
                 template[: block_match.start()]
@@ -299,7 +292,7 @@ def _replace_blocks_with_replacements(
             # Replace default content with replacement
             else:
                 replacement = replacements[block_name].replace(
-                    r"{{ block.super }}".encode("utf-8"), block_content
+                    r"{{ block.super }}", block_content
                 )
 
                 template = (
@@ -311,15 +304,15 @@ def _replace_blocks_with_replacements(
     return template
 
 
-def _find_next_hash_comment(template: bytes) -> "re.Match[bytes]":
+def _find_next_hash_comment(template: str):
     return _PRECOMPILED_HASH_COMMENT_PATTERN.search(template)
 
 
-def _find_next_block_comment(template: bytes) -> "re.Match[bytes]":
+def _find_next_block_comment(template: str):
     return _PRECOMPILED_BLOCK_COMMENT_PATTERN.search(template)
 
 
-def _remove_comments(template: bytes) -> bytes:
+def _remove_comments(template: str):
     # Remove hash comments: {# ... #}
     while (comment_match := _find_next_hash_comment(template)) is not None:
         template = template[: comment_match.start()] + template[comment_match.end() :]
@@ -331,7 +324,7 @@ def _remove_comments(template: bytes) -> bytes:
     return template
 
 
-def _find_next_token(template: bytes) -> "re.Match[bytes]":
+def _find_next_token(template: str):
     return _PRECOMPILED_TOKEN_PATTERN.search(template)
 
 
@@ -343,10 +336,6 @@ def _create_template_function(  # pylint: disable=,too-many-locals,too-many-bran
     context_name: str = "context",
     dry_run: bool = False,
 ) -> "Generator[str] | str":
-    # Workaround for bug in re module https://github.com/adafruit/circuitpython/issues/6860
-    # TODO: Remove .encode() and .decode() when bug is fixed
-    template: bytes = template.encode("utf-8")
-
     # Resolve includes, blocks and extends
     template = _resolve_includes_blocks_and_extends(template)
 
@@ -363,10 +352,10 @@ def _create_template_function(  # pylint: disable=,too-many-locals,too-many-bran
 
     # Resolve tokens
     while (token_match := _find_next_token(template)) is not None:
-        token: str = token_match.group(0).decode("utf-8")
+        token = token_match.group(0)
 
         # Add the text before the token
-        if text_before_token := template[: token_match.start()].decode("utf-8"):
+        if text_before_token := template[: token_match.start()]:
             function_string += (
                 indent * indentation_level + f"yield {repr(text_before_token)}\n"
             )
@@ -457,9 +446,7 @@ def _create_template_function(  # pylint: disable=,too-many-locals,too-many-bran
 
     # Add the text after the last token (if any)
     if template:
-        function_string += (
-            indent * indentation_level + f"yield {repr(template.decode('utf-8'))}\n"  #
-        )
+        function_string += indent * indentation_level + f"yield {repr(template)}\n"
 
     # If dry run, return the template function string
     if dry_run: