Search in sources :

Example 1 with PyNames

use of com.jetbrains.python.PyNames in project intellij-community by JetBrains.

the class PyClassTypeImpl method doResolveMember.

@Nullable
private List<? extends RatedResolveResult> doResolveMember(@NotNull String name, @Nullable PyExpression location, @NotNull AccessDirection direction, @NotNull PyResolveContext resolveContext, boolean inherited) {
    final TypeEvalContext context = resolveContext.getTypeEvalContext();
    PsiElement classMember = //overriding members provers have priority to normal resolve
    resolveByOverridingMembersProviders(this, name, location, context);
    if (classMember != null) {
        return ResolveResultList.to(classMember);
    }
    if (resolveContext.allowProperties()) {
        final Ref<ResolveResultList> resultRef = findProperty(name, direction, true, resolveContext.getTypeEvalContext());
        if (resultRef != null) {
            return resultRef.get();
        }
    }
    if ("super".equals(getClassQName()) && isBuiltin() && location instanceof PyCallExpression) {
        // methods of super() call are not of class super!
        PyExpression first_arg = ((PyCallExpression) location).getArgument(0, PyExpression.class);
        if (first_arg != null) {
            // the usual case: first arg is the derived class that super() is proxying for
            PyType first_arg_type = context.getType(first_arg);
            if (first_arg_type instanceof PyClassType) {
                PyClass derived_class = ((PyClassType) first_arg_type).getPyClass();
                final Iterator<PyClass> base_it = derived_class.getAncestorClasses(context).iterator();
                if (base_it.hasNext()) {
                    return new PyClassTypeImpl(base_it.next(), true).resolveMember(name, location, direction, resolveContext);
                } else {
                    // no base classes = super() cannot proxy anything meaningful from a base class
                    return null;
                }
            }
        }
    }
    final List<? extends RatedResolveResult> classMembers = resolveInner(myClass, myIsDefinition, name, location);
    if (PyNames.__CLASS__.equals(name)) {
        return resolveDunderClass(context, classMembers);
    }
    if (!classMembers.isEmpty()) {
        return classMembers;
    }
    if (PyNames.DOC.equals(name)) {
        return Optional.ofNullable(PyBuiltinCache.getInstance(myClass).getObjectType()).map(type -> type.resolveMember(name, location, direction, resolveContext)).orElse(Collections.emptyList());
    }
    classMember = resolveByOverridingAncestorsMembersProviders(this, name, location);
    if (classMember != null) {
        final ResolveResultList list = new ResolveResultList();
        int rate = RatedResolveResult.RATE_NORMAL;
        for (PyResolveResultRater rater : Extensions.getExtensions(PyResolveResultRater.EP_NAME)) {
            rate += rater.getMemberRate(classMember, this, context);
        }
        list.poke(classMember, rate);
        return list;
    }
    if (inherited) {
        for (PyClassLikeType type : myClass.getAncestorTypes(context)) {
            if (type instanceof PyClassType) {
                if (!myIsDefinition) {
                    type = type.toInstance();
                }
                final List<? extends RatedResolveResult> superMembers = resolveInner(((PyClassType) type).getPyClass(), myIsDefinition, name, location);
                if (!superMembers.isEmpty()) {
                    return superMembers;
                }
            }
            if (type != null) {
                final List<? extends RatedResolveResult> results = type.resolveMember(name, location, direction, resolveContext, false);
                if (results != null && !results.isEmpty()) {
                    return results;
                }
            }
        }
    }
    if (inherited && !PyNames.INIT.equals(name) && !PyNames.NEW.equals(name)) {
        final List<? extends RatedResolveResult> typeMembers = resolveMetaClassMember(name, location, direction, resolveContext);
        if (typeMembers != null) {
            return typeMembers;
        }
    }
    if (inherited) {
        classMember = resolveByMembersProviders(this, name, location, //ask providers after real class introspection as providers have less priority
        context);
    }
    if (classMember != null) {
        return ResolveResultList.to(classMember);
    }
    if (inherited) {
        for (PyClassLikeType type : myClass.getAncestorTypes(context)) {
            if (type instanceof PyClassType) {
                final PyClass pyClass = ((PyClassType) type).getPyClass();
                PsiElement superMember = resolveByMembersProviders(new PyClassTypeImpl(pyClass, isDefinition()), name, location, resolveContext.getTypeEvalContext());
                if (superMember != null) {
                    return ResolveResultList.to(superMember);
                }
            }
        }
    }
    return Collections.emptyList();
}
Also used : com.intellij.openapi.util(com.intellij.openapi.util) PyNames(com.jetbrains.python.PyNames) java.util(java.util) ArrayUtil(com.intellij.util.ArrayUtil) ContainerUtil(com.intellij.util.containers.ContainerUtil) PyUtil.as(com.jetbrains.python.psi.PyUtil.as) PyResolveProcessor(com.jetbrains.python.psi.resolve.PyResolveProcessor) PsiTreeUtil(com.intellij.psi.util.PsiTreeUtil) CompletionVariantsProcessor(com.jetbrains.python.psi.resolve.CompletionVariantsProcessor) Lists(com.google.common.collect.Lists) PsiElement(com.intellij.psi.PsiElement) ResolveResultList(com.jetbrains.python.psi.impl.ResolveResultList) PsiFile(com.intellij.psi.PsiFile) com.jetbrains.python.psi(com.jetbrains.python.psi) QualifiedName(com.intellij.psi.util.QualifiedName) CompletionUtil(com.intellij.codeInsight.completion.CompletionUtil) PyCustomMemberUtils(com.jetbrains.python.codeInsight.PyCustomMemberUtils) PsiInvalidElementAccessException(com.intellij.psi.PsiInvalidElementAccessException) ProcessingContext(com.intellij.util.ProcessingContext) Extensions(com.intellij.openapi.extensions.Extensions) LookupElementBuilder(com.intellij.codeInsight.lookup.LookupElementBuilder) PyReferenceImpl(com.jetbrains.python.psi.impl.references.PyReferenceImpl) LookupElement(com.intellij.codeInsight.lookup.LookupElement) PyResolveContext(com.jetbrains.python.psi.resolve.PyResolveContext) Collectors(java.util.stream.Collectors) Sets(com.google.common.collect.Sets) RatedResolveResult(com.jetbrains.python.psi.resolve.RatedResolveResult) PyBuiltinCache(com.jetbrains.python.psi.impl.PyBuiltinCache) Nullable(org.jetbrains.annotations.Nullable) PyResolveResultRater(com.jetbrains.python.psi.impl.PyResolveResultRater) Maybe(com.jetbrains.python.toolbox.Maybe) PyResolveImportUtil.resolveTopLevelMember(com.jetbrains.python.psi.resolve.PyResolveImportUtil.resolveTopLevelMember) Processor(com.intellij.util.Processor) PyCustomMember(com.jetbrains.python.codeInsight.PyCustomMember) ObjectUtils(com.intellij.util.ObjectUtils) PyResolveImportUtil.fromFoothold(com.jetbrains.python.psi.resolve.PyResolveImportUtil.fromFoothold) NotNull(org.jetbrains.annotations.NotNull) PyResolveResultRater(com.jetbrains.python.psi.impl.PyResolveResultRater) ResolveResultList(com.jetbrains.python.psi.impl.ResolveResultList) PsiElement(com.intellij.psi.PsiElement) Nullable(org.jetbrains.annotations.Nullable)

Aggregations

Lists (com.google.common.collect.Lists)1 Sets (com.google.common.collect.Sets)1 CompletionUtil (com.intellij.codeInsight.completion.CompletionUtil)1 LookupElement (com.intellij.codeInsight.lookup.LookupElement)1 LookupElementBuilder (com.intellij.codeInsight.lookup.LookupElementBuilder)1 Extensions (com.intellij.openapi.extensions.Extensions)1 com.intellij.openapi.util (com.intellij.openapi.util)1 PsiElement (com.intellij.psi.PsiElement)1 PsiFile (com.intellij.psi.PsiFile)1 PsiInvalidElementAccessException (com.intellij.psi.PsiInvalidElementAccessException)1 PsiTreeUtil (com.intellij.psi.util.PsiTreeUtil)1 QualifiedName (com.intellij.psi.util.QualifiedName)1 ArrayUtil (com.intellij.util.ArrayUtil)1 ObjectUtils (com.intellij.util.ObjectUtils)1 ProcessingContext (com.intellij.util.ProcessingContext)1 Processor (com.intellij.util.Processor)1 ContainerUtil (com.intellij.util.containers.ContainerUtil)1 PyNames (com.jetbrains.python.PyNames)1 PyCustomMember (com.jetbrains.python.codeInsight.PyCustomMember)1 PyCustomMemberUtils (com.jetbrains.python.codeInsight.PyCustomMemberUtils)1