Merge pull request #62 from ydb-platform/fix_declare_filling

vgvoleg · web-flow · commit 97526e5f05d9 · 2024-11-29T10:51:07.000+03:00
Fix declare param_name cutting
diff --git 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()
             ]
         )

Original file line number	Diff line number	Diff line change
`@@ -354,7 +354,7 @@ def _format_variables(`
`354`	`354`	`def _add_declare_for_yql_stmt_vars_impl(self, statement, parameters_types):`
`355`	`355`	`declarations = "\n".join(`
`356`	`356`	`[`
`357`		- f"DECLARE $`{param_name[1:]}` as {str(param_type)};"
	`357`	+ f"DECLARE $`{param_name[1:] if param_name.startswith('$') else param_name}` as {str(param_type)};"
`358`	`358`	`for param_name, param_type in parameters_types.items()`
`359`	`359`	`]`
`360`	`360`	`)`