NameAndStatusTransformer.java

package com.jetbrains.python.refactoring.classes;

import com.google.common.base.Function;
import com.intellij.refactoring.classMembers.MemberInfoModel;
import com.jetbrains.python.psi.PyElement;
import com.jetbrains.python.refactoring.classes.membersManager.PyMemberInfo;
import org.jetbrains.annotations.NotNull;

/**
* @author Ilya.Kazakevich
*/
public class NameAndStatusTransformer implements Function<PyMemberInfo<PyElement>, PyPresenterTestMemberEntry> {
  @NotNull
  private final MemberInfoModel<PyElement, PyMemberInfo<PyElement>> myMemberInfoModel;

  public NameAndStatusTransformer(@NotNull final MemberInfoModel<PyElement, PyMemberInfo<PyElement>> memberInfoModel) {
    myMemberInfoModel = memberInfoModel;
  }

  @Override
  public PyPresenterTestMemberEntry apply(final PyMemberInfo<PyElement> input) {
    return new PyPresenterTestMemberEntry(input.getDisplayName(), myMemberInfoModel.isMemberEnabled(input), input.isStatic(), myMemberInfoModel.isAbstractEnabled(input));
  }
}