pytrilogy 0.0.3.11__tar.gz → 0.0.3.12__tar.gz

{pytrilogy-0.0.3.11/pytrilogy.egg-info → pytrilogy-0.0.3.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: pytrilogy
-Version: 0.0.3.11
+Version: 0.0.3.12
 Summary: Declarative, typed query language that compiles to SQL.
 Home-page: 
 Author: 

{pytrilogy-0.0.3.11 → pytrilogy-0.0.3.12/pytrilogy.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: pytrilogy
-Version: 0.0.3.11
+Version: 0.0.3.12
 Summary: Declarative, typed query language that compiles to SQL.
 Home-page: 
 Author: 

{pytrilogy-0.0.3.11 → pytrilogy-0.0.3.12}/tests/test_query_render.py

@@ -10,4 +10,4 @@ def test_sql_generators():
     for dialect in Dialects:
         generator = get_dialect_generator(dialect)
         compiled = generator.compile_statement(processed)
-        assert compiled.startswith("""SELECT\n    :a"""), f'{dialect} compiled'
+        assert compiled.startswith("""SELECT\n    :a"""), f"{dialect} compiled"

{pytrilogy-0.0.3.11 → pytrilogy-0.0.3.12}/trilogy/__init__.py

@@ -4,6 +4,6 @@ from trilogy.dialect.enums import Dialects
 from trilogy.executor import Executor
 from trilogy.parser import parse
 
-__version__ = "0.0.3.11"
+__version__ = "0.0.3.12"
 
 __all__ = ["parse", "Executor", "Dialects", "Environment", "CONFIG"]

{pytrilogy-0.0.3.11 → pytrilogy-0.0.3.12}/trilogy/engine.py

@@ -18,6 +18,15 @@ class EngineConnection(Protocol):
     def execute(self, statement: str, parameters: Any | None = None) -> EngineResult:
         pass
 
+    def commit(self):
+        raise NotImplementedError()
+
+    def begin(self):
+        raise NotImplementedError()
+
+    def rollback(self):
+        raise NotImplementedError()
+
 
 class ExecutionEngine(Protocol):
     pass
@@ -40,13 +49,24 @@ class SqlAlchemyResult(EngineResult):
 
 class SqlAlchemyConnection(EngineConnection):
     def __init__(self, connection: Connection):
-        self.connection = connection
+        from sqlalchemy.future import Connection
+
+        self.connection: Connection = connection
 
     def execute(
         self, statement: str, parameters: Any | None = None
     ) -> SqlAlchemyResult:
         return SqlAlchemyResult(self.connection.execute(statement, parameters))
 
+    def commit(self):
+        self.connection.commit()
+
+    def begin(self):
+        self.connection.begin()
+
+    def rollback(self):
+        self.connection.rollback()
+
 
 class SqlAlchemyEngine(ExecutionEngine):
     def __init__(self, engine: Engine):