python/edu/learn-python/src/com/jetbrains/python/edu/highlighting/StudyVisitorFilter.java


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18

package com.jetbrains.python.edu.highlighting;

import com.intellij.psi.PsiFile;
import com.jetbrains.python.edu.StudyTaskManager;
import com.jetbrains.python.inspections.PythonVisitorFilter;
import com.jetbrains.python.inspections.unresolvedReference.PyUnresolvedReferencesInspection;
import org.jetbrains.annotations.NotNull;

public class StudyVisitorFilter implements PythonVisitorFilter {
  @Override
  public boolean isSupported(@NotNull final Class visitorClass, @NotNull final PsiFile file) {
    if (StudyTaskManager.getInstance(file.getProject()).getCourse() == null) return true;
    if (visitorClass == PyUnresolvedReferencesInspection.class) {
      return false;
    }
    return true;
  }
}