justhtml 0.6.0__py3-none-any.whl

justhtml/parser.py

@@ -0,0 +1,86 @@
+"""Minimal JustHTML parser entry point."""
+
+from .tokenizer import Tokenizer, TokenizerOpts
+from .treebuilder import TreeBuilder
+
+
+class StrictModeError(SyntaxError):
+    """Raised when strict mode encounters a parse error.
+
+    Inherits from SyntaxError to provide Python 3.11+ enhanced error display
+    with source location highlighting.
+    """
+
+    def __init__(self, error):
+        self.error = error
+        # Use the ParseError's as_exception() to get enhanced display
+        exc = error.as_exception()
+        super().__init__(exc.msg)
+        # Copy SyntaxError attributes for enhanced display
+        self.filename = exc.filename
+        self.lineno = exc.lineno
+        self.offset = exc.offset
+        self.text = exc.text
+        self.end_lineno = getattr(exc, "end_lineno", None)
+        self.end_offset = getattr(exc, "end_offset", None)
+
+
+class JustHTML:
+    __slots__ = ("debug", "errors", "fragment_context", "root", "tokenizer", "tree_builder")
+
+    def __init__(
+        self,
+        html,
+        *,
+        collect_errors=False,
+        debug=False,
+        fragment_context=None,
+        iframe_srcdoc=False,
+        strict=False,
+        tokenizer_opts=None,
+        tree_builder=None,
+    ):
+        self.debug = bool(debug)
+        self.fragment_context = fragment_context
+
+        # Enable error collection if strict mode is on
+        should_collect = collect_errors or strict
+
+        self.tree_builder = tree_builder or TreeBuilder(
+            fragment_context=fragment_context,
+            iframe_srcdoc=iframe_srcdoc,
+            collect_errors=should_collect,
+        )
+        opts = tokenizer_opts or TokenizerOpts()
+
+        # For RAWTEXT fragment contexts, set initial tokenizer state and rawtext tag
+        if fragment_context and not fragment_context.namespace:
+            rawtext_elements = {"textarea", "title", "style"}
+            tag_name = fragment_context.tag_name.lower()
+            if tag_name in rawtext_elements:
+                opts.initial_state = Tokenizer.RAWTEXT
+                opts.initial_rawtext_tag = tag_name
+            elif tag_name in ("plaintext", "script"):
+                opts.initial_state = Tokenizer.PLAINTEXT
+
+        self.tokenizer = Tokenizer(self.tree_builder, opts, collect_errors=should_collect)
+        # Link tokenizer to tree_builder for position info
+        self.tree_builder.tokenizer = self.tokenizer
+
+        self.tokenizer.run(html or "")
+        self.root = self.tree_builder.finish()
+
+        # Merge errors from both tokenizer and tree builder
+        self.errors = self.tokenizer.errors + self.tree_builder.errors
+
+        # In strict mode, raise on first error
+        if strict and self.errors:
+            raise StrictModeError(self.errors[0])
+
+    def query(self, selector):
+        """Query the document using a CSS selector. Delegates to root.query()."""
+        return self.root.query(selector)
+
+    def to_html(self, pretty=True, indent_size=2):
+        """Serialize the document to HTML. Delegates to root.to_html()."""
+        return self.root.to_html(indent=0, indent_size=indent_size, pretty=pretty)