Skip to content

Commit 8ea65de

Browse files
author
kabulov kozim
committed
fix style
1 parent dc2c836 commit 8ea65de

File tree

2 files changed

+17
-4
lines changed

2 files changed

+17
-4
lines changed

test/test_core.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -95,10 +95,12 @@ class TestDirectories(TablesTest):
9595

9696
def prepare_table(self, engine):
9797
base = declarative_base()
98+
9899
class Table(base):
99100
__tablename__ = 'dir/test'
100101
id = Column(Integer, primary_key=True)
101102
text = Column(Unicode)
103+
102104
base.metadata.create_all(engine)
103105
session = sessionmaker(bind=engine)()
104106
session.add(Table(id=2, text="foo"))

ydb_sqlalchemy/sqlalchemy/__init__.py

Lines changed: 15 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)