ydb-sqlalchemy 0.1.16__py2.py3-none-any.whl → 0.1.17__py2.py3-none-any.whl

ydb_sqlalchemy/_version.py

@@ -1 +1 @@
-VERSION = "0.1.16"
+VERSION = "0.1.17"

ydb_sqlalchemy/sqlalchemy/__init__.py

@@ -206,6 +206,7 @@ class YqlDialect(StrCompileDialect):
         json_serializer=None,
         json_deserializer=None,
         _add_declare_for_yql_stmt_vars=False,
+        _statement_prefixes_list=None,
         **kwargs,
     ):
         super().__init__(**kwargs)
@@ -215,6 +216,7 @@ class YqlDialect(StrCompileDialect):
         # NOTE: _add_declare_for_yql_stmt_vars is temporary and is soon to be removed.
         # no need in declare in yql statement here since ydb 24-1
         self._add_declare_for_yql_stmt_vars = _add_declare_for_yql_stmt_vars
+        self._statement_prefixes = tuple(_statement_prefixes_list) if _statement_prefixes_list else ()
 
     def _describe_table(self, connection, table_name, schema=None) -> ydb.TableDescription:
         if schema is not None:
@@ -405,6 +407,12 @@ class YqlDialect(StrCompileDialect):
         )
         return f"{declarations}\n{statement}"
 
+    def _apply_statement_prefixes_impl(self, statement: str) -> str:
+        if not self._statement_prefixes:
+            return statement
+        prefixes = "\n".join(self._statement_prefixes) + "\n"
+        return f"{prefixes}{statement}"
+
     def __merge_parameters_values_and_types(
         self, values: Mapping[str, Any], types: Mapping[str, Any], execute_many: bool
     ) -> Sequence[Mapping[str, ydb.TypedValue]]:
@@ -438,9 +446,12 @@ class YqlDialect(StrCompileDialect):
             statement, parameters = self._format_variables(statement, parameters, execute_many)
             if self._add_declare_for_yql_stmt_vars:
                 statement = self._add_declare_for_yql_stmt_vars_impl(statement, parameters_types)
+            statement = self._apply_statement_prefixes_impl(statement)
             return statement, parameters
 
         statement, parameters = self._format_variables(statement, parameters, execute_many)
+        if not is_ddl:
+            statement = self._apply_statement_prefixes_impl(statement)
         return statement, parameters
 
     def do_ping(self, dbapi_connection: ydb_dbapi.Connection) -> bool:

ydb_sqlalchemy/sqlalchemy/test_sqlalchemy.py

@@ -114,6 +114,30 @@ def test_types_compilation():
     assert compile_type(struct) == "Struct<a:Int32,b:List<Int32>>"
 
 
+def test_statement_prefixes_prepended_to_query():
+    dialect = YqlDialect(_statement_prefixes_list=["PRAGMA DistinctOverKeys;"])
+    result = dialect._apply_statement_prefixes_impl("SELECT 1")
+    assert result == "PRAGMA DistinctOverKeys;\nSELECT 1"
+
+
+def test_statement_prefixes_empty_list_unchanged():
+    dialect = YqlDialect(_statement_prefixes_list=[])
+    result = dialect._apply_statement_prefixes_impl("SELECT 1")
+    assert result == "SELECT 1"
+
+
+def test_statement_prefixes_none_unchanged():
+    dialect = YqlDialect()
+    result = dialect._apply_statement_prefixes_impl("SELECT 1")
+    assert result == "SELECT 1"
+
+
+def test_statement_prefixes_multiple():
+    dialect = YqlDialect(_statement_prefixes_list=["PRAGMA Foo;", "PRAGMA Bar;"])
+    result = dialect._apply_statement_prefixes_impl("SELECT 1")
+    assert result == "PRAGMA Foo;\nPRAGMA Bar;\nSELECT 1"
+
+
 def test_optional_type_compilation():
     dialect = YqlDialect()
     type_compiler = dialect.type_compiler

{ydb_sqlalchemy-0.1.16.dist-info → ydb_sqlalchemy-0.1.17.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ydb-sqlalchemy
-Version: 0.1.16
+Version: 0.1.17
 Summary: YDB Dialect for SQLAlchemy
 Home-page: http://github.com/ydb-platform/ydb-sqlalchemy
 Author: Yandex LLC

{ydb_sqlalchemy-0.1.16.dist-info → ydb_sqlalchemy-0.1.17.dist-info}/RECORD

@@ -5,22 +5,22 @@ test/test_inspect.py,sha256=c4kc3jc48MCOfllO-ciiYf1vO-HOfuv0xVoXYT1Jxro,1106
 test/test_orm.py,sha256=jQVVld50zbUwxwgW9ySIWGaNDEOLzHKXjTkdpsG9TpA,1825
 test/test_suite.py,sha256=JYBGZjaRbg_ZiAqTHeCfL7DLnB6N6xkXN82gnooCyd8,31063
 ydb_sqlalchemy/__init__.py,sha256=hX7Gy-KOiHk7B5-0wj3ZmLjk4YDJnSMHIAqxVGn_PJY,181
-ydb_sqlalchemy/_version.py,sha256=EP5EHnJPa1QWoZlQBIhmsmHs-zCq83eQ9QAPmqT8HTo,19
-ydb_sqlalchemy/sqlalchemy/__init__.py,sha256=qLPF8gobyuwiXznh1-MPSQb5iTJc1p_Ypm7yz9owbGQ,17951
+ydb_sqlalchemy/_version.py,sha256=CQKgFdKOd8j2eckBUFbwPNJt1oASUfatjq5eO0eF8cA,19
+ydb_sqlalchemy/sqlalchemy/__init__.py,sha256=8_wPXgqlhyIW0CGmG1vIsjuepGyCOl8rAm7E0Hv90DY,18500
 ydb_sqlalchemy/sqlalchemy/datetime_types.py,sha256=wrI9kpsI_f7Jhbm7Fu0o_S1QoGCLIe6A9jfUwb41aMM,1929
 ydb_sqlalchemy/sqlalchemy/dbapi_adapter.py,sha256=7FDjganh9QStIkoXYPFfcRRhd07YCX63_8OmMnge1FI,3542
 ydb_sqlalchemy/sqlalchemy/dml.py,sha256=k_m6PLOAY7dVzG1gsyo2bB3Lp-o3rhzN0oSX_nfkbFU,310
 ydb_sqlalchemy/sqlalchemy/json.py,sha256=b4ydjlQjBhlhqGP_Sy2uZVKmt__D-9M7-YLGQMdYGME,1043
 ydb_sqlalchemy/sqlalchemy/requirements.py,sha256=zm6fcLormtk3KHnbtrBvxfkbG9ZyzNan38HrRB6vC3c,2505
-ydb_sqlalchemy/sqlalchemy/test_sqlalchemy.py,sha256=4wyRHmE8YQaMElQPHX6ToEj7A9F8Mvv909aaz_0wRnA,4535
+ydb_sqlalchemy/sqlalchemy/test_sqlalchemy.py,sha256=kSQ2sg90X5QoEAKfJSj3CIqI6j95tLUoUW8Fa197Um0,5410
 ydb_sqlalchemy/sqlalchemy/types.py,sha256=lFopPGW8WLPIrhgXEcnSAqzJMyrV74ryLSaeTnaKBzA,6125
 ydb_sqlalchemy/sqlalchemy/compiler/__init__.py,sha256=QqA6r-_bw1R97nQZy5ZSJN724znXg88l4mi5PpqAOxI,492
 ydb_sqlalchemy/sqlalchemy/compiler/base.py,sha256=JyDIEk3CvJggidFLSNoeI4Khjk3a-b0SjS80GrjgSIk,20248
 ydb_sqlalchemy/sqlalchemy/compiler/sa14.py,sha256=LanxAnwOiMnsnrY05B0jpmvGn5NXuOKMcxi_6N3obVM,1186
 ydb_sqlalchemy/sqlalchemy/compiler/sa20.py,sha256=rvVhe-pq5bOyuW4KMMMAD7JIWMzy355eijymBvuPwKw,3421
-ydb_sqlalchemy-0.1.16.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-ydb_sqlalchemy-0.1.16.dist-info/METADATA,sha256=F3A9xNjcBN9pL4wLc4V8-NKyz3TKN1Kcm25_xJ47vDg,5395
-ydb_sqlalchemy-0.1.16.dist-info/WHEEL,sha256=I3glN-nznogni2CWkAvi6vPRkBhTbVeYvJZnIsln6uc,109
-ydb_sqlalchemy-0.1.16.dist-info/entry_points.txt,sha256=iJxbKYuliWNBmL0iIiw8MxvOXrSEz5xe5fuEBqMRwCE,267
-ydb_sqlalchemy-0.1.16.dist-info/top_level.txt,sha256=iS69Y1GTAcTok0u0oQdxP-Q5iVgUGI71XBsaEUrWhMg,20
-ydb_sqlalchemy-0.1.16.dist-info/RECORD,,
+ydb_sqlalchemy-0.1.17.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+ydb_sqlalchemy-0.1.17.dist-info/METADATA,sha256=fhxTYNZXhbJs_hoyerJNTPmdqjVXSDbUa84YmsacZts,5395
+ydb_sqlalchemy-0.1.17.dist-info/WHEEL,sha256=I3glN-nznogni2CWkAvi6vPRkBhTbVeYvJZnIsln6uc,109
+ydb_sqlalchemy-0.1.17.dist-info/entry_points.txt,sha256=iJxbKYuliWNBmL0iIiw8MxvOXrSEz5xe5fuEBqMRwCE,267
+ydb_sqlalchemy-0.1.17.dist-info/top_level.txt,sha256=iS69Y1GTAcTok0u0oQdxP-Q5iVgUGI71XBsaEUrWhMg,20
+ydb_sqlalchemy-0.1.17.dist-info/RECORD,,