From ff500cd7108807b2b2a469ee93d19337001cc84d Mon Sep 17 00:00:00 2001 From: Oleg Ovcharuk Date: Wed, 4 Dec 2024 11:15:28 +0300 Subject: [PATCH] Add slash to DB name --- ydb_sqlalchemy/sqlalchemy/__init__.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/ydb_sqlalchemy/sqlalchemy/__init__.py b/ydb_sqlalchemy/sqlalchemy/__init__.py index 7175e00..7bdf8f8 100644 --- a/ydb_sqlalchemy/sqlalchemy/__init__.py +++ b/ydb_sqlalchemy/sqlalchemy/__init__.py @@ -307,6 +307,15 @@ def reset_ydb_request_settings(self, dbapi_connection: ydb_dbapi.Connection): def get_ydb_request_settings(self, dbapi_connection: ydb_dbapi.Connection) -> ydb.BaseRequestSettings: return dbapi_connection.get_ydb_request_settings() + def create_connect_args(self, url): + args, kwargs = super().create_connect_args(url) + # YDB database name should start with '/' + if "database" in kwargs: + if not kwargs["database"].startswith("/"): + kwargs["database"] = "/" + kwargs["database"] + + return [args, kwargs] + def connect(self, *cargs, **cparams): return self.dbapi.connect(*cargs, **cparams)