Refactor extract get_code_actions()

lieryan · lieryan · commit 6c51814bfe72 · 2021-09-29T09:01:01.000+10:00
diff --git a/pymode/rope.py b/pymode/rope.py
@@ -463,10 +463,11 @@ def run(self):
                 if not input_str:
                     return False
 
+                code_actions = self.get_code_actions()
                 action = env.user_input_choices(
-                    'Choose what to do:', 'perform', 'preview',
-                    'perform in class hierarchy',
-                    'preview in class hierarchy')
+                    'Choose what to do:',
+                    *code_actions,
+                )
 
                 in_hierarchy = action.endswith("in class hierarchy")
 
@@ -492,6 +493,14 @@ def run(self):
             except Exception as e: # noqa
                 env.error('Unhandled exception in Pymode: %s' % e)
 
+    def get_code_actions(self):
+        return [
+            'perform',
+            'preview',
+            'perform in class hierarchy',
+            'preview in class hierarchy',
+        ]
+
     @staticmethod
     def get_refactor(ctx):
         """ Get refactor object. """
