From e0711e191ce67ca510acb70b09e882e7840caab4 Mon Sep 17 00:00:00 2001 From: Denis Artyushin Date: Sun, 18 Feb 2024 18:04:42 +0300 Subject: [PATCH] Refactor with black --- bump_pydantic/codemods/replace_config.py | 28 +++++++++++++----------- 1 file changed, 15 insertions(+), 13 deletions(-) diff --git a/bump_pydantic/codemods/replace_config.py b/bump_pydantic/codemods/replace_config.py index 9eef776..0792444 100644 --- a/bump_pydantic/codemods/replace_config.py +++ b/bump_pydantic/codemods/replace_config.py @@ -217,20 +217,22 @@ def leave_config_class_childless(self, original_node: cst.ClassDef, updated_node AddImportsVisitor.add_needed_import(context=self.context, **needed_import) # type: ignore[arg-type] block = cst.ensure_type(updated_node.body, cst.IndentedBlock) body = [ - cst.SimpleStatementLine( - body=[ - cst.Assign( - targets=[cst.AssignTarget(target=cst.Name("model_config"))], - value=cst.Call( - func=cst.Name("SettingsConfigDict" if self.is_base_settings else "ConfigDict"), - args=self.config_args, - ), - ) - ], - leading_lines=self._leading_lines_from_removed_keys(self.config_args), + ( + cst.SimpleStatementLine( + body=[ + cst.Assign( + targets=[cst.AssignTarget(target=cst.Name("model_config"))], + value=cst.Call( + func=cst.Name("SettingsConfigDict" if self.is_base_settings else "ConfigDict"), + args=self.config_args, + ), + ) + ], + leading_lines=self._leading_lines_from_removed_keys(self.config_args), + ) + if m.matches(statement, m.ClassDef(name=m.Name(value="Config"))) + else statement ) - if m.matches(statement, m.ClassDef(name=m.Name(value="Config"))) - else statement for statement in block.body ] self.is_base_settings = False