@@ -584,7 +584,14 @@ class YqlDialect(StrCompileDialect):
584584 def import_dbapi (cls : Any ):
585585 return dbapi .YdbDBApi ()
586586
587- def __init__ (self , json_serializer = None , json_deserializer = None , _add_declare_for_yql_stmt_vars = False , directories = [], ** kwargs ):
587+ def __init__ (
588+ self ,
589+ json_serializer = None ,
590+ json_deserializer = None ,
591+ _add_declare_for_yql_stmt_vars = False ,
592+ directories = [],
593+ ** kwargs ,
594+ ):
588595 super ().__init__ (** kwargs )
589596
590597 self ._json_deserializer = json_deserializer
@@ -677,7 +684,7 @@ def do_commit(self, dbapi_connection: dbapi.Connection) -> None:
677684 def _fix_variable_name (self , variable ):
678685 for directory in self ._directories :
679686 if variable .startswith (f"{ directory } /" ):
680- return f"{ directory } _" + variable [len (directory ) + 1 :]
687+ return f"{ directory } _" + variable [len (directory ) + 1 :]
681688 return variable
682689
683690 def _format_variables (
@@ -696,12 +703,16 @@ def _format_variables(
696703 formatted_parameters = []
697704 for i in range (len (parameters_sequence )):
698705 variable_names .update (set (parameters_sequence [i ].keys ()))
699- formatted_parameters .append ({f"${ self ._fix_variable_name (k )} " : v for k , v in parameters_sequence [i ].items ()})
706+ formatted_parameters .append (
707+ {f"${ self ._fix_variable_name (k )} " : v for k , v in parameters_sequence [i ].items ()}
708+ )
700709 else :
701710 variable_names = set (parameters .keys ())
702711 formatted_parameters = {f"${ self ._fix_variable_name (k )} " : v for k , v in parameters .items ()}
703712
704- formatted_variable_names = {variable_name : f"${ self ._fix_variable_name (variable_name )} " for variable_name in variable_names }
713+ formatted_variable_names = {
714+ variable_name : f"${ self ._fix_variable_name (variable_name )} " for variable_name in variable_names
715+ }
705716 formatted_statement = formatted_statement % formatted_variable_names
706717
707718 formatted_statement = formatted_statement .replace ("%%" , "%" )
0 commit comments