From 89fab4571bf005f7d41a973e0b920f875167e42f Mon Sep 17 00:00:00 2001
From: Elizaveta Shashkova <Elizaveta.Shashkova@jetbrains.com>
Date: Fri, 6 Dec 2019 15:06:04 +0300
Subject: [PATCH] PY-26367 Disable Pep8 inside code fragments

GitOrigin-RevId: d3262b71c18e420efc186c21a90fe1f0d56999bd
---
 python/src/META-INF/python-core-common.xml     |  4 ++++
 .../PyExpressionCodeFragmentVisitorFilter.kt   | 18 ++++++++++++++++++
 2 files changed, 22 insertions(+)
 create mode 100644 python/src/com/jetbrains/python/inspections/PyExpressionCodeFragmentVisitorFilter.kt

diff --git a/python/src/META-INF/python-core-common.xml b/python/src/META-INF/python-core-common.xml
index 5ff6a6ddac66..8177aa9f677e 100644
--- a/python/src/META-INF/python-core-common.xml
+++ b/python/src/META-INF/python-core-common.xml
@@ -703,6 +703,10 @@
     <visitorFilter language="PyFunctionTypeComment"
                    implementationClass="com.jetbrains.python.inspections.PyFunctionTypeAnnotationVisitorFilter"/>
 
+    <!-- PyExpressionCodeFragment -->
+    <visitorFilter language="Python"
+                   implementationClass="com.jetbrains.python.inspections.PyExpressionCodeFragmentVisitorFilter"/>
+
     <!-- Pipenv -->
     <pyAddSdkProvider implementation="com.jetbrains.python.sdk.pipenv.PyAddPipEnvSdkProvider"/>
     <pythonFlavorProvider implementation="com.jetbrains.python.sdk.pipenv.PyPipEnvSdkFlavorProvider"/>
diff --git a/python/src/com/jetbrains/python/inspections/PyExpressionCodeFragmentVisitorFilter.kt b/python/src/com/jetbrains/python/inspections/PyExpressionCodeFragmentVisitorFilter.kt
new file mode 100644
index 000000000000..9e6630618cb4
--- /dev/null
+++ b/python/src/com/jetbrains/python/inspections/PyExpressionCodeFragmentVisitorFilter.kt
@@ -0,0 +1,18 @@
+// Copyright 2000-2019 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license that can be found in the LICENSE file.
+package com.jetbrains.python.inspections
+
+import com.intellij.psi.PsiFile
+import com.jetbrains.python.psi.PyElementVisitor
+import com.jetbrains.python.psi.PyExpressionCodeFragment
+import com.jetbrains.python.psi.PythonVisitorFilter
+
+class PyExpressionCodeFragmentVisitorFilter : PythonVisitorFilter {
+  override fun isSupported(visitorClass: Class<out PyElementVisitor>, file: PsiFile): Boolean {
+    if (file is PyExpressionCodeFragment) {
+      if (visitorClass == PyPep8Inspection::class.java) {
+        return false
+      }
+    }
+    return true
+  }
+}
\ No newline at end of file