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_CRIT) sys.exit(nagios.STATE_CRIT) sleep(delay)