Examples with CLASSMETHOD com.jetbrains.python.psi.PyFunction.Modifier.CLASSMETHOD used on opensource projects

Example 1 with CLASSMETHOD

use of com.jetbrains.python.psi.PyFunction.Modifier.CLASSMETHOD in project intellij-community by JetBrains.

the class PyFunctionImpl method getModifier.

/**
   * Looks for two standard decorators to a function, or a wrapping assignment that closely follows it.
   *
   * @return a flag describing what was detected.
   */
@Nullable
public Modifier getModifier() {
    final String deconame = getClassOrStaticMethodDecorator();
    if (PyNames.CLASSMETHOD.equals(deconame)) {
        return CLASSMETHOD;
    } else if (PyNames.STATICMETHOD.equals(deconame)) {
        return STATICMETHOD;
    }
    // implicit staticmethod __new__
    final PyClass cls = getContainingClass();
    if (cls != null && PyNames.NEW.equals(getName()) && cls.isNewStyleClass(null)) {
        return STATICMETHOD;
    }
    final PyFunctionStub stub = getStub();
    if (stub != null) {
        return getModifierFromStub(stub);
    }
    final String funcName = getName();
    if (funcName != null) {
        PyAssignmentStatement currentAssignment = PsiTreeUtil.getNextSiblingOfType(this, PyAssignmentStatement.class);
        while (currentAssignment != null) {
            final String modifier = currentAssignment.getTargetsToValuesMapping().stream().filter(pair -> pair.getFirst() instanceof PyTargetExpression && funcName.equals(pair.getFirst().getName())).filter(pair -> pair.getSecond() instanceof PyCallExpression).map(pair -> interpretAsModifierWrappingCall((PyCallExpression) pair.getSecond(), this)).filter(interpreted -> interpreted != null && interpreted.getSecond() == this).map(interpreted -> interpreted.getFirst()).filter(wrapperName -> PyNames.CLASSMETHOD.equals(wrapperName) || PyNames.STATICMETHOD.equals(wrapperName)).findAny().orElse(null);
            if (PyNames.CLASSMETHOD.equals(modifier)) {
                return CLASSMETHOD;
            } else if (PyNames.STATICMETHOD.equals(modifier)) {
                return STATICMETHOD;
            }
            currentAssignment = PsiTreeUtil.getNextSiblingOfType(currentAssignment, PyAssignmentStatement.class);
        }
    }
    return null;
}