Merge pull request #4 from ydb-platform/add_pragma_table_prefix

Add table prefix pragma to queries

Author: vgvoleg

ydb_dbapi/cursors.py

@@ -39,6 +39,8 @@ def __init__(self) -> None:
         self._description: list[tuple] | None = None
         self._state: CursorStatus = CursorStatus.ready
 
+        self._table_path_prefix: str = ""
+
     @property
     def description(self) -> list[tuple] | None:
         return self._description
@@ -134,6 +136,12 @@ def _fetchall_from_buffer(self) -> list:
         self._raise_if_closed()
         return list(self._rows or iter([]))
 
+    def _append_table_path_prefix(self, query: str) -> str:
+        if self._table_path_prefix:
+            prgm = f'PRAGMA TablePathPrefix = "{self._table_path_prefix}";\n'
+            return prgm + query
+        return query
+
 
 class Cursor(BufferedCursor):
     def __init__(
@@ -225,6 +233,9 @@ def execute(
     ) -> None:
         self._raise_if_closed()
         self._raise_if_running()
+
+        query = self._append_table_path_prefix(query)
+
         if self._tx_context is not None:
             self._stream = self._execute_transactional_query(
                 tx_context=self._tx_context, query=query, parameters=parameters
@@ -376,6 +387,9 @@ async def execute(
     ) -> None:
         self._raise_if_closed()
         self._raise_if_running()
+
+        query = self._append_table_path_prefix(query)
+
         if self._tx_context is not None:
             self._stream = await self._execute_transactional_query(
                 tx_context=self._tx_context, query=query, parameters=parameters