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