flagsmith-sql-flag-engine 0.1.2__py3-none-any.whl → 0.2.0__py3-none-any.whl

flagsmith_sql_flag_engine/__init__.py

@@ -4,11 +4,22 @@ Public API:
     translate_segment(segment, ctx) -> str | None
     TranslateContext
 
+By default the translator inlines each segment value as an escaped SQL
+string literal. Pass a `Binder` on the `TranslateContext` to bind
+value-bearing literals as query parameters instead — read its params off
+`Binder.params` after translation. See `flagsmith_sql_flag_engine.binder`.
+
 See README.md for usage. The translator is dialect-aware via the `Dialect`
 protocol; `flagsmith_sql_flag_engine.dialects.clickhouse.ClickHouseDialect`
 is the only implementation today.
 """
 
+from flagsmith_sql_flag_engine.binder import (
+    Binder,
+    ClickHouseServerParamStyle,
+    ParamStyle,
+    PyformatParamStyle,
+)
 from flagsmith_sql_flag_engine.dialect import Dialect
 from flagsmith_sql_flag_engine.translator import (
     TRANSLATABLE_OPERATORS,
@@ -20,7 +31,11 @@ from flagsmith_sql_flag_engine.translator import (
 
 __all__ = [
     "TRANSLATABLE_OPERATORS",
+    "Binder",
+    "ClickHouseServerParamStyle",
     "Dialect",
+    "ParamStyle",
+    "PyformatParamStyle",
     "TranslateContext",
     "translate_condition",
     "translate_rule",

flagsmith_sql_flag_engine/binder.py

@@ -0,0 +1,50 @@
+from typing import Protocol
+
+
+class ParamStyle(Protocol):
+    """A driver's placeholder syntax for a named bound parameter."""
+
+    def placeholder(self, name: str) -> str:
+        """The placeholder token referencing bound parameter `name`."""
+        ...
+
+
+class PyformatParamStyle:
+    """`%(name)s`
+
+    Used by `clickhouse-driver` which substitutes parameters
+    client-side via `query % params`."""
+
+    def placeholder(self, name: str) -> str:
+        return f"%({name})s"
+
+
+class ClickHouseServerParamStyle:
+    """`{name:String}`
+
+    ClickHouse's native server-side parameter syntax,
+    used by `clickhouse-connect`."""
+
+    def placeholder(self, name: str) -> str:
+        return "{" + name + ":String}"
+
+
+class Binder:
+    """Collects bound parameter values and mints their placeholders.
+
+    Not thread-safe; use one `Binder` per predicate translation.
+    """
+
+    def __init__(self, style: ParamStyle, prefix: str = "") -> None:
+        self.params: dict[str, str] = {}
+        self._style = style
+        self._prefix = prefix
+        self._count = 0
+
+    def add(self, value: str) -> str:
+        """Record `value` under a fresh namespaced name and return its
+        placeholder token for the active paramstyle."""
+        name = f"{self._prefix}p{self._count}"
+        self._count += 1
+        self.params[name] = value
+        return self._style.placeholder(name)

flagsmith_sql_flag_engine/dialect.py

@@ -3,6 +3,8 @@ regex, padded-version comparison, type-aware trait predicates, regex flavour."""
 
 from typing import Protocol
 
+from flagsmith_sql_flag_engine.binder import Binder
+
 
 class Dialect(Protocol):
     """Per-dialect SQL fragments.
@@ -10,6 +12,10 @@ class Dialect(Protocol):
     Methods return SQL string fragments. Inputs are already-formatted SQL
     strings (column refs, string literals); the dialect only chooses the
     right syntax for the operation.
+
+    Methods that embed a segment- or context-derived value take an
+    optional `binder`: when provided, the value is emitted as a bound
+    query parameter rather than an inline literal.
     """
 
     name: str  # human-readable, used in test ids and error messages
@@ -35,7 +41,14 @@ class Dialect(Protocol):
         """
         ...
 
-    def trait_eq(self, alias: str, trait_key: str, value: object, negate: bool) -> str:
+    def trait_eq(
+        self,
+        alias: str,
+        trait_key: str,
+        value: object,
+        negate: bool,
+        binder: Binder | None = None,
+    ) -> str:
         """Type-aware EQUAL / NOT_EQUAL predicate on a trait, mirroring
         `flag_engine`'s per-type coercion: the segment value is cast to
         the trait's runtime type before compare, and a cast failure
@@ -45,7 +58,13 @@ class Dialect(Protocol):
         """
         ...
 
-    def trait_in(self, alias: str, trait_key: str, items: list[str]) -> str:
+    def trait_in(
+        self,
+        alias: str,
+        trait_key: str,
+        items: list[str],
+        binder: Binder | None = None,
+    ) -> str:
         """Type-aware IN predicate on a trait, mirroring engine semantics:
         string trait does direct lookup; integer trait stringifies and
         looks up; other trait types never match. `items` is the parsed
@@ -77,14 +96,19 @@ class Dialect(Protocol):
         to `flag_engine`."""
         ...
 
-    def regexp_anchored_match(self, value_expr: str, pattern: str) -> str:
+    def regexp_anchored_match(
+        self,
+        value_expr: str,
+        pattern: str,
+        binder: Binder | None = None,
+    ) -> str:
         """Boolean: equivalent to Python `re.match(pattern, value)` —
         anchored at position 0, may be a prefix of the value, not a
         full-match.
 
-        `pattern` is the raw Python regex string; the dialect handles
-        its own escaping into a SQL literal, since regex flavours
-        differ in how backslashes are treated."""
+        `pattern` is the raw Python regex string. With no `binder`, the
+        dialect handles its own escaping into a SQL literal, since regex
+        flavours differ in how backslashes are treated."""
         ...
 
     def regexp_nth_digit_run(self, value_expr: str, n: int) -> str:

flagsmith_sql_flag_engine/dialects/clickhouse.py

@@ -76,7 +76,8 @@ typical Flagsmith trait vocabularies; we should monitor.
 ClickHouse Cloud as of 25.12 (no longer experimental on OSS 25.x).
 Callers should apply this setting at session creation."""
 
-from flagsmith_sql_flag_engine.utils import re2_safe, string_literal
+from flagsmith_sql_flag_engine.binder import Binder
+from flagsmith_sql_flag_engine.utils import bind_or_inline, re2_safe
 
 SCHEMA_DDL = """\
 CREATE TABLE IF NOT EXISTS IDENTITIES (
@@ -154,10 +155,17 @@ class ClickHouseDialect:
         sub = self._sub(alias, trait_key)
         return f"if({sub} IS NULL, NULL, toString({sub}))"
 
-    def trait_eq(self, alias: str, trait_key: str, value: object, negate: bool) -> str:
+    def trait_eq(
+        self,
+        alias: str,
+        trait_key: str,
+        value: object,
+        negate: bool,
+        binder: Binder | None = None,
+    ) -> str:
         sub = self._sub(alias, trait_key)
         str_value = str(value)
-        str_lit = string_literal(str_value)
+        str_lit = bind_or_inline(binder, str_value)
         # Engine bool cast: `v not in ("False", "false")`. A JSON true matches
         # every segment value except literal "False" / "false"; those two coerce
         # to False and match a JSON false.
@@ -223,7 +231,13 @@ class ClickHouseDialect:
             f"(({str_sub} IS NOT NULL AND {str_sub} <> {str_lit}) OR {bool_branch} OR {num_branch})"
         )
 
-    def trait_in(self, alias: str, trait_key: str, items: list[str]) -> str:
+    def trait_in(
+        self,
+        alias: str,
+        trait_key: str,
+        items: list[str],
+        binder: Binder | None = None,
+    ) -> str:
         # `toString(<sub>)` returns the canonical string form for any JSON
         # value type in a single subcolumn read. Engine semantics only
         # match String and integer trait types — bool / float / array
@@ -235,7 +249,7 @@ class ClickHouseDialect:
         bool_sub = f"{sub}.:Bool"
         float_sub = f"{sub}.:Float64"
         str_path = f"toString({sub})"
-        item_lits = ",".join(string_literal(v) for v in items)
+        item_lits = ",".join(bind_or_inline(binder, v) for v in items)
         return f"({bool_sub} IS NULL AND {float_sub} IS NULL AND {str_path} IN ({item_lits}))"
 
     # ----- string operations -----
@@ -267,13 +281,21 @@ class ClickHouseDialect:
         doubled = pattern.replace("\\", "\\\\").replace("'", "''")
         return f"'{doubled}'"
 
-    def regexp_anchored_match(self, value_expr: str, pattern: str) -> str:
+    def regexp_anchored_match(
+        self, value_expr: str, pattern: str, binder: Binder | None = None
+    ) -> str:
         # `match` is RE2 but unanchored — equivalent to `re.search`. Prepend
         # `^` to get `re.match` semantics (start-anchored, prefix-allowed).
         # Wrapping in `(...)` keeps the user's top-level alternation from
         # binding tighter than the anchor.
         anchored = "^(" + pattern + ")"
-        return f"match({_non_null(value_expr)}, {self._regex_literal(anchored)})"
+        # Bind the raw pattern when a binder is active: the driver escapes
+        # it, and — crucially — no `%` from a character class like
+        # `[a-z%]` lands in the query text to trip a `%`-substituting
+        # driver. Inline, `_regex_literal` doubles backslashes so RE2 sees
+        # the pattern the segment author wrote.
+        pattern_lit = binder.add(anchored) if binder is not None else self._regex_literal(anchored)
+        return f"match({_non_null(value_expr)}, {pattern_lit})"
 
     def regexp_nth_digit_run(self, value_expr: str, n: int) -> str:
         # `extractAll` returns the matches array; subscript is 1-indexed

flagsmith_sql_flag_engine/translator.py

@@ -22,12 +22,12 @@ from flag_engine.context.types import (
 from flag_engine.segments.evaluator import is_context_in_segment
 from flag_engine.segments.types import ConditionOperator
 
+from flagsmith_sql_flag_engine.binder import Binder
 from flagsmith_sql_flag_engine.dialect import Dialect
 from flagsmith_sql_flag_engine.utils import (
-    escape_string,
+    bind_or_inline,
     modulo_literal,
     numeric_literal,
-    string_literal,
 )
 
 TRANSLATABLE_OPERATORS: frozenset[ConditionOperator] = frozenset(
@@ -75,7 +75,10 @@ class TranslateContext:
     being configured here. `identities_alias` is the table alias for
     `IDENTITIES` in the surrounding query — defaults to `i`.
     `segment_key` salts `PERCENTAGE_SPLIT` and is auto-injected from
-    the segment's `key` field by `translate_segment`.
+    the segment's `key` field by `translate_segment`. `binder`, when
+    provided, promotes value-bearing literals to bound query parameters
+    instead of inlining them — see `flagsmith_sql_flag_engine.binder`;
+    the default `None` inlines them as escaped SQL string literals.
     """
 
     def __init__(
@@ -84,11 +87,13 @@ class TranslateContext:
         dialect: Dialect,
         identities_alias: str = "i",
         segment_key: str | None = None,
+        binder: Binder | None = None,
     ) -> None:
         self.evaluation_context = evaluation_context
         self.dialect = dialect
         self.identities_alias = identities_alias
         self.segment_key = segment_key
+        self.binder = binder
 
     @property
     def identity_key_expr(self) -> str:
@@ -114,6 +119,7 @@ class TranslateContext:
             dialect=self.dialect,
             identities_alias=self.identities_alias,
             segment_key=key,
+            binder=self.binder,
         )
 
 
@@ -133,7 +139,7 @@ def _percentage_split_expr(
     the engine recurses with doubled input; we don't.
     """
     d = ctx.dialect
-    seg_lit = string_literal(seg_key)
+    seg_lit = bind_or_inline(ctx.binder, seg_key)
     hash_subject = f"{seg_lit} || ',' || ({ctx_value_sql})"
     h = d.md5_hex(hash_subject)
     s1 = d.parse_hex_chunk(h, 1)
@@ -289,18 +295,19 @@ def _comparison(
     if value is None:
         return "FALSE"
     d = ctx.dialect
-    lit = string_literal(str(value))
     str_expr = expr if is_jsonpath else d.cast_string(expr)
+    # Bind the operand lazily
     if op == "EQUAL":
-        return f"{str_expr} = {lit}"
+        return f"{str_expr} = {bind_or_inline(ctx.binder, str(value))}"
     if op == "NOT_EQUAL":
-        return f"{str_expr} <> {lit}"
+        return f"{str_expr} <> {bind_or_inline(ctx.binder, str(value))}"
     if op == "IN":
-        items = "','".join(escape_string(v.strip()) for v in str(value).split(","))
-        return f"{str_expr} IN ('{items}')"
+        items = ",".join(bind_or_inline(ctx.binder, v.strip()) for v in str(value).split(","))
+        return f"{str_expr} IN ({items})"
     if op == "CONTAINS":
-        return d.position(lit, str_expr)
+        return d.position(bind_or_inline(ctx.binder, str(value)), str_expr)
     if op == "NOT_CONTAINS":
+        lit = bind_or_inline(ctx.binder, str(value))
         return f"({expr} IS NOT NULL AND NOT ({d.position(lit, str_expr)}))"
     if op in {"GREATER_THAN", "LESS_THAN", "GREATER_THAN_INCLUSIVE", "LESS_THAN_INCLUSIVE"}:
         numeric_lit = numeric_literal(value)
@@ -327,7 +334,7 @@ def _comparison(
         pattern = str(value)
         if not d.regex_supports(pattern):
             return None
-        return f"({expr} IS NOT NULL AND {d.regexp_anchored_match(str_expr, pattern)})"
+        return f"({expr} IS NOT NULL AND {d.regexp_anchored_match(str_expr, pattern, ctx.binder)})"
     raise AssertionError(  # pragma: no cover - all TRANSLATABLE_OPERATORS handled above
         f"unhandled translatable operator in _comparison: {op}"
     )
@@ -369,7 +376,7 @@ def _translate_trait_op(
     # content, which is what the fall-through handlers already do.
     if isinstance(val, str) and val.endswith(":semver") and op in _SEMVER_OPS:
         bare = val[:-7]
-        bare_lit = string_literal(bare)
+        bare_lit = bind_or_inline(ctx.binder, bare)
         col_str = ctx.dialect.cast_string(path)
         return (
             f"({path} IS NOT NULL AND "
@@ -382,7 +389,13 @@ def _translate_trait_op(
     # casts, and short-circuit pitfalls are all engine-specific.
     if op in {"EQUAL", "NOT_EQUAL"} and val is not None:
         negate = op == "NOT_EQUAL"
-        eq_pred = ctx.dialect.trait_eq(ctx.identities_alias, trait_key, val, negate=negate)
+        eq_pred = ctx.dialect.trait_eq(
+            ctx.identities_alias,
+            trait_key,
+            val,
+            negate=negate,
+            binder=ctx.binder,
+        )
         return f"({path} IS NOT NULL AND {eq_pred})"
     if op == "IN":
         items = _engine_in_values(val)
@@ -390,7 +403,12 @@ def _translate_trait_op(
             # Bad IN value — neither a string nor a list. Engine returns
             # False.
             return "FALSE"
-        in_pred = ctx.dialect.trait_in(ctx.identities_alias, trait_key, items)
+        in_pred = ctx.dialect.trait_in(
+            ctx.identities_alias,
+            trait_key,
+            items,
+            binder=ctx.binder,
+        )
         return f"({path} IS NOT NULL AND {in_pred})"
 
     return _comparison(ctx, op, path, val, is_jsonpath=False)
@@ -432,7 +450,6 @@ def translate_condition(cond: SegmentCondition, ctx: TranslateContext) -> str |
             # Engine: float() on the threshold raises → returns False.
             return "FALSE"
         threshold = float(threshold_lit)
-        identity: dict[str, object] = ctx.evaluation_context.get("identity") or {}  # type: ignore[assignment]
         kind = classification.kind
         if not prop:
             # In traditional engine implementations, this branch implies
@@ -457,13 +474,18 @@ def translate_condition(cond: SegmentCondition, ctx: TranslateContext) -> str |
             # future work and let the caller fall back to the engine.
             return None
         else:
-            # Plain trait key, or `$.identity.traits.<X>` rewritten to
-            # the bare key. Hash subject pulls from `i.traits:"<key>"`
-            # per row.
-            traits = identity.get("traits") or {}
-            if not isinstance(traits, dict) or prop not in traits:
-                return "FALSE"
-            value_expr = ctx.dialect.cast_string(ctx.trait_path(prop))
+            # Plain trait key, or `$.identity.traits.<X>` rewritten to the
+            # bare key. Hash the per-row trait value from `i.traits:"<key>"`.
+            # A row missing the trait has no value to bucket, so it cannot
+            # match — mirroring the engine's `context_value is None -> False`.
+            trait_sql = ctx.trait_path(prop)
+            split = _percentage_split_expr(
+                ctx,
+                ctx.segment_key,
+                ctx.dialect.cast_string(trait_sql),
+                threshold,
+            )
+            return f"({trait_sql} IS NOT NULL AND {split})"
         return _percentage_split_expr(ctx, ctx.segment_key, value_expr, threshold)
 
     if not prop:

flagsmith_sql_flag_engine/utils.py

@@ -21,6 +21,21 @@ on the `Dialect` protocol instead.
 """
 
 import re
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from flagsmith_sql_flag_engine.binder import Binder
+
+
+def bind_or_inline(binder: "Binder | None", value: str) -> str:
+    """Return a bound-parameter placeholder for `value` when a `binder` is
+    active, else a single-quoted SQL string literal.
+
+    The single seam that lets a value-bearing literal either be inlined
+    (the default) or promoted to a query parameter, chosen per call site
+    by whether the caller threaded a `Binder` onto the context.
+    """
+    return binder.add(value) if binder is not None else string_literal(value)
 
 
 def escape_string(value: str) -> str:

{flagsmith_sql_flag_engine-0.1.2.dist-info → flagsmith_sql_flag_engine-0.2.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flagsmith-sql-flag-engine
-Version: 0.1.2
+Version: 0.2.0
 Summary: SQL translator for Flagsmith segment predicates.
 Author: Flagsmith
 Author-email: Flagsmith <engineering@flagsmith.com>
@@ -66,6 +66,36 @@ translator can't handle — typically a REGEX pattern the active dialect's
 regex flavour can't compile. Callers should fall back to
 `flag_engine.is_context_in_segment` for those segments.
 
+## Bound parameters
+
+By default the translator inlines each segment value as an escaped SQL string literal. Pass a `Binder` on the `TranslateContext` to bind value-bearing literals as query parameters instead. 
+
+```python
+from flagsmith_sql_flag_engine import (
+    Binder,
+    PyformatParamStyle,
+    TranslateContext,
+    translate_segment,
+)
+from flagsmith_sql_flag_engine.dialects import ClickHouseDialect
+
+binder = Binder(PyformatParamStyle())
+ctx = TranslateContext(
+    evaluation_context=eval_context,
+    dialect=ClickHouseDialect(),
+    binder=binder,
+)
+where_expr = translate_segment(segment, ctx)
+```
+
+Hand both to the driver:
+
+```python
+   cursor.execute(f"... WHERE ({where_expr})", binder.params)
+```
+
+Currently, `%`-prefixed style `PyformatParamStyle` and ClickHouse-specific `ClickHouseServerParamStyle` are supported.
+
 ## Schema
 
 Each dialect publishes the table layout it expects via a `schema_ddl`

flagsmith_sql_flag_engine-0.2.0.dist-info/RECORD

@@ -0,0 +1,11 @@
+flagsmith_sql_flag_engine/__init__.py,sha256=gsyNqBdTyELjV-xdfMYfVTAIDcS9LE1jiv1Yt4JWFFw,1223
+flagsmith_sql_flag_engine/binder.py,sha256=eSb_OiHIN_ZmZcz7I0cGurm1vn2XmMeljB6w6yNBrEM,1386
+flagsmith_sql_flag_engine/dialect.py,sha256=xofskXKrVMFzv73_vVzNsygUH9uvfKRq2VnZ1KTVZ9Y,5005
+flagsmith_sql_flag_engine/dialects/__init__.py,sha256=4pkblw-Jr04CLnSbpMud5Vyj5alBgB6WhSm0Cz9OWSI,141
+flagsmith_sql_flag_engine/dialects/clickhouse.py,sha256=dcrfyOQ4XNlmsFLTWOjp47GtOmoV_u4zXQ2LFAYSoYg,16218
+flagsmith_sql_flag_engine/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+flagsmith_sql_flag_engine/translator.py,sha256=_oTYHAeaxNiWC9ew0FjI_cg5tGTGe-4A7vVXn3bx6zE,23833
+flagsmith_sql_flag_engine/utils.py,sha256=FEh_W5_2BLCdJaucgIm3s4YaCS_H4JU7Q19Fj4c7VVk,3800
+flagsmith_sql_flag_engine-0.2.0.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
+flagsmith_sql_flag_engine-0.2.0.dist-info/METADATA,sha256=TSMcscM4_yC8XPbkOf76EDxnxVntEp91lZJphZLcqAk,6974
+flagsmith_sql_flag_engine-0.2.0.dist-info/RECORD,,

flagsmith_sql_flag_engine-0.1.2.dist-info/RECORD

@@ -1,10 +0,0 @@
-flagsmith_sql_flag_engine/__init__.py,sha256=DgiUBg8KnfdisZiMNsWpzKYtDqJoLvxnKctXQGESdZg,710
-flagsmith_sql_flag_engine/dialect.py,sha256=G4rzXszXUVM8iV-r0w_YJgFAWtqWx1hfJI48P9zuTmE,4514
-flagsmith_sql_flag_engine/dialects/__init__.py,sha256=4pkblw-Jr04CLnSbpMud5Vyj5alBgB6WhSm0Cz9OWSI,141
-flagsmith_sql_flag_engine/dialects/clickhouse.py,sha256=Flwy2QNEv-A53rvZoaWv8Q0wytuO8NY4IaRogGr0Ztg,15523
-flagsmith_sql_flag_engine/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-flagsmith_sql_flag_engine/translator.py,sha256=UfqttdhK3ew7mV2PbLIiKEjzL2YAUOBYjhH9wzCl2oQ,22999
-flagsmith_sql_flag_engine/utils.py,sha256=ygP8cijp1jQjszFj87odK95jdfV3DwaT02eXrQdikAA,3202
-flagsmith_sql_flag_engine-0.1.2.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
-flagsmith_sql_flag_engine-0.1.2.dist-info/METADATA,sha256=xcP1ByC00k1Y0pHxlQbCoKttWvcFua-5v2kNdhGPAF8,6137
-flagsmith_sql_flag_engine-0.1.2.dist-info/RECORD,,