Rename to ydb_table_path_prefix

Author: LuckySting

test/test_core.py

@@ -997,7 +997,7 @@ def test_index_with_join_usage(self, connection: sa.Connection, metadata: sa.Met
         assert cursor.one() == ("Sarah Connor", "wanted")
 
 
-class TestRootDirectory(TablesTest):
+class TestTablePathPrefix(TablesTest):
     __backend__ = True
 
     @classmethod
@@ -1013,8 +1013,8 @@ def insert_data(cls, connection: sa.Connection):
         connection.execute(sa.insert(table).values({"id": 1}))
         connection.execute(sa.insert(root_table).values({"id": 2}))
 
-    def test_root_directory(self):
-        engine = sa.create_engine(config.db_url, connect_args={"ydb_root_directory": "/local/some_dir/nested_dir"})
+    def test_select(self):
+        engine = sa.create_engine(config.db_url, connect_args={"ydb_table_path_prefix": "/local/some_dir/nested_dir"})
         rel_table = Table("table", sa.MetaData(), sa.Column("id", sa.Integer, primary_key=True))
         abs_table = Table("/local/table", sa.MetaData(), sa.Column("id", sa.Integer, primary_key=True))
 
@@ -1026,8 +1026,10 @@ def test_root_directory(self):
         assert result2 == 2
 
     def test_two_engines(self):
-        create_engine = sa.create_engine(config.db_url, connect_args={"ydb_root_directory": "/local/two/engines/test"})
-        select_engine = sa.create_engine(config.db_url, connect_args={"ydb_root_directory": "/local/two"})
+        create_engine = sa.create_engine(
+            config.db_url, connect_args={"ydb_table_path_prefix": "/local/two/engines/test"}
+        )
+        select_engine = sa.create_engine(config.db_url, connect_args={"ydb_table_path_prefix": "/local/two"})
         table_to_create = Table("table", sa.MetaData(), sa.Column("id", sa.Integer, primary_key=True))
         table_to_select = Table("engines/test/table", sa.MetaData(), sa.Column("id", sa.Integer, primary_key=True))
 
@@ -1044,7 +1046,7 @@ def test_two_engines(self):
         assert result == 42
 
     def test_reflection(self):
-        reflection_engine = sa.create_engine(config.db_url, connect_args={"ydb_root_directory": "/local/some_dir"})
+        reflection_engine = sa.create_engine(config.db_url, connect_args={"ydb_table_path_prefix": "/local/some_dir"})
         metadata = sa.MetaData()
 
         metadata.reflect(reflection_engine)

ydb_sqlalchemy/dbapi/connection.py

@@ -38,7 +38,7 @@ def __init__(
         self.database = database
         self.conn_kwargs = conn_kwargs
         self.credentials = self.conn_kwargs.pop("credentials", None)
-        self.root_directory = self.conn_kwargs.pop("ydb_root_directory", "")
+        self.table_path_prefix = self.conn_kwargs.pop("ydb_table_path_prefix", "")
 
         if "ydb_session_pool" in self.conn_kwargs:  # Use session pool managed manually
             self._shared_session_pool = True
@@ -59,20 +59,20 @@ def __init__(
         self.tx_context: Optional[ydb.TxContext] = None
 
     def cursor(self):
-        return self._cursor_class(self.session_pool, self.tx_mode, self.tx_context, self.root_directory)
+        return self._cursor_class(self.session_pool, self.tx_mode, self.tx_context, self.table_path_prefix)
 
     def describe(self, table_path: str) -> ydb.TableDescription:
-        abs_table_path = posixpath.join(self.database, self.root_directory, table_path)
+        abs_table_path = posixpath.join(self.database, self.table_path_prefix, table_path)
         cursor = self.cursor()
         return cursor.describe_table(abs_table_path)
 
     def check_exists(self, table_path: str) -> ydb.SchemeEntry:
-        abs_table_path = posixpath.join(self.database, self.root_directory, table_path)
+        abs_table_path = posixpath.join(self.database, self.table_path_prefix, table_path)
         cursor = self.cursor()
         return cursor.check_exists(abs_table_path)
 
     def get_table_names(self) -> List[str]:
-        abs_dir_path = posixpath.join(self.database, self.root_directory)
+        abs_dir_path = posixpath.join(self.database, self.table_path_prefix)
         cursor = self.cursor()
         return [posixpath.relpath(path, abs_dir_path) for path in cursor.get_table_names(abs_dir_path)]
 

ydb_sqlalchemy/dbapi/cursor.py

@@ -79,7 +79,7 @@ def __init__(
         session_pool: Union[ydb.SessionPool, ydb.aio.SessionPool],
         tx_mode: ydb.AbstractTransactionModeBuilder,
         tx_context: Optional[ydb.BaseTxContext] = None,
-        root_directory: str = "",
+        table_path_prefix: str = "",
     ):
         self.session_pool = session_pool
         self.tx_mode = tx_mode
@@ -88,7 +88,7 @@ def __init__(
         self.arraysize = 1
         self.rows = None
         self._rows_prefetched = None
-        self.root_directory = root_directory
+        self.root_directory = table_path_prefix
 
     @_handle_ydb_errors
     def describe_table(self, abs_table_path: str) -> ydb.TableDescription: