[python] Add methods visitPyKeyValueExpression to PyElementVisitor and PyAstElementVisitor

GitOrigin-RevId: 247a02160e18c93a00b1630e2ed70cea6dcefbbf

python/python-ast/src/com/jetbrains/python/ast/PyAstElementVisitor.kt

@@ -381,4 +381,8 @@ open class PyAstElementVisitor : PsiElementVisitor() {
   open fun visitPyTypeParameterList(node: PyAstTypeParameterList) {
     visitPyElement(node)
   }
+
+  open fun visitPyKeyValueExpression(node: PyAstKeyValueExpression) {
+    visitPyExpression(node)
+  }
 }

python/python-ast/src/com/jetbrains/python/ast/PyAstKeyValueExpression.java

@@ -20,4 +20,9 @@ public interface PyAstKeyValueExpression extends PyAstExpression {
   default PyAstExpression getValue() {
     return PsiTreeUtil.getNextSiblingOfType(getKey(), PyAstExpression.class);
   }
+
+  @Override
+  default void acceptPyVisitor(PyAstElementVisitor pyVisitor) {
+    pyVisitor.visitPyKeyValueExpression(this);
+  }
 }

python/python-psi-api/src/com/jetbrains/python/psi/PyElementVisitor.java

@@ -397,4 +397,8 @@ public class PyElementVisitor extends PsiElementVisitor {
   public void visitPyTypeParameterList(@NotNull PyTypeParameterList node) {
     visitPyElement(node);
   }
+
+  public void visitPyKeyValueExpression(@NotNull PyKeyValueExpression node) {
+    visitPyExpression(node);
+  }
 }

python/python-psi-impl/src/com/jetbrains/python/psi/impl/PyKeyValueExpressionImpl.java

@@ -2,6 +2,7 @@
 package com.jetbrains.python.psi.impl;
 
 import com.intellij.lang.ASTNode;
+import com.jetbrains.python.psi.PyElementVisitor;
 import com.jetbrains.python.psi.PyExpression;
 import com.jetbrains.python.psi.PyKeyValueExpression;
 import com.jetbrains.python.psi.types.PyTupleType;
@@ -17,6 +18,11 @@ public class PyKeyValueExpressionImpl extends PyElementImpl implements PyKeyValu
     super(astNode);
   }
 
+  @Override
+  protected void acceptPyVisitor(PyElementVisitor pyVisitor) {
+    pyVisitor.visitPyKeyValueExpression(this);
+  }
+
   @Override
   public PyType getType(@NotNull TypeEvalContext context, @NotNull TypeEvalContext.Key key) {
     final PyType keyType = context.getType(getKey());