sqlsaber-viz 0.1.1__py3-none-any.whl

sqlsaber_viz/transforms.py

@@ -0,0 +1,155 @@
+"""Transform pipeline for visualization data."""
+
+from __future__ import annotations
+
+import re
+from datetime import datetime
+
+from .spec import (
+    FilterConfig,
+    FilterTransform,
+    LimitTransform,
+    SortItem,
+    SortTransform,
+    Transform,
+)
+
+
+def apply_transforms(rows: list[dict], transforms: list[Transform]) -> list[dict]:
+    """Apply transform pipeline to rows."""
+    result = list(rows)
+    for transform in transforms:
+        if isinstance(transform, SortTransform):
+            result = apply_sort(result, transform.sort)
+        elif isinstance(transform, LimitTransform):
+            result = result[: transform.limit]
+        elif isinstance(transform, FilterTransform):
+            result = apply_filter(result, transform.filter)
+    return result
+
+
+def apply_sort(rows: list[dict], sorts: list[SortItem]) -> list[dict]:
+    """Sort rows by multiple fields."""
+    result = list(rows)
+    for sort in reversed(sorts):
+        field = sort.field
+        descending = sort.dir == "desc"
+        result = sorted(result, key=lambda row: _sort_key(row.get(field)))
+        if descending:
+            result.reverse()
+        # Always push None/missing values to the end.
+        non_none = [row for row in result if row.get(field) is not None]
+        none_rows = [row for row in result if row.get(field) is None]
+        result = non_none + none_rows
+    return result
+
+
+def apply_filter(rows: list[dict], filter_config: FilterConfig) -> list[dict]:
+    """Filter rows by condition."""
+    field = filter_config.field
+    op = filter_config.op
+    target = filter_config.value
+
+    filtered: list[dict] = []
+    for row in rows:
+        value = row.get(field)
+        if _compare(value, op, target):
+            filtered.append(row)
+    return filtered
+
+
+def _sort_key(value: object) -> tuple[int, object]:
+    if value is None:
+        return (3, "")
+
+    numeric = _coerce_number(value)
+    if numeric is not None:
+        return (0, numeric)
+
+    time_val = _coerce_time(value)
+    if time_val is not None:
+        return (1, time_val)
+
+    return (2, str(value).lower())
+
+
+def _compare(value: object, op: str, target: object) -> bool:
+    if op in ("==", "!="):
+        result = _equals(value, target)
+        return result if op == "==" else not result
+
+    left_num = _coerce_number(value)
+    right_num = _coerce_number(target)
+    if left_num is not None and right_num is not None:
+        return _compare_ordered(left_num, right_num, op)
+
+    left_time = _coerce_time(value)
+    right_time = _coerce_time(target)
+    if left_time is not None and right_time is not None:
+        return _compare_ordered(left_time, right_time, op)
+
+    return False
+
+
+def _equals(value: object, target: object) -> bool:
+    if value is None or target is None:
+        return value is target
+
+    left_num = _coerce_number(value)
+    right_num = _coerce_number(target)
+    if left_num is not None and right_num is not None:
+        return left_num == right_num
+
+    left_time = _coerce_time(value)
+    right_time = _coerce_time(target)
+    if left_time is not None and right_time is not None:
+        return left_time == right_time
+
+    return value == target
+
+
+def _compare_ordered(left: object, right: object, op: str) -> bool:
+    if op == ">":
+        return left > right
+    if op == "<":
+        return left < right
+    if op == ">=":
+        return left >= right
+    if op == "<=":
+        return left <= right
+    return False
+
+
+def _coerce_number(value: object) -> float | None:
+    if isinstance(value, bool):
+        return None
+    if isinstance(value, (int, float)):
+        return float(value)
+    if isinstance(value, str):
+        try:
+            return float(value)
+        except ValueError:
+            return None
+    return None
+
+
+def _coerce_time(value: object) -> datetime | None:
+    if isinstance(value, datetime):
+        return value
+    if isinstance(value, str):
+        # Handle Z suffix (e.g., "2024-01-01T00:00:00Z")
+        normalized = value
+        if value.endswith("Z"):
+            normalized = value[:-1] + "+00:00"
+        try:
+            return datetime.fromisoformat(normalized)
+        except ValueError:
+            pass
+        # Try YYYY-MM format (e.g., "2023-06")
+        if re.match(r"^\d{4}-\d{2}$", value):
+            try:
+                return datetime.fromisoformat(f"{value}-01")
+            except ValueError:
+                pass
+        return None
+    return None

sqlsaber_viz-0.1.1.dist-info/METADATA

@@ -0,0 +1,12 @@
+Metadata-Version: 2.4
+Name: sqlsaber-viz
+Version: 0.1.1
+Summary: SQLsaber terminal visualization plugin
+Requires-Python: >=3.12
+Requires-Dist: plotext>=5.3.0
+Requires-Dist: sqlsaber>=0.54.0
+Description-Content-Type: text/markdown
+
+# SQLSaber Visualization Plugin
+
+Provides the `viz` tool for SQLSaber via the `sqlsaber-viz` plugin. The tool generates a declarative chart spec and renders ASCII charts in the terminal using plotext.

sqlsaber_viz-0.1.1.dist-info/RECORD

@@ -0,0 +1,16 @@
+sqlsaber_viz/__init__.py,sha256=HdXfKzS8O7f1qD1nQo9XBPIB5Y6NvIroWmlaulqdvDs,394
+sqlsaber_viz/data_loader.py,sha256=mTFGCd6GZP_9KK110gupKumJQYsg0QoHVGXeqBefYbY,4380
+sqlsaber_viz/prompts.py,sha256=UCoIH32qfv8FguL3T0L77l8n-DjSYR7q_FOrZg-4_3U,1508
+sqlsaber_viz/spec.py,sha256=_zgUBrHPrjztnyECRZycLps1KeEiLn29x9B0fSdqf10,2923
+sqlsaber_viz/spec_agent.py,sha256=LDH5gOd6zw-dno-0yNT7wKR7OnU4ILlH829cdzxOwXc,4694
+sqlsaber_viz/templates.py,sha256=8WDYYVIjEDmKGUo9VPuv_vOOoieqnqGLl39hQpGEWL0,5022
+sqlsaber_viz/tools.py,sha256=d6axy4Ac_TRrJJqqGr9u9k_-5vFHE5XZUVEjzy0jEIM,7756
+sqlsaber_viz/transforms.py,sha256=I_PxG4fy1ckdwC_9in3KVEa9cAQ9oLmq113eEGmvLEk,4440
+sqlsaber_viz/renderers/__init__.py,sha256=oIDE7QqaulKVN_yVKXqERuPskS-Pvxa2ILk234JbkIs,174
+sqlsaber_viz/renderers/base.py,sha256=V-8dCvgvE1AMpVlwOPRxnNwqumUbjlZmwMr-KpculyI,312
+sqlsaber_viz/renderers/html_renderer.py,sha256=f48JyfpcYIcnwCb_zmi9HlWflir94OOMuSVgi8PQhPA,367
+sqlsaber_viz/renderers/plotext_renderer.py,sha256=QFZbaFxvfEr32BCOFRq_fxtZYD_dhAJtbhkcr9VGzP8,13371
+sqlsaber_viz-0.1.1.dist-info/METADATA,sha256=LPq6Wg5944bec4J41-3YreZeUM3K7CPzLnoX2n5ri8M,432
+sqlsaber_viz-0.1.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+sqlsaber_viz-0.1.1.dist-info/entry_points.txt,sha256=us67FM9sQiA3XG-UxYbHI9GAKQB5Xpqe_STfQtKEWys,51
+sqlsaber_viz-0.1.1.dist-info/RECORD,,

sqlsaber_viz-0.1.1.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.28.0
+Root-Is-Purelib: true
+Tag: py3-none-any

sqlsaber_viz-0.1.1.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[sqlsaber.tools]
+viz = sqlsaber_viz:register_tools