pytrilogy 0.3.138__cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl

trilogy/parsing/render.py

@@ -0,0 +1,750 @@
+from collections import defaultdict
+from contextlib import contextmanager
+from dataclasses import dataclass
+from datetime import date, datetime
+from functools import singledispatchmethod
+from typing import Any
+
+from jinja2 import Template
+
+from trilogy.constants import DEFAULT_NAMESPACE, VIRTUAL_CONCEPT_PREFIX, MagicConstants
+from trilogy.core.enums import (
+    ConceptSource,
+    DatasourceState,
+    DatePart,
+    FunctionType,
+    Modifier,
+    PersistMode,
+    Purpose,
+    ValidationScope,
+)
+from trilogy.core.models.author import (
+    AggregateWrapper,
+    AlignClause,
+    AlignItem,
+    CaseElse,
+    CaseWhen,
+    Comment,
+    Comparison,
+    Concept,
+    ConceptRef,
+    Conditional,
+    FilterItem,
+    Function,
+    FunctionCallWrapper,
+    Grain,
+    OrderBy,
+    Ordering,
+    OrderItem,
+    Parenthetical,
+    SubselectComparison,
+    WhereClause,
+    WindowItem,
+)
+from trilogy.core.models.core import (
+    ArrayType,
+    DataType,
+    ListWrapper,
+    MapWrapper,
+    NumericType,
+    TraitDataType,
+    TupleWrapper,
+)
+from trilogy.core.models.datasource import (
+    Address,
+    ColumnAssignment,
+    Datasource,
+    Query,
+    RawColumnExpr,
+)
+from trilogy.core.models.environment import Environment, Import
+from trilogy.core.statements.author import (
+    ArgBinding,
+    ConceptDeclarationStatement,
+    ConceptDerivationStatement,
+    ConceptTransform,
+    CopyStatement,
+    FunctionDeclaration,
+    ImportStatement,
+    KeyMergeStatement,
+    MergeStatementV2,
+    MultiSelectStatement,
+    PersistStatement,
+    RawSQLStatement,
+    RowsetDerivationStatement,
+    SelectItem,
+    SelectStatement,
+    TypeDeclaration,
+    ValidateStatement,
+)
+
+QUERY_TEMPLATE = Template(
+    """{% if where %}WHERE
+{{ where }}
+{% endif %}SELECT{%- for select in select_columns %}
+{{ select }},{% endfor %}{% if having %}
+HAVING
+{{ having }}
+{% endif %}{%- if order_by %}
+ORDER BY{% for order in order_by %}
+{{ order }}{% if not loop.last %},{% endif %}{% endfor %}{% endif %}{%- if limit is not none %}
+LIMIT {{ limit }}{% endif %}
+;"""
+)
+
+
+@dataclass
+class IndentationContext:
+    """Tracks indentation state during rendering"""
+
+    depth: int = 0
+    indent_string: str = "    "  # 4 spaces by default
+
+    @property
+    def current_indent(self) -> str:
+        return self.indent_string * self.depth
+
+    def increase_depth(self, extra_levels: int = 1) -> "IndentationContext":
+        return IndentationContext(
+            depth=self.depth + extra_levels, indent_string=self.indent_string
+        )
+
+
+class Renderer:
+
+    def __init__(
+        self, environment: Environment | None = None, indent_string: str = "    "
+    ):
+        self.environment = environment
+        self.indent_context = IndentationContext(indent_string=indent_string)
+
+    @contextmanager
+    def indented(self, levels: int = 1):
+        """Context manager for temporarily increasing indentation"""
+        old_context = self.indent_context
+        self.indent_context = self.indent_context.increase_depth(levels)
+        try:
+            yield
+        finally:
+            self.indent_context = old_context
+
+    def indent_lines(self, text: str, extra_levels: int = 0) -> str:
+        """Apply current indentation to all lines in text"""
+        if not text:
+            return text
+
+        indent = self.indent_context.indent_string * (
+            self.indent_context.depth + extra_levels
+        )
+        lines = text.split("\n")
+        indented_lines = []
+
+        for line in lines:
+            if line.strip():  # Only indent non-empty lines
+                indented_lines.append(indent + line)
+            else:
+                indented_lines.append(line)  # Keep empty lines as-is
+
+        return "\n".join(indented_lines)
+
+    def render_statement_string(self, list_of_statements: list[Any]) -> str:
+        new = []
+        last_statement_type = None
+        for stmt in list_of_statements:
+            stmt_type = type(stmt)
+            if last_statement_type is None:
+                pass
+            elif last_statement_type == Comment:
+                new.append("\n")
+            elif stmt_type != last_statement_type:
+                new.append("\n\n")
+            else:
+                new.append("\n")
+            new.append(self.to_string(stmt))
+            last_statement_type = stmt_type
+        return "".join(new)
+
+    @singledispatchmethod
+    def to_string(self, arg):
+        raise NotImplementedError("Cannot render type {}".format(type(arg)))
+
+    @to_string.register
+    def _(self, arg: Environment):
+        output_concepts = []
+        constants: list[Concept] = []
+        keys: list[Concept] = []
+        properties = defaultdict(list)
+        metrics = []
+        # first, keys
+        for concept in arg.concepts.values():
+            if "__preql_internal" in concept.address:
+                continue
+
+            # don't render anything that came from an import
+            if concept.namespace in arg.imports:
+                continue
+            if (
+                concept.metadata
+                and concept.metadata.concept_source == ConceptSource.AUTO_DERIVED
+            ):
+                continue
+            elif not concept.lineage and concept.purpose == Purpose.CONSTANT:
+                constants.append(concept)
+            elif not concept.lineage and concept.purpose == Purpose.KEY:
+                keys.append(concept)
+
+            elif not concept.lineage and concept.purpose == Purpose.PROPERTY:
+                if concept.keys:
+                    # avoid duplicate declarations
+                    # but we need better composite key support
+                    for key in sorted(list(concept.keys))[:1]:
+                        properties[key].append(concept)
+                else:
+                    keys.append(concept)
+            else:
+                metrics.append(concept)
+
+        output_concepts = constants
+        for key_concept in keys:
+            output_concepts += [key_concept]
+            output_concepts += properties.get(key_concept.name, [])
+        output_concepts += metrics
+
+        rendered_concepts = [
+            self.to_string(ConceptDeclarationStatement(concept=concept))
+            for concept in output_concepts
+        ]
+
+        rendered_datasources = [
+            # extra padding between datasources
+            # todo: make this more generic
+            self.to_string(datasource) + "\n"
+            for datasource in arg.datasources.values()
+            if datasource.namespace == DEFAULT_NAMESPACE
+        ]
+        rendered_imports = []
+        for _, imports in arg.imports.items():
+            for import_statement in imports:
+                rendered_imports.append(self.to_string(import_statement))
+        components = []
+        if rendered_imports:
+            components.append(rendered_imports)
+        if rendered_concepts:
+            components.append(rendered_concepts)
+        if rendered_datasources:
+            components.append(rendered_datasources)
+
+        final = "\n\n".join("\n".join(x) for x in components)
+        return final
+
+    @to_string.register
+    def _(self, arg: TypeDeclaration):
+        return f"type {arg.type.name} {self.to_string(arg.type.type)};"
+
+    @to_string.register
+    def _(self, arg: ArgBinding):
+        if arg.default:
+            return f"{arg.name}={self.to_string(arg.default)}"
+        return f"{arg.name}"
+
+    @to_string.register
+    def _(self, arg: FunctionDeclaration):
+        args = ", ".join([self.to_string(x) for x in arg.args])
+        return f"def {arg.name}({args}) -> {self.to_string(arg.expr)};"
+
+    @to_string.register
+    def _(self, arg: Datasource):
+        with self.indented():
+            assignments = ",\n".join(
+                [self.indent_lines(self.to_string(x)) for x in arg.columns]
+            )
+
+        if arg.non_partial_for:
+            non_partial = f"\ncomplete where {self.to_string(arg.non_partial_for)}"
+        else:
+            non_partial = ""
+
+        base = f"""datasource {arg.name} (
+{assignments}
+)
+{self.to_string(arg.grain) if arg.grain.components else ''}{non_partial}
+{self.to_string(arg.address)}"""
+        if arg.where:
+            base += f"\nwhere {self.to_string(arg.where)}"
+
+        if arg.incremental_by:
+            base += f"\nincremental by {','.join(self.to_string(x) for x in arg.incremental_by)}"
+
+        if arg.partition_by:
+            base += f"\npartition by {','.join(self.to_string(x) for x in arg.partition_by)}"
+        if arg.status != DatasourceState.PUBLISHED:
+            base += f"\nstate {arg.status.value.lower()}"
+
+        base += ";"
+        return base
+
+    @to_string.register
+    def _(self, arg: "Grain"):
+        final = []
+        for comp in arg.components:
+            if comp.startswith(DEFAULT_NAMESPACE):
+                final.append(comp.split(".", 1)[1])
+            else:
+                final.append(comp)
+        final = sorted(final)
+        components = ",".join(x for x in final)
+        return f"grain ({components})"
+
+    @to_string.register
+    def _(self, arg: "Query"):
+        return f"""query {arg.text}"""
+
+    @to_string.register
+    def _(self, arg: RowsetDerivationStatement):
+        return f"""rowset {arg.name} <- {self.to_string(arg.select)}"""
+
+    @to_string.register
+    def _(self, arg: "CaseWhen"):
+        return (
+            f"""WHEN {self.to_string(arg.comparison)} THEN {self.to_string(arg.expr)}"""
+        )
+
+    @to_string.register
+    def _(self, arg: "CaseElse"):
+        return f"""ELSE {self.to_string(arg.expr)}"""
+
+    @to_string.register
+    def _(self, arg: "FunctionCallWrapper"):
+        args = [self.to_string(c) for c in arg.args]
+        arg_string = ", ".join(args)
+        return f"""@{arg.name}({arg_string})"""
+
+    @to_string.register
+    def _(self, arg: "Parenthetical"):
+        return f"""({self.to_string(arg.content)})"""
+
+    @to_string.register
+    def _(self, arg: DataType):
+        return arg.value
+
+    @to_string.register
+    def _(self, arg: "NumericType"):
+        return f"""Numeric({arg.precision},{arg.scale})"""
+
+    @to_string.register
+    def _(self, arg: TraitDataType):
+        traits = "::".join([x for x in arg.traits])
+        return f"{self.to_string(arg.data_type)}::{traits}"
+
+    @to_string.register
+    def _(self, arg: ListWrapper):
+        return "[" + ", ".join([self.to_string(x) for x in arg]) + "]"
+
+    @to_string.register
+    def _(self, arg: TupleWrapper):
+        return "(" + ", ".join([self.to_string(x) for x in arg]) + ")"
+
+    @to_string.register
+    def _(self, arg: MapWrapper):
+        def process_key_value(key, value):
+            return f"{self.to_string(key)}: {self.to_string(value)}"
+
+        return (
+            "{"
+            + ", ".join([process_key_value(key, value) for key, value in arg.items()])
+            + "}"
+        )
+
+    @to_string.register
+    def _(self, arg: DatePart):
+        return arg.value
+
+    @to_string.register
+    def _(self, arg: "Address"):
+        if arg.is_query:
+            if arg.location.startswith("("):
+                return f"query '''{arg.location[1:-1]}'''"
+            return f"query '''{arg.location}'''"
+        return f"address {arg.location}"
+
+    @to_string.register
+    def _(self, arg: "RawSQLStatement"):
+        return f"raw_sql('''{arg.text}''');"
+
+    @to_string.register
+    def _(self, arg: "MagicConstants"):
+        if arg == MagicConstants.NULL:
+            return "null"
+        return arg.value
+
+    @to_string.register
+    def _(self, arg: "ColumnAssignment"):
+        if arg.modifiers:
+            modifiers = "".join(
+                [self.to_string(modifier) for modifier in sorted(arg.modifiers)]
+            )
+        else:
+            modifiers = ""
+        if isinstance(arg.alias, str):
+            return f"{arg.alias}: {modifiers}{self.to_string(arg.concept)}"
+        return f"{self.to_string(arg.alias)}: {modifiers}{self.to_string(arg.concept)}"
+
+    @to_string.register
+    def _(self, arg: "RawColumnExpr"):
+        return f"raw('''{arg.text}''')"
+
+    @to_string.register
+    def _(self, arg: "ConceptDeclarationStatement"):
+        concept = arg.concept
+        if concept.metadata and concept.metadata.description:
+            base_description = concept.metadata.description
+        else:
+            base_description = None
+        if concept.namespace and concept.namespace != DEFAULT_NAMESPACE:
+            namespace = f"{concept.namespace}."
+        else:
+            namespace = ""
+        if not concept.lineage:
+            if concept.purpose == Purpose.PROPERTY and concept.keys:
+                if len(concept.keys) == 1:
+                    output = f"{concept.purpose.value} {self.to_string(ConceptRef(address=list(concept.keys)[0]))}.{namespace}{concept.name} {self.to_string(concept.datatype)};"
+                else:
+                    keys = ",".join(
+                        sorted(
+                            list(
+                                self.to_string(ConceptRef(address=x))
+                                for x in concept.keys
+                            )
+                        )
+                    )
+                    output = f"{concept.purpose.value} <{keys}>.{namespace}{concept.name} {self.to_string(concept.datatype)};"
+            else:
+                output = f"{concept.purpose.value} {namespace}{concept.name} {self.to_string(concept.datatype)};"
+        else:
+            output = f"{concept.purpose.value} {namespace}{concept.name} <- {self.to_string(concept.lineage)};"
+        if base_description:
+            lines = "\n#".join(base_description.split("\n"))
+            output += f" #{lines}"
+        return output
+
+    @to_string.register
+    def _(self, arg: ArrayType):
+        return f"list<{self.to_string(arg.value_data_type)}>"
+
+    @to_string.register
+    def _(self, arg: DataType):
+        return arg.value
+
+    @to_string.register
+    def _(self, arg: date):
+        return f"'{arg.isoformat()}'::date"
+
+    @to_string.register
+    def _(self, arg: datetime):
+        return f"'{arg.isoformat()}'::datetime"
+
+    @to_string.register
+    def _(self, arg: ConceptDerivationStatement):
+        # this is identical rendering;
+        return self.to_string(ConceptDeclarationStatement(concept=arg.concept))
+
+    @to_string.register
+    def _(self, arg: PersistStatement):
+        if arg.persist_mode == PersistMode.APPEND:
+            keyword = "APPEND"
+        else:
+            keyword = "OVERWRITE"
+        if arg.partition_by:
+            partition_by = (
+                f"BY {', '.join(self.to_string(x) for x in arg.partition_by)}"
+            )
+            return f"{keyword} {arg.identifier} INTO {arg.address.location} {partition_by} FROM {self.to_string(arg.select)}"
+        return f"{keyword} {arg.identifier} INTO {arg.address.location} FROM {self.to_string(arg.select)}"
+
+    @to_string.register
+    def _(self, arg: SelectItem):
+        prefixes = []
+        if Modifier.HIDDEN in arg.modifiers:
+            prefixes.append("--")
+        if Modifier.PARTIAL in arg.modifiers:
+            prefixes.append("~")
+        final = "".join(prefixes)
+        return f"{final}{self.to_string(arg.content)}"
+
+    @to_string.register
+    def _(self, arg: ValidateStatement):
+        targets = ",".join(arg.targets) if arg.targets else "*"
+        if arg.scope.value == ValidationScope.ALL:
+            return "validate all;"
+        return f"validate {arg.scope.value} {targets};"
+
+    @to_string.register
+    def _(self, arg: SelectStatement):
+        with self.indented():
+            select_columns = [
+                self.indent_lines(self.to_string(c)) for c in arg.selection
+            ]
+            where_clause = None
+            if arg.where_clause:
+                where_clause = self.indent_lines(self.to_string(arg.where_clause))
+            having_clause = None
+            if arg.having_clause:
+                having_clause = self.indent_lines(self.to_string(arg.having_clause))
+            order_by = None
+            if arg.order_by:
+                order_by = [
+                    self.indent_lines(self.to_string(c)) for c in arg.order_by.items
+                ]
+
+        return QUERY_TEMPLATE.render(
+            select_columns=select_columns,
+            where=where_clause,
+            having=having_clause,
+            order_by=order_by,
+            limit=arg.limit,
+        )
+
+    @to_string.register
+    def _(self, arg: MultiSelectStatement):
+        # Each select gets its own indentation
+        select_parts = []
+        for select in arg.selects:
+            select_parts.append(
+                self.to_string(select)[:-2]
+            )  # Remove the trailing ";\n"
+
+        base = "\nMERGE\n".join(select_parts)
+        base += self.to_string(arg.align)
+        if arg.where_clause:
+            base += f"\nWHERE\n{self.to_string(arg.where_clause)}"
+        if arg.order_by:
+            base += f"\nORDER BY\n{self.to_string(arg.order_by)}"
+        if arg.limit:
+            base += f"\nLIMIT {arg.limit}"
+        base += "\n;"
+        return base
+
+    @to_string.register
+    def _(self, arg: CopyStatement):
+        return f"COPY INTO {arg.target_type.value.upper()} '{arg.target}' FROM {self.to_string(arg.select)}"
+
+    @to_string.register
+    def _(self, arg: AlignClause):
+        with self.indented():
+            align_items = [self.indent_lines(self.to_string(c)) for c in arg.items]
+        return "\nALIGN\n" + ",\n".join(align_items)
+
+    @to_string.register
+    def _(self, arg: AlignItem):
+        return f"{arg.alias}:{','.join([self.to_string(c) for c in arg.concepts])}"
+
+    @to_string.register
+    def _(self, arg: OrderBy):
+        with self.indented():
+            order_items = [self.indent_lines(self.to_string(c)) for c in arg.items]
+        return ",\n".join(order_items)
+
+    @to_string.register
+    def _(self, arg: Ordering):
+        return arg.value
+
+    @to_string.register
+    def _(self, arg: "WhereClause"):
+        base = f"{self.to_string(arg.conditional)}"
+        if base[0] == "(" and base[-1] == ")":
+            return base[1:-1]
+        return base
+
+    @to_string.register
+    def _(self, arg: "Conditional"):
+        return f"{self.to_string(arg.left)} {arg.operator.value} {self.to_string(arg.right)}"
+
+    @to_string.register
+    def _(self, arg: "SubselectComparison"):
+        return f"{self.to_string(arg.left)} {arg.operator.value} {self.to_string(arg.right)}"
+
+    @to_string.register
+    def _(self, arg: "Comparison"):
+        return f"{self.to_string(arg.left)} {arg.operator.value} {self.to_string(arg.right)}"
+
+    @to_string.register
+    def _(self, arg: "Comment"):
+        lines = "\n#".join(arg.text.split("\n"))
+        return f"{lines}"
+
+    @to_string.register
+    def _(self, arg: "WindowItem"):
+        over = ",".join(self.to_string(c) for c in arg.over)
+        order = ",".join(self.to_string(c) for c in arg.order_by)
+        if over and order:
+            return (
+                f"{arg.type.value} {self.to_string(arg.content)} by {order} over {over}"
+            )
+        elif over:
+            return f"{arg.type.value} {self.to_string(arg.content)} over {over}"
+        return f"{arg.type.value} {self.to_string(arg.content)} by {order}"
+
+    @to_string.register
+    def _(self, arg: "FilterItem"):
+        return f"filter {self.to_string(arg.content)} where {self.to_string(arg.where)}"
+
+    @to_string.register
+    def _(self, arg: "ConceptRef"):
+        if arg.address == "__preql_internal.all_rows":
+            return "*"
+        if arg.name.startswith(VIRTUAL_CONCEPT_PREFIX) and self.environment:
+            return self.to_string(self.environment.concepts[arg.address])
+
+        ns, base = arg.address.rsplit(".", 1)
+        if ns == DEFAULT_NAMESPACE:
+            return base
+        return arg.address
+
+    @to_string.register
+    def _(self, arg: "ImportStatement"):
+        path: str = str(arg.path).replace("\\", ".")
+        path = path.replace("/", ".")
+        if path.endswith(".preql"):
+            path = path.rsplit(".", 1)[0]
+        if path.startswith("."):
+            path = path[1:]
+        if arg.alias == DEFAULT_NAMESPACE or not arg.alias:
+            return f"import {path};"
+        return f"import {path} as {arg.alias};"
+
+    @to_string.register
+    def _(self, arg: "Import"):
+        path: str = str(arg.path).replace("\\", ".")
+        path = path.replace("/", ".")
+        if path.endswith(".preql"):
+            path = path.rsplit(".", 1)[0]
+        if path.startswith("."):
+            path = path[1:]
+        if arg.alias == DEFAULT_NAMESPACE or not arg.alias:
+            return f"import {path};"
+        return f"import {path} as {arg.alias};"
+
+    @to_string.register
+    def _(self, arg: "Concept"):
+        if arg.name.startswith(VIRTUAL_CONCEPT_PREFIX):
+            return self.to_string(arg.lineage)
+        if arg.namespace == DEFAULT_NAMESPACE:
+            return arg.name
+        return arg.address
+
+    @to_string.register
+    def _(self, arg: "ConceptTransform"):
+        return f"{self.to_string(arg.function)} -> {arg.output.name}"
+
+    @to_string.register
+    def _(self, arg: "Function"):
+        args = [self.to_string(c) for c in arg.arguments]
+
+        if arg.operator == FunctionType.SUBTRACT:
+            return " - ".join(args)
+        if arg.operator == FunctionType.ADD:
+            return " + ".join(args)
+        if arg.operator == FunctionType.MULTIPLY:
+            return " * ".join(args)
+        if arg.operator == FunctionType.DIVIDE:
+            return " / ".join(args)
+        if arg.operator == FunctionType.MOD:
+            return f"{args[0]} % {args[1]}"
+        if arg.operator == FunctionType.PARENTHETICAL:
+            return f"({args[0]})"
+        if arg.operator == FunctionType.GROUP:
+            arg_string = ", ".join(args[1:])
+            if len(args) == 1:
+                return f"group({args[0]})"
+            return f"group({args[0]}) by {arg_string}"
+
+        if arg.operator == FunctionType.CONSTANT:
+            return f"{', '.join(args)}"
+        if arg.operator == FunctionType.CAST:
+            return f"{self.to_string(arg.arguments[0])}::{self.to_string(arg.arguments[1])}"
+        if arg.operator == FunctionType.INDEX_ACCESS:
+            return f"{self.to_string(arg.arguments[0])}[{self.to_string(arg.arguments[1])}]"
+
+        if arg.operator == FunctionType.CASE:
+            with self.indented():
+                indented_args = [
+                    self.indent_lines(self.to_string(a)) for a in arg.arguments
+                ]
+            inputs = "\n".join(indented_args)
+            return f"CASE\n{inputs}\n{self.indent_context.current_indent}END"
+
+        if arg.operator == FunctionType.STRUCT:
+            # zip arguments to pairs
+            input_pairs = zip(arg.arguments[0::2], arg.arguments[1::2])
+            with self.indented():
+                pair_strings = []
+                for k, v in input_pairs:
+                    pair_line = f"{self.to_string(k)}-> {v}"
+                    pair_strings.append(self.indent_lines(pair_line))
+            inputs = ",\n".join(pair_strings)
+            return f"struct(\n{inputs}\n{self.indent_context.current_indent})"
+        if arg.operator == FunctionType.ALIAS:
+            return f"{self.to_string(arg.arguments[0])}"
+        inputs = ",".join(args)
+        return f"{arg.operator.value}({inputs})"
+
+    @to_string.register
+    def _(self, arg: "OrderItem"):
+        return f"{self.to_string(arg.expr)} {arg.order.value}"
+
+    @to_string.register
+    def _(self, arg: AggregateWrapper):
+        if arg.by:
+            by = ", ".join([self.to_string(x) for x in arg.by])
+            return f"{self.to_string(arg.function)} by {by}"
+        return f"{self.to_string(arg.function)}"
+
+    @to_string.register
+    def _(self, arg: MergeStatementV2):
+        if len(arg.sources) == 1:
+            return f"MERGE {self.to_string(arg.sources[0])} into {''.join([self.to_string(modifier) for modifier in arg.modifiers])}{self.to_string(arg.targets[arg.sources[0].address])};"
+        return f"MERGE {arg.source_wildcard}.* into {''.join([self.to_string(modifier) for modifier in arg.modifiers])}{arg.target_wildcard}.*;"
+
+    @to_string.register
+    def _(self, arg: KeyMergeStatement):
+        keys = ", ".join(sorted(list(arg.keys)))
+        return f"MERGE PROPERTY <{keys}> from {arg.target.address};"
+
+    @to_string.register
+    def _(self, arg: Modifier):
+        if arg == Modifier.PARTIAL:
+            return "~"
+        elif arg == Modifier.HIDDEN:
+            return "--"
+        elif arg == Modifier.NULLABLE:
+            return "?"
+        return arg.value
+
+    @to_string.register
+    def _(self, arg: int):
+        return f"{arg}"
+
+    @to_string.register
+    def _(self, arg: str):
+        return f"'{arg}'"
+
+    @to_string.register
+    def _(self, arg: float):
+        return f"{arg}"
+
+    @to_string.register
+    def _(self, arg: bool):
+        return f"{arg}"
+
+    @to_string.register
+    def _(self, arg: list):
+        base = ", ".join([self.to_string(x) for x in arg])
+        return f"[{base}]"
+
+
+def render_query(query: "SelectStatement") -> str:
+    return Renderer().to_string(query)
+
+
+def render_environment(environment: "Environment") -> str:
+    return Renderer().to_string(environment)