Skip to content

Commit 28300f3

Browse files
authored
Merge pull request #65 from ydb-platform/update_db_name
Add slash to DB name
2 parents dd95939 + bb63d01 commit 28300f3

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

ydb_sqlalchemy/sqlalchemy/__init__.py

+9
Original file line numberDiff line numberDiff line change
@@ -307,6 +307,15 @@ def reset_ydb_request_settings(self, dbapi_connection: ydb_dbapi.Connection):
307307
def get_ydb_request_settings(self, dbapi_connection: ydb_dbapi.Connection) -> ydb.BaseRequestSettings:
308308
return dbapi_connection.get_ydb_request_settings()
309309

310+
def create_connect_args(self, url):
311+
args, kwargs = super().create_connect_args(url)
312+
# YDB database name should start with '/'
313+
if "database" in kwargs:
314+
if not kwargs["database"].startswith("/"):
315+
kwargs["database"] = "/" + kwargs["database"]
316+
317+
return [args, kwargs]
318+
310319
def connect(self, *cargs, **cparams):
311320
return self.dbapi.connect(*cargs, **cparams)
312321

0 commit comments

Comments
 (0)