diff options
-rw-r--r-- | scripts/codemods/rename_pyqt.py | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/scripts/codemods/rename_pyqt.py b/scripts/codemods/rename_pyqt.py index ba20f4f54..e4cf82b0b 100644 --- a/scripts/codemods/rename_pyqt.py +++ b/scripts/codemods/rename_pyqt.py @@ -117,6 +117,13 @@ class RenameCommand(VisitorBasedCodemodCommand): # this so that we do not end up with two of the same import. self.bypass_import = False + def per_module_reset(self): + """Reset attributes that are cached on self.""" + # They should probably really be on self.context.scratch? + self.as_name = None + self.scheduled_removals = set() + self.bypass_import = False + def visit_Import(self, node: cst.Import) -> None: for import_alias in node.names: alias_name = get_full_name_for_node(import_alias.name) @@ -315,6 +322,9 @@ class RenameCommand(VisitorBasedCodemodCommand): return updated_node + def visit_Module(self, node: cst.Module) -> Optional[bool]: + self.per_module_reset() + def leave_Module( self, original_node: cst.Module, updated_node: cst.Module ) -> cst.Module: |