Skip to content

Commit 8b86e15

Browse files
committed
unnecesasry else
1 parent e56b127 commit 8b86e15

28 files changed

Lines changed: 132 additions & 151 deletions

pyproject.toml

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -208,3 +208,8 @@ module = [
208208
"json_stream.*"
209209
]
210210
ignore_missing_imports = true
211+
212+
[tool.ruff.lint]
213+
select = [
214+
"RET505", # supkkerfluous-else-return
215+
]

sqlmesh/cli/main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -96,7 +96,7 @@ def cli(
9696
if ctx.invoked_subcommand in SKIP_CONTEXT_COMMANDS:
9797
ctx.obj = path
9898
return
99-
elif ctx.invoked_subcommand in SKIP_LOAD_COMMANDS:
99+
if ctx.invoked_subcommand in SKIP_LOAD_COMMANDS:
100100
load = False
101101

102102
configs = load_configs(config, Context.CONFIG_TYPE, paths)

sqlmesh/core/audit/definition.py

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -95,18 +95,15 @@ def audit_map_validator(cls: t.Type, v: t.Any, values: t.Any) -> t.Dict[str, t.A
9595
return dict([_maybe_parse_arg_pair(v.unnest())])
9696
if isinstance(v, (exp.Tuple, exp.Array)):
9797
return dict(map(_maybe_parse_arg_pair, v.expressions))
98-
elif isinstance(v, dict):
98+
if isinstance(v, dict):
9999
dialect = get_dialect(values)
100100
return {
101101
key: value
102102
if isinstance(value, exp.Expression)
103103
else d.parse_one(str(value), dialect=dialect)
104104
for key, value in v.items()
105105
}
106-
else:
107-
raise_config_error(
108-
"Defaults must be a tuple of exp.EQ or a dict", error_type=AuditConfigError
109-
)
106+
raise_config_error("Defaults must be a tuple of exp.EQ or a dict", error_type=AuditConfigError)
110107
return {}
111108

112109

sqlmesh/core/config/root.py

Lines changed: 3 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -255,12 +255,9 @@ def get_gateway(self, name: t.Optional[str] = None) -> GatewayConfig:
255255
raise ConfigError(f"Missing gateway with name '{name}'.")
256256

257257
return self.gateways[name]
258-
else:
259-
if name is not None:
260-
raise ConfigError(
261-
"Gateway name is not supported when only one gateway is configured."
262-
)
263-
return self.gateways
258+
if name is not None:
259+
raise ConfigError("Gateway name is not supported when only one gateway is configured.")
260+
return self.gateways
264261

265262
def get_connection(self, gateway_name: t.Optional[str] = None) -> ConnectionConfig:
266263
return self.get_gateway(gateway_name).connection or self.default_connection

sqlmesh/core/engine_adapter/base.py

Lines changed: 24 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -375,33 +375,32 @@ def replace_query(
375375
column_descriptions=column_descriptions,
376376
**kwargs,
377377
)
378-
else:
379-
if self_referencing:
380-
with self.temp_table(
381-
self._select_columns(columns_to_types).from_(target_table),
382-
name=target_table,
383-
columns_to_types=columns_to_types,
384-
**kwargs,
385-
) as temp_table:
386-
for source_query in source_queries:
387-
source_query.add_transform(
388-
lambda node: ( # type: ignore
389-
temp_table # type: ignore
390-
if isinstance(node, exp.Table)
391-
and quote_identifiers(node) == quote_identifiers(target_table)
392-
else node
393-
)
378+
if self_referencing:
379+
with self.temp_table(
380+
self._select_columns(columns_to_types).from_(target_table),
381+
name=target_table,
382+
columns_to_types=columns_to_types,
383+
**kwargs,
384+
) as temp_table:
385+
for source_query in source_queries:
386+
source_query.add_transform(
387+
lambda node: ( # type: ignore
388+
temp_table # type: ignore
389+
if isinstance(node, exp.Table)
390+
and quote_identifiers(node) == quote_identifiers(target_table)
391+
else node
394392
)
395-
return self._insert_overwrite_by_condition(
396-
target_table,
397-
source_queries,
398-
columns_to_types,
399393
)
400-
return self._insert_overwrite_by_condition(
401-
target_table,
402-
source_queries,
403-
columns_to_types,
404-
)
394+
return self._insert_overwrite_by_condition(
395+
target_table,
396+
source_queries,
397+
columns_to_types,
398+
)
399+
return self._insert_overwrite_by_condition(
400+
target_table,
401+
source_queries,
402+
columns_to_types,
403+
)
405404

406405
def create_index(
407406
self,

sqlmesh/core/engine_adapter/bigquery.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -602,9 +602,12 @@ def insert_overwrite_by_partition(
602602
raise SQLMeshError(
603603
f"The partition expression '{partition_sql}' doesn't contain a column."
604604
)
605-
with self.session({}), self.temp_table(
606-
query_or_df, name=table_name, partitioned_by=partitioned_by
607-
) as temp_table_name:
605+
with (
606+
self.session({}),
607+
self.temp_table(
608+
query_or_df, name=table_name, partitioned_by=partitioned_by
609+
) as temp_table_name,
610+
):
608611
if columns_to_types is None or columns_to_types[
609612
partition_column.name
610613
] == exp.DataType.build("unknown"):
@@ -1158,7 +1161,7 @@ def _is_retryable(self, error: BaseException) -> bool:
11581161

11591162
if isinstance(error, self.retryable_errors):
11601163
return True
1161-
elif isinstance(error, Forbidden) and any(
1164+
if isinstance(error, Forbidden) and any(
11621165
e["reason"] == "rateLimitExceeded" for e in error.errors
11631166
):
11641167
return True

sqlmesh/core/macros.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -281,7 +281,7 @@ def evaluate_macros(
281281
self.parse_one(node.sql(dialect=self.dialect, copy=False))
282282
for node in transformed
283283
]
284-
elif isinstance(transformed, exp.Expression):
284+
if isinstance(transformed, exp.Expression):
285285
return self.parse_one(transformed.sql(dialect=self.dialect, copy=False))
286286

287287
return transformed
@@ -1265,7 +1265,7 @@ def resolve_template(
12651265
if mode.lower() == "table":
12661266
return exp.to_table(result, dialect=evaluator.dialect)
12671267
return exp.Literal.string(result)
1268-
elif evaluator.runtime_stage != RuntimeStage.LOADING.value:
1268+
if evaluator.runtime_stage != RuntimeStage.LOADING.value:
12691269
# only error if we are CREATING, EVALUATING or TESTING and @this_model is not present; this could indicate a bug
12701270
# otherwise, for LOADING, it's a no-op
12711271
raise SQLMeshError(
@@ -1391,7 +1391,7 @@ def _coerce(
13911391
return tuple(expr.expressions)
13921392
if generic[-1] is ...:
13931393
return tuple(_coerce(expr, generic[0], dialect, path) for expr in expr.expressions)
1394-
elif len(generic) == len(expr.expressions):
1394+
if len(generic) == len(expr.expressions):
13951395
return tuple(
13961396
_coerce(expr, generic[i], dialect, path)
13971397
for i, expr in enumerate(expr.expressions)

sqlmesh/core/model/definition.py

Lines changed: 13 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -2159,19 +2159,18 @@ def load_sql_based_model(
21592159
time_column_format=time_column_format,
21602160
**common_kwargs,
21612161
)
2162-
else:
2163-
seed_properties = {
2164-
p.name.lower(): p.args.get("value") for p in common_kwargs.pop("kind").expressions
2165-
}
2166-
try:
2167-
return create_seed_model(
2168-
name,
2169-
SeedKind(**seed_properties),
2170-
**common_kwargs,
2171-
)
2172-
except Exception as ex:
2173-
raise_config_error(str(ex), path)
2174-
raise
2162+
seed_properties = {
2163+
p.name.lower(): p.args.get("value") for p in common_kwargs.pop("kind").expressions
2164+
}
2165+
try:
2166+
return create_seed_model(
2167+
name,
2168+
SeedKind(**seed_properties),
2169+
**common_kwargs,
2170+
)
2171+
except Exception as ex:
2172+
raise_config_error(str(ex), path)
2173+
raise
21752174

21762175

21772176
def create_sql_model(
@@ -2565,7 +2564,7 @@ def _split_sql_model_statements(
25652564
if not query_positions:
25662565
return None, sql_statements, [], on_virtual_update, inline_audits
25672566

2568-
elif len(query_positions) > 1:
2567+
if len(query_positions) > 1:
25692568
raise_config_error("Only one SELECT query is allowed per model", path)
25702569

25712570
query, pos = query_positions[0]

sqlmesh/core/renderer.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -346,8 +346,7 @@ def _expand(node: exp.Expression) -> exp.Expression:
346346
alias=node.alias or model.view_name,
347347
copy=False,
348348
)
349-
else:
350-
logger.warning("Failed to expand the nested model '%s'", name)
349+
logger.warning("Failed to expand the nested model '%s'", name)
351350
return node
352351

353352
expression = expression.transform(_expand, copy=False) # type: ignore

sqlmesh/core/schema_diff.py

Lines changed: 10 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -90,15 +90,13 @@ def from_struct_kwarg(cls, struct: exp.ColumnDef) -> TableAlterColumn:
9090

9191
if kwarg_type.is_type(exp.DataType.Type.STRUCT):
9292
return cls.struct(name, quoted=quoted)
93-
elif kwarg_type.is_type(exp.DataType.Type.ARRAY):
93+
if kwarg_type.is_type(exp.DataType.Type.ARRAY):
9494
if kwarg_type.expressions and kwarg_type.expressions[0].is_type(
9595
exp.DataType.Type.STRUCT
9696
):
9797
return cls.array_of_struct(name, quoted=quoted)
98-
else:
99-
return cls.array_of_primitive(name, quoted=quoted)
100-
else:
101-
return cls.primitive(name, quoted=quoted)
98+
return cls.array_of_primitive(name, quoted=quoted)
99+
return cls.primitive(name, quoted=quoted)
102100

103101
@property
104102
def is_array(self) -> bool:
@@ -268,22 +266,21 @@ def expression(
268266
)
269267
],
270268
)
271-
elif self.is_add:
269+
if self.is_add:
272270
alter_table = exp.Alter(this=exp.to_table(table_name), kind="TABLE")
273271
column = self.column_def(array_element_selector)
274272
alter_table.set("actions", [column])
275273
if self.add_position:
276274
column.set("position", self.add_position.column_position_node)
277275
return alter_table
278-
elif self.is_drop:
276+
if self.is_drop:
279277
alter_table = exp.Alter(this=exp.to_table(table_name), kind="TABLE")
280278
drop_column = exp.Drop(
281279
this=self.column(array_element_selector), kind="COLUMN", cascade=self.cascade
282280
)
283281
alter_table.set("actions", [drop_column])
284282
return alter_table
285-
else:
286-
raise ValueError(f"Unknown operation {self.op}")
283+
raise ValueError(f"Unknown operation {self.op}")
287284

288285

289286
class SchemaDiffer(PydanticModel):
@@ -593,7 +590,7 @@ def _alter_operation(
593590
)
594591
if self._is_coerceable_type(current_type, new_type):
595592
return []
596-
elif self._is_compatible_type(current_type, new_type):
593+
if self._is_compatible_type(current_type, new_type):
597594
struct.expressions.pop(pos)
598595
struct.expressions.insert(pos, new_kwarg)
599596
col_pos = (
@@ -610,10 +607,9 @@ def _alter_operation(
610607
col_pos,
611608
)
612609
]
613-
else:
614-
return self._drop_operation(
615-
columns, root_struct, pos, root_struct
616-
) + self._add_operation(columns, pos, new_kwarg, struct, root_struct)
610+
return self._drop_operation(columns, root_struct, pos, root_struct) + self._add_operation(
611+
columns, pos, new_kwarg, struct, root_struct
612+
)
617613

618614
def _resolve_alter_operations(
619615
self,

0 commit comments

Comments
 (0)