PY-50642 Use a more generic name for PyVersionAwareElementVisitor so it can be used with arbitrary if-conditions

GitOrigin-RevId: 55f6afe70f9b8ed0a2170c7279b5a57252ce865f
This commit is contained in:
Petr
2025-11-28 19:07:18 +00:00
committed by intellij-monorepo-bot
parent 87e2e9f316
commit 90fac13e8c
5 changed files with 8 additions and 11 deletions
@@ -7,11 +7,8 @@ import com.jetbrains.python.psi.PyIfStatement
import com.jetbrains.python.psi.PyRecursiveElementVisitor
import org.jetbrains.annotations.ApiStatus
/**
* @see [Version and Platform Checks](https://typing.readthedocs.io/en/latest/source/stubs.html.version-and-platform-checks)
*/
@ApiStatus.Internal
open class PyVersionAwareElementVisitor(languageLevel: LanguageLevel?) : PyRecursiveElementVisitor() {
open class PyTypeCheckedElementVisitor(languageLevel: LanguageLevel?) : PyRecursiveElementVisitor() {
private val version = languageLevel?.let { Version(it.majorVersion, it.minorVersion, 0) }
override fun visitPyIfStatement(node: PyIfStatement) {
@@ -9,7 +9,7 @@ import com.jetbrains.python.psi.PyFunction
import org.jetbrains.annotations.ApiStatus
@ApiStatus.Internal
abstract class PyVersionAwareTopLevelElementVisitor(languageLevel: LanguageLevel) : PyVersionAwareElementVisitor(languageLevel) {
abstract class PyTypeCheckedTopLevelElementVisitor(languageLevel: LanguageLevel) : PyTypeCheckedElementVisitor(languageLevel) {
override fun visitPyElement(node: PyElement) {
super.visitPyElement(node)
checkAddElement(node)