コード例 #1
0
 @Override
 protected void performInplaceIntroduce(IntroduceOperation operation) {
   final PsiElement statement = performRefactoring(operation);
   // put caret on identifier after "self."
   if (statement instanceof PyAssignmentStatement) {
     final List<PsiElement> occurrences = operation.getOccurrences();
     final PsiElement occurrence = findOccurrenceUnderCaret(occurrences, operation.getEditor());
     PyTargetExpression target =
         (PyTargetExpression) ((PyAssignmentStatement) statement).getTargets()[0];
     putCaretOnFieldName(operation.getEditor(), occurrence != null ? occurrence : target);
     final InplaceVariableIntroducer<PsiElement> introducer =
         new PyInplaceFieldIntroducer(target, operation, occurrences);
     introducer.performInplaceRefactoring(new LinkedHashSet<>(operation.getSuggestedNames()));
   }
 }
コード例 #2
0
 public PyInplaceFieldIntroducer(
     PyTargetExpression target, IntroduceOperation operation, List<PsiElement> occurrences) {
   super(
       target,
       operation.getEditor(),
       operation.getProject(),
       "Introduce Field",
       occurrences.toArray(new PsiElement[occurrences.size()]),
       null);
   myTarget = target;
   myOperation = operation;
   if (operation.getAvailableInitPlaces().size() > 1) {
     myPanel = new PyIntroduceFieldPanel(myProject, operation.getAvailableInitPlaces());
   } else {
     myPanel = null;
   }
 }
コード例 #3
0
 @Override
 protected boolean checkEnabled(IntroduceOperation operation) {
   if (PyUtil.getContainingClassOrSelf(operation.getElement()) == null) {
     CommonRefactoringUtil.showErrorHint(
         operation.getProject(),
         operation.getEditor(),
         "Cannot introduce field: not in class",
         myDialogTitle,
         getHelpId());
     return false;
   }
   if (dependsOnLocalScopeValues(operation.getElement())) {
     operation.removeAvailableInitPlace(InitPlace.CONSTRUCTOR);
     operation.removeAvailableInitPlace(InitPlace.SET_UP);
   }
   return true;
 }