Class RLStringHelper

rl_string_helper/rl_string_helper/string_helper.py:315–389

class RLStringHelper:
    def __init__(
        self,
        string: str,
        quote_html_type: list[str] = ["full"],
        _default_bang_char: str = "R",
    ):
        logger.trace("Initializing RLStringHelper")
        self.string: str = quote_symbol(string)
        self.templates: list[tuple[tuple[int, int], Template]] = []
        self.quote_replaces: list[tuple[tuple[int, int], str]] = []
        self.replaces: list[tuple[tuple[int, int], str]] = []
        self.quote_html_type = quote_html_type
        self.utf16_handler = UTF16Handler(_default_bang_char)
        self.template_renderer = TemplateRenderer()
        self.string_replacer = StringReplacer()

    def set_template(self, start: int, end: int, template: str | Template):
        logger.trace(f"Setting template for range {start}:{end}")
        if not isinstance(template, Template):
            template = jinja_env.from_string(template)
        self.templates.append(((start, end), template))
        logger.trace(f"Mutation: Added template for range {start}:{end}")

    def set_replace(self, start: int, end: int, replace_with: str):
        logger.trace(f"Setting replacement for range {start}:{end}")
        self.replaces.append(((start, end), replace_with))
        logger.trace(
            f"Mutation: Added replacement '{replace_with}' for range {start}:{end}"
        )

    def __str__(self):
        logger.trace("Converting RLStringHelper to string")
        string = StringAssignmentMixin(self.string)

        string_pos_matrix = list(range(len(string)))
        updated_text, string_pos_matrix, utf_16_bang_list = (
            self.utf16_handler.pre_utf_16_bang(string, string_pos_matrix)
        )

        if self.quote_html_type:
            logger.trace("Applying HTML quoting")
            self.quote_replaces = list(
                quote_html(str(updated_text), self.quote_html_type)
            )
            logger.trace(
                f"Mutation: Added {len(self.quote_replaces)} HTML quote replacements"
            )

        if not self.templates and not self.replaces and not self.quote_replaces:
            logger.trace("No modifications needed, returning original string")
            return self.string

        updated_text, string_pos_matrix, utf_16_bang_list = (
            self.template_renderer.render_templates(
                updated_text, string_pos_matrix, utf_16_bang_list, self.templates
            )
        )