diff --git a/src/codemodder/codemodder.py b/src/codemodder/codemodder.py index 0aabb82d..93c86b04 100644 --- a/src/codemodder/codemodder.py +++ b/src/codemodder/codemodder.py @@ -89,8 +89,7 @@ def apply_codemods( for codemod in codemods_to_run: # NOTE: this may be used as a progress indicator by upstream tools logger.info("running codemod %s", codemod.id) - codemod_token_usage = codemod.apply(context) - if codemod_token_usage: + if codemod_token_usage := codemod.apply(context): log_token_usage(f"Codemod {codemod.id}", codemod_token_usage) token_usage += codemod_token_usage diff --git a/src/codemodder/codemods/utils_mixin.py b/src/codemodder/codemods/utils_mixin.py index a6bce24c..a1b3111f 100644 --- a/src/codemodder/codemods/utils_mixin.py +++ b/src/codemodder/codemods/utils_mixin.py @@ -308,9 +308,9 @@ def class_has_method(self, classdef: cst.ClassDef, method_name: str) -> bool: """Check if a given class definition implements a method of name `method_name`.""" for node in classdef.body.body: match node: - case cst.FunctionDef( - name=cst.Name(value=value) - ) if value == method_name: + case cst.FunctionDef(name=cst.Name(value=value)) if ( + value == method_name + ): return True return False @@ -331,7 +331,9 @@ def is_value_of_assignment( | cst.Assign(value=value) | cst.WithItem(item=value) | cst.NamedExpr(value=value) - ) if expr == value: # type: ignore + ) if ( + expr == value + ): # type: ignore return parent return None diff --git a/src/codemodder/utils/format_string_parser.py b/src/codemodder/utils/format_string_parser.py index f0ef769a..e3250832 100644 --- a/src/codemodder/utils/format_string_parser.py +++ b/src/codemodder/utils/format_string_parser.py @@ -128,7 +128,7 @@ def expressions_from_replacements( def dict_to_values_dict( - expr_dict: dict[cst.BaseExpression, cst.BaseExpression] + expr_dict: dict[cst.BaseExpression, cst.BaseExpression], ) -> dict[str | cst.BaseExpression, cst.BaseExpression]: return { extract_raw_value(k): v