Source code for spicerack.debmonitor

"""Debmonitor module."""
import logging

import requests
from wmflib.requests import http_session

from spicerack.exceptions import SpicerackError

logger = logging.getLogger(__name__)

[docs]class DebmonitorError(SpicerackError): """Custom exception class for errors of the Debmonitor class."""
[docs]class Debmonitor: """Class to interact with a Debmonitor website.""" def __init__(self, host: str, cert: str, key: str, dry_run: bool = True) -> None: """Initialize the instance. Arguments: host (str): the hostname of the Debmonitor server (without protocol). cert (str): the path to the TLS certificate to use to authenticate on Debmonitor. key (str): the path to the TLS key to use to authenticate on Debmonitor. dry_run (bool, optional): whether this is a DRY-RUN. """ self._base_url: str = "https://{host}".format(host=host) self._cert = cert self._key = key self._dry_run = dry_run self._http_session = http_session(".".join((self.__module__, self.__class__.__name__)))
[docs] def host_delete(self, hostname: str) -> None: """Remove a host and all its packages from Debmonitor. Arguments: host (str): the FQDN of the host to remove from Debmonitor. Raises: spicerack.debmonitor.DebmonitorError: on failure to delete. It doesn't raise if the host is already absent in Debmonitor. """ if self._dry_run: logger.debug("Skip removing host %s from Debmonitor in DRY-RUN", hostname) return url = "{base}/hosts/{host}".format(base=self._base_url, host=hostname) response = self._http_session.delete(url, cert=(self._cert, self._key)) if response.status_code ==["no_content"]:"Removed host %s from Debmonitor", hostname) elif response.status_code ==["not_found"]:"Host %s already missing on Debmonitor", hostname) else: raise DebmonitorError( "Unable to remove host {host} from Debmonitor, got: {code} {msg}".format( host=hostname, code=response.status_code, msg=response.reason ) )