From 78714d572e1411abdc89e143050657ff6b8b3f03 Mon Sep 17 00:00:00 2001 From: Oleg Ovcharuk Date: Thu, 28 Nov 2024 16:20:53 +0300 Subject: [PATCH] Fix declare param_name cutting --- ydb_sqlalchemy/sqlalchemy/__init__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ydb_sqlalchemy/sqlalchemy/__init__.py b/ydb_sqlalchemy/sqlalchemy/__init__.py index c7ad20e..7175e00 100644 --- a/ydb_sqlalchemy/sqlalchemy/__init__.py +++ b/ydb_sqlalchemy/sqlalchemy/__init__.py @@ -354,7 +354,7 @@ def _format_variables( def _add_declare_for_yql_stmt_vars_impl(self, statement, parameters_types): declarations = "\n".join( [ - f"DECLARE $`{param_name[1:]}` as {str(param_type)};" + f"DECLARE $`{param_name[1:] if param_name.startswith('$') else param_name}` as {str(param_type)};" for param_name, param_type in parameters_types.items() ] )