teaql 0.2.0__tar.gz → 0.2.2__tar.gz

{teaql-0.2.0 → teaql-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: teaql
-Version: 0.2.0
+Version: 0.2.2
 Summary: TeaQL Runtime and Data Service for Python
 Author: TeaQL
 License-Expression: Apache-2.0

{teaql-0.2.0 → teaql-0.2.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "teaql"
-version = "0.2.0"
+version = "0.2.2"
 description = "TeaQL Runtime and Data Service for Python"
 readme = "README.md"
 requires-python = ">=3.10"

{teaql-0.2.0 → teaql-0.2.2}/src/teaql/runtime/__init__.py

@@ -1,4 +1,7 @@
-from .context import ContextEntityRef, ContextRootError, UserContext, TeaqlRuntime, SqlLogEntry, SqlLogOperation
+from .context import (
+    ContextEntityRef, ContextRootError, DiagnosticSqlLogSink, UserContext,
+    TeaqlRuntime, SqlLogEntry, SqlLogOperation, TextDiagnosticSqlLogSink,
+)
 from .tools import (
     ContextTools, ExecutableHttpTool, HTTP_TOOL, HttpIntentPhase, HttpTool,
     HttpToolProvider, ToolDeniedError, ToolError, ToolPolicy, ToolRisk,
@@ -10,7 +13,7 @@ from .audit import RawAuditEvent, SafeAuditEvent, MutationAuditKind
 from .i18n import CheckException, CheckResult, I18nCatalog, Locale, ObjectLocation, UnsupportedLocaleError
 from teaql.core.entity import EntityKey, EntityChangeSet, EntityRoot
 
-__all__ = ["EntityKey", "EntityChangeSet", "EntityRoot", "ContextEntityRef", "ContextRootError", "CheckException", "CheckResult", "I18nCatalog", "Locale", "ObjectLocation", "UnsupportedLocaleError", "UserContext", "TeaqlRuntime", "SqlLogEntry", "SqlLogOperation", "ServiceRuntimeFromEnv", "RuntimeModule", "DataStore", "RawAuditEvent", "SafeAuditEvent", "MutationAuditKind", "ContextTools", "ExecutableHttpTool", "HTTP_TOOL", "HttpIntentPhase", "HttpTool", "HttpToolProvider", "ToolDeniedError", "ToolError", "ToolPolicy", "ToolRisk", "Tools", "ToolToken", "ToolUnavailableError"]
+__all__ = ["EntityKey", "EntityChangeSet", "EntityRoot", "ContextEntityRef", "ContextRootError", "CheckException", "CheckResult", "I18nCatalog", "Locale", "ObjectLocation", "UnsupportedLocaleError", "UserContext", "TeaqlRuntime", "SqlLogEntry", "SqlLogOperation", "DiagnosticSqlLogSink", "TextDiagnosticSqlLogSink", "ServiceRuntimeFromEnv", "RuntimeModule", "DataStore", "RawAuditEvent", "SafeAuditEvent", "MutationAuditKind", "ContextTools", "ExecutableHttpTool", "HTTP_TOOL", "HttpIntentPhase", "HttpTool", "HttpToolProvider", "ToolDeniedError", "ToolError", "ToolPolicy", "ToolRisk", "Tools", "ToolToken", "ToolUnavailableError"]
 
 
 def __getattr__(name):

{teaql-0.2.0 → teaql-0.2.2}/src/teaql/runtime/context.py

@@ -1,4 +1,6 @@
 from typing import Dict, Any, Optional, List, Callable, TypeVar
+import asyncio
+import contextvars
 from dataclasses import dataclass
 from array import array
 
@@ -11,6 +13,13 @@ class ContextEntityRef:
     entity: str
     id: int
 
+@dataclass(frozen=True)
+class FixEvidence:
+    entity_type: str
+    model_path: str
+    source: str
+    source_label: str
+
 class ContextRootError(Exception):
     def __init__(self, reason: str, expected_type: str, active_root: Optional[ContextEntityRef] = None):
         self.reason = reason
@@ -84,6 +93,35 @@ class UserContext:
         self._id_set_plan = "ID_SET_DISABLED"
         self._id_set_count = 0
         self._id_set_count_accuracy = "UNKNOWN"
+        self._graph_save_active = False
+        self._graph_save_lock = asyncio.Lock()
+        self._graph_save_owner = contextvars.ContextVar(
+            f"teaql_graph_save_owner_{id(self)}", default=None
+        )
+        self._graph_commit_actions: List[Any] = []
+        self._graph_rollback_actions: List[Any] = []
+        self._fix_evidence_current: List[FixEvidence] = []
+        self._fix_evidence_last: List[FixEvidence] = []
+
+    def begin_fix_evidence(self):
+        self._fix_evidence_current = []
+        return self
+
+    def record_fix_evidence(self, entity_type: str, model_path: str, source: str, source_label: str):
+        normalized = source_label.lower()
+        if not entity_type or not model_path or source not in ("clock", "context") or not source_label \
+                or "authorization" in normalized or "cookie" in normalized or "token=" in normalized:
+            raise ValueError("Fix evidence must contain only safe framework provenance labels")
+        self._fix_evidence_current.append(FixEvidence(entity_type, model_path, source, source_label))
+        return self
+
+    def finish_fix_evidence(self):
+        self._fix_evidence_last = list(self._fix_evidence_current)
+        self._fix_evidence_current = []
+        return self
+
+    def last_fix_evidence(self):
+        return tuple(self._fix_evidence_last)
 
     @classmethod
     def new(cls) -> 'UserContext':
@@ -103,6 +141,74 @@ class UserContext:
             raise ContextRootError("type_mismatch", expected_type, root)
         return root
 
+    async def execute_graph_save(self, work):
+        """Run one generated entity graph in one provider transaction."""
+        if self._graph_save_owner.get() is not None:
+            return await work()
+        async with self._graph_save_lock:
+            provider = self.require_resource("dataService")
+            begin = getattr(provider, "begin", None)
+            if not callable(begin):
+                raise RuntimeError("Configured dataService does not support graph transactions")
+            try:
+                transaction = await begin(self)
+            except TypeError:
+                transaction = await begin()
+            owner_token = self._graph_save_owner.set(object())
+            self._graph_save_active = True
+            self._graph_commit_actions = []
+            self._graph_rollback_actions = []
+            self.insert_resource("fix_time", datetime.now())
+            self.begin_fix_evidence()
+            self.insert_resource("dataService", transaction)
+            try:
+                result = await work()
+            except BaseException:
+                try:
+                    await self._finish_graph_transaction(transaction, "rollback")
+                finally:
+                    for action in reversed(self._graph_rollback_actions):
+                        action()
+                raise
+            else:
+                try:
+                    await self._finish_graph_transaction(transaction, "commit")
+                except BaseException:
+                    try:
+                        await self._finish_graph_transaction(transaction, "rollback")
+                    finally:
+                        for action in reversed(self._graph_rollback_actions):
+                            action()
+                    raise
+                for action in self._graph_commit_actions:
+                    action()
+                return result
+            finally:
+                self.insert_resource("dataService", provider)
+                self._graph_save_active = False
+                self._graph_commit_actions = []
+                self._graph_rollback_actions = []
+                self._resources.pop("fix_time", None)
+                self.finish_fix_evidence()
+                self._graph_save_owner.reset(owner_token)
+
+    async def _finish_graph_transaction(self, transaction, operation: str) -> None:
+        finish = getattr(transaction, operation)
+        try:
+            await finish(self)
+        except TypeError:
+            await finish()
+
+    def after_graph_commit(self, work) -> None:
+        if not self._graph_save_active:
+            raise RuntimeError("No graph save is active")
+        self._graph_commit_actions.append(work)
+
+    def after_graph_rollback(self, work) -> None:
+        if not self._graph_save_active:
+            raise RuntimeError("No graph save is active")
+        self._graph_rollback_actions.append(work)
+
     async def get_in_store(self, key: str) -> Optional[Any]:
         store = self.get_resource("data_store")
         if store and hasattr(store, "get"):
@@ -454,12 +560,17 @@ class UserContext:
         if not entity or record is None:
             return
         from datetime import datetime
-        self.insert_resource("fix_time", datetime.now())
+        owns_fix_time = self.get_resource("fix_time") is None
+        if owns_fix_time:
+            self.insert_resource("fix_time", datetime.now())
+            self.begin_fix_evidence()
         self.insert_resource("fix_operation", type(mutation).__name__.replace("Command", "").lower())
         try:
             self.check_and_fix_record(entity, record)
         finally:
-            self._resources.pop("fix_time", None)
+            if owns_fix_time:
+                self._resources.pop("fix_time", None)
+                self.finish_fix_evidence()
             self._resources.pop("fix_operation", None)
 
     def translate_check_results(self, results: Any):
@@ -507,10 +618,17 @@ class UserContext:
     def sql_log_options(self) -> 'SqlLogOptions':
         opts = self.get_resource("sql_log_options")
         if not opts:
-            opts = SqlLogOptions.all()
+            opts = SqlLogOptions.disabled()
             self.insert_resource("sql_log_options", opts)
         return opts
 
+    def with_diagnostic_sql_log_sink(self, sink: 'DiagnosticSqlLogSink') -> 'UserContext':
+        self.insert_resource("diagnostic_sql_log_sink", sink)
+        return self
+
+    def set_diagnostic_sql_log_sink(self, sink: Optional['DiagnosticSqlLogSink']):
+        self.insert_resource("diagnostic_sql_log_sink", sink)
+
     def enable_select_sql_log(self):
         self.set_sql_log_options(SqlLogOptions.select_only())
         self.clear_sql_logs()
@@ -581,6 +699,9 @@ class UserContext:
         logs = self.sql_logs()
         logs.append(entry)
         self._resources["sql_logs"] = logs
+        sink = self.get_resource("diagnostic_sql_log_sink")
+        if sink is not None:
+            sink.write(entry)
         buf = self.get_resource("UnifiedLogBuffer")
         if buf:
             buf.entries.append(UnifiedLogEntry(
@@ -619,6 +740,9 @@ class UserContext:
         logs = self.sql_logs()
         logs.append(entry)
         self._resources["sql_logs"] = logs
+        sink = self.get_resource("diagnostic_sql_log_sink")
+        if sink is not None:
+            sink.write(entry)
         
         buf = self.get_resource("UnifiedLogBuffer")
         if buf:
@@ -857,6 +981,22 @@ class SqlLogEntry:
     affected_rows: Optional[int]
     result_summary: str
 
+class DiagnosticSqlLogSink:
+    """Explicit value-bearing diagnostic SQL destination; never installed by default."""
+    def write(self, entry: SqlLogEntry) -> None:
+        raise NotImplementedError
+
+class TextDiagnosticSqlLogSink(DiagnosticSqlLogSink):
+    def __init__(self, writer=print):
+        self._writer = writer
+
+    def write(self, entry: SqlLogEntry) -> None:
+        elapsed_us = int(entry.elapsed.total_seconds() * 1_000_000) if entry.elapsed else 0
+        self._writer(
+            f"[TeaQL SQL][{entry.operation.name.lower()}][{elapsed_us}us] "
+            f"{entry.result_summary}\n{entry.debug_sql}"
+        )
+
 @dataclass
 class InfoLogEntry:
     message: str

{teaql-0.2.0 → teaql-0.2.2}/src/teaql/runtime/i18n.py

@@ -17,6 +17,9 @@ class ObjectLocation:
     def index(self, value: int) -> ObjectLocation:
         return ObjectLocation(self.segments + (("index", value),))
 
+    def prefixed_by(self, prefix: ObjectLocation) -> ObjectLocation:
+        return ObjectLocation(prefix.segments + self.segments)
+
     @builtins.property
     def model_path(self) -> str:
         return self._render(lambda value: value)

{teaql-0.2.0 → teaql-0.2.2}/src/teaql.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: teaql
-Version: 0.2.0
+Version: 0.2.2
 Summary: TeaQL Runtime and Data Service for Python
 Author: TeaQL
 License-Expression: Apache-2.0