@Nullable
  default PyAstExpression getCondition() {
    ASTNode n = getNode().findChildByType(PythonDialectsTokenSetProvider.getInstance().getExpressionTokens());
    if (n != null) {
      return (PyAstExpression)n.getPsi();
    }
    return null;
  }
@Nullable
  default PyAstExpression getCondition()