Add slash to DB name

ydb-platform · Dec 4, 2024 · ff500cd · ff500cd
1 parent 89bba30
commit ff500cd
Showing 1 changed file with 9 additions and 0 deletions.
diff --git 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)