icinga2-checks/checker/http.py

import sys
from time import sleep

import requests

from . import nagios
from .result import print_icinga2_check_status


def fetch_with_retry(url, method: str = 'get', retries=3, delay=1, **kwargs):
    """
    Wrapper function for requests.get() with a retry mechanism.

    :param method: HTTP request type: get, post
    :param url: URL to send the GET request
    :param retries: Number of retries in case of HTTP failures (default: 3)
    :param delay: Time delay between retries in seconds (default: 1)
    :param kwargs: Additional keyword arguments for requests.get()
    :return: Response object
    """

    for i in range(retries):
        try:
            if method == 'get':
                response = requests.get(url, **kwargs)
            elif method == 'post':
                response = requests.post(url, **kwargs)
            else:
                raise ValueError('Invalid method! Must be get or post.')
            response.raise_for_status()
            return response
        except requests.exceptions.RequestException as e:
            if i == retries - 1:
                # raise e
                print_icinga2_check_status(f'HTTP request failed after {i + 1} retries: {url}\n{e}', nagios.STATE_UNKNOWN)
                sys.exit(nagios.STATE_UNKNOWN)
            sleep(delay)