@Override
  protected PyExpression getCondition(PyStatement statement) {
    if (statement instanceof PyWhileStatement) {
      return ((PyWhileStatement)statement).getWhilePart().getCondition();
    }
    return null;
  }
@Override
  protected PyExpression getCondition(PyStatement statement)