Source code for spicerack

"""Spicerack package."""
from logging import Logger, getLogger
from pathlib import Path
from socket import gethostname
from typing import TYPE_CHECKING, Any, Callable, Dict, Optional, Sequence

from git import Repo
from pkg_resources import DistributionNotFound, get_distribution
from wmflib import requests
from wmflib.actions import ActionsDict
from wmflib.config import load_ini_config, load_yaml_config
from wmflib.dns import Dns
from wmflib.interactive import get_username
from wmflib.phabricator import Phabricator, create_phabricator
from wmflib.prometheus import Prometheus, Thanos

from spicerack._log import irc_logger
from spicerack.administrative import Reason
from spicerack.alerting import AlertingHosts
from spicerack.alertmanager import AlertmanagerHosts
from spicerack.confctl import Confctl, ConftoolEntity
from spicerack.debmonitor import Debmonitor
from spicerack.dhcp import DHCP
from spicerack.dnsdisc import Discovery
from spicerack.elasticsearch_cluster import ElasticsearchClusters, create_elasticsearch_clusters
from spicerack.exceptions import SpicerackError
from spicerack.ganeti import Ganeti
from spicerack.icinga import ICINGA_DOMAIN, IcingaHosts
from spicerack.interactive import get_management_password
from spicerack.ipmi import Ipmi
from spicerack.k8s import Kubernetes
from spicerack.kafka import Kafka
from spicerack.mediawiki import MediaWiki
from spicerack.mysql import Mysql
from spicerack.mysql_legacy import MysqlLegacy
from spicerack.netbox import Netbox, NetboxServer
from spicerack.puppet import PuppetHosts, PuppetMaster, get_puppet_ca_hostname
from spicerack.redfish import Redfish, RedfishDell
from spicerack.redis_cluster import RedisCluster
from spicerack.remote import Remote, RemoteHosts
from spicerack.reposync import RepoSync
from spicerack.service import Catalog
from spicerack.toolforge.etcdctl import EtcdctlController
from spicerack.typing import TypeHosts

if TYPE_CHECKING:  # Imported only during type checking, prevents cyclic imports at runtime
    from spicerack._menu import BaseItem  # pragma: no cover


logger = getLogger(__name__)

try:
    __version__: str = get_distribution("wikimedia-spicerack").version  # Must be the same used as 'name' in setup.py
    """:py:class:`str`: the version of the current Spicerack module."""
except DistributionNotFound:  # pragma: no cover - this should never happen during tests
    pass  # package is not installed


[docs]class Spicerack: # pylint: disable=too-many-instance-attributes """Spicerack service locator.""" def __init__( self, *, verbose: bool = False, dry_run: bool = True, cumin_config: str = "/etc/cumin/config.yaml", cumin_installer_config: str = "/etc/cumin/config-installer.yaml", conftool_config: str = "/etc/conftool/config.yaml", conftool_schema: str = "/etc/conftool/schema.yaml", debmonitor_config: str = "/etc/debmonitor.conf", spicerack_config_dir: str = "/etc/spicerack", http_proxy: str = "", get_cookbook_callback: Optional[Callable[["Spicerack", str, Sequence[str]], Optional["BaseItem"]]] = None, ) -> None: """Initialize the service locator for the Spicerack library. Arguments: verbose (bool, optional): whether to set the verbose mode. dry_run (bool, optional): whether this is a DRY-RUN. cumin_config (str, optional): the path to Cumin's configuration file. cumin_installer_config (str, optional): the path to Cumin's configuration file for the Debian installer. conftool_config (str, optional): the path to Conftool's configuration file. conftool_schema (str, optional): the path to Conftool's schema file. debmonitor_config (str, optional): the path to Debmonitor's INI configuration file. It must have at least the following schema:: [DEFAULT] server=debmonitor.example.com cert=/etc/debmonitor/ssl/cert.pem key=/etc/debmonitor/ssl/server.key spicerack_config_dir (str, optional): the path for the root configuration directory for Spicerack. Module-specific configuration will be loaded from `config_dir/module_name/`. http_proxy (str, optional): the scheme://url:port of the HTTP proxy to use for external calls. get_cookbook_callback (callable, optional): a callable to retrieve a CookbookItem to execute a cookbook from inside another cookbook. """ # Attributes self._verbose = verbose self._dry_run = dry_run self._http_proxy = http_proxy self._cumin_config = cumin_config self._cumin_installer_config = cumin_installer_config self._conftool_config = conftool_config self._conftool_schema = conftool_schema self._debmonitor_config = debmonitor_config self._spicerack_config_dir = Path(spicerack_config_dir) self._get_cookbook_callback = get_cookbook_callback self._username = get_username() self._current_hostname = gethostname() self._irc_logger = irc_logger self._confctl: Optional[Confctl] = None self._service_catalog: Optional[Catalog] = None self._management_password: str = "" self._actions = ActionsDict() @property def dry_run(self) -> bool: """Getter for the ``dry_run`` property. Returns: bool: True if the DRY-RUN mode is set, False otherwise. """ return self._dry_run @property def verbose(self) -> bool: """Getter for the ``verbose`` property. Returns: bool: True if the verbose mode is set, False otherwise. """ return self._verbose @property def username(self) -> str: """Getter for the current username. Returns: str: the name of the effective running user. """ return self._username @property def config_dir(self) -> Path: """Getter for Spicerack's configuration file directory. Returns: str: a filesystem location of configuration files. """ return self._spicerack_config_dir @property def http_proxy(self) -> str: """Getter for the HTTP PROXY to use for external calls. Returns: str: the scheme://url:port of the proxy. """ return self._http_proxy @property def requests_proxies(self) -> Optional[Dict[str, str]]: """Getter to return the HTTP proxy configuration for the Requests module. Returns: dict: with the proxies as required by Requests documentation. :py:data:`None`: if no HTTP proxy is set. See Also: https://requests.readthedocs.io/en/master/user/advanced/#proxies """ if not self._http_proxy: return None return {"http": self._http_proxy, "https": self._http_proxy} @property def irc_logger(self) -> Logger: """Getter for the ``irc_logger`` property. Returns: logging.Logger: the logger instance to write to IRC. """ return self._irc_logger @property def actions(self) -> ActionsDict: """Getter for the ``actions`` property. Returns: wmflib.actions.ActionsDict: a dictionary to log and record cookbook actions. """ return self._actions @property def icinga_master_host(self) -> RemoteHosts: """Getter for the ``icinga_master_host`` property. Returns: spicerack.remote.RemoteHosts: the instance to execute commands on the Icinga master host. """ return self.remote().query(self.dns().resolve_cname(ICINGA_DOMAIN)) @property def netbox_master_host(self) -> RemoteHosts: """Getter for the ``netbox_master_host`` property. Returns: spicerack.remote.RemoteHosts: the instance to execute commands on the Netbox master host. """ dns = self.dns() netbox_hostname = dns.resolve_ptr(dns.resolve_ips("netbox.discovery.wmnet")[0])[0] return self.remote().query(netbox_hostname) @property def management_password(self) -> str: """Getter for the ``management_password`` property. Returns: str: the management password. It will be asked to the user if not already cached by the current instance. """ if not self._management_password: self._management_password = get_management_password() return self._management_password
[docs] def run_cookbook(self, cookbook: str, args: Sequence[str] = ()) -> int: """Run another Cookbook within the current run. The other Cookbook will be executed with the current setup and will log in the same file of the current Cookbook that is running. Arguments: cookbook (str): the path to the cookbook to execute, either in Spicerack's dot notation or the relative path to the Python file to execute. args (sequence, optional): an iterable sequence of strings with the Cookbook's argument. The Cookbook will be executed with the same global arguments used for the current run. Returns: int: the exit code of the Cookbook, 0 if successful, non-zero if not. Raises: spicerack.exceptions.SpicerackError: if the get_cookbook_callback callback is not set or unable to find the cookbook with the given name. """ if self._get_cookbook_callback is None: raise SpicerackError("Unable to run other cookbooks, get_cookbook_callback is not set.") cookbook_item = self._get_cookbook_callback(self, cookbook, args) if cookbook_item is None: raise SpicerackError(f"Unable to find cookbook {cookbook}") logger.debug("Executing cookbook %s with args: %s", cookbook, args) return cookbook_item.run() or 0 # Force the return code to be 0 if the cookbook returns None
[docs] def remote(self, installer: bool = False) -> Remote: """Get a Remote instance. Arguments: installer (bool, optional): whether to use the special configuration to connect to a Debian installer or freshly re-imaged host prior to its first Puppet run. Returns: spicerack.remote.Remote: the Remote instance. """ return Remote(self._cumin_installer_config if installer else self._cumin_config, dry_run=self._dry_run)
[docs] def confctl(self, entity_name: str) -> ConftoolEntity: """Access a Conftool specific entity instance. Arguments: entity_name (str): the name of a Conftool entity. Possible values: ``node``, ``service``, ``discovery``, ``mwconfig``. Returns: spicerack.confctl.ConftoolEntity: the confctl entity instance. """ if self._confctl is None: self._confctl = Confctl( config=self._conftool_config, schema=self._conftool_schema, dry_run=self._dry_run, ) return self._confctl.entity(entity_name)
[docs] def dhcp(self, remote_hosts: RemoteHosts) -> DHCP: """Return a DHCP configuration manager for specified site. Arguments: remote_hosts (RemoteHosts instance): Hosts to operate on, which are normally install servers with dhcp. Returns: spicerack.dhcp.DHCP : A DHCP configuration instance """ return DHCP(remote_hosts, dry_run=self._dry_run)
[docs] def dns(self) -> Dns: """Get a Dns instance. Returns: wmflib.dns.Dns: a Dns instance that will use the operating system default namserver(s). """ return Dns()
[docs] def discovery(self, *records: str) -> Discovery: """Get a Discovery instance. Arguments: *records (str): arbitrary positional arguments, each one must be a Discovery DNS record name. Returns: spicerack.dnsdisc.Discovery: the pre-configured Discovery instance for the given records. """ return Discovery( self.confctl("discovery"), self.remote(), list(records), dry_run=self._dry_run, )
[docs] def kubernetes(self, group: str, cluster: str) -> Kubernetes: """Get a kubernetes client for the specified cluster. Arguments: group (str): the cluster group cluster (str): the kubernetes cluster Returns: spicerack.k8s.Kubernetes: a pre-configured kubernetes client """ return Kubernetes(group, cluster, dry_run=self._dry_run)
[docs] def mediawiki(self) -> MediaWiki: """Get a MediaWiki instance. Returns: spicerack.mediawiki.MediaWiki: the MediaWiki instance. """ return MediaWiki( self.confctl("mwconfig"), self.remote(), self._username, dry_run=self._dry_run, )
[docs] def mysql(self) -> Mysql: """Get a Mysql instance. Returns: spicerack.mysql.Mysql: the Mysql instance. """ return Mysql(dry_run=self._dry_run)
[docs] def mysql_legacy(self) -> MysqlLegacy: """Get a MysqlLegacy instance. Returns: spicerack.mysql_legacy.MysqlLegacy: the MysqlLegacy instance. """ return MysqlLegacy(self.remote(), dry_run=self._dry_run)
[docs] def redis_cluster(self, cluster: str) -> RedisCluster: """Get a RedisCluster instance. Arguments: cluster (str): the name of the cluster. Returns: spicerack.redis_cluster.RedisCluster: the cluster instance. """ return RedisCluster( cluster, self._spicerack_config_dir / "redis_cluster", dry_run=self._dry_run, )
[docs] def reposync(self, name: str) -> RepoSync: """Get a Reposync instance. Arguments: name (str): the name of the repo to sync. Returns: spicerack.reposync.RepoSync: the reposync instance. """ config = load_yaml_config(self._spicerack_config_dir / "reposync" / "config.yaml") if name not in config["repos"]: raise SpicerackError(f"Unknown repo {name}") repo_dir = Path(config["base_dir"], name) query = ",".join(config["remotes"]) remote_hosts = self.remote().query(query) if not repo_dir.is_dir(): raise SpicerackError(f"The repo directory ({repo_dir}) does not exist") repo = Repo(repo_dir) if not repo.bare: raise SpicerackError(f"The repo directory ({repo_dir}) is not a bare git repository") return RepoSync(repo, self.username, remote_hosts, dry_run=self._dry_run)
[docs] def elasticsearch_clusters( self, clustergroup: str, write_queue_datacenters: Sequence[str] ) -> ElasticsearchClusters: """Get an ElasticsearchClusters instance. Arguments: clustergroup (str): name of cluster group e.g ``search_eqiad``. write_queue_datacenters (Sequence[str]): Sequence of which core DCs to query write queues for. Returns: spicerack.elasticsearch_cluster.ElasticsearchClusters: ElasticsearchClusters instance. """ configuration = load_yaml_config(self._spicerack_config_dir / "elasticsearch" / "config.yaml") return create_elasticsearch_clusters( configuration, clustergroup, write_queue_datacenters, self.remote(), self.prometheus(), dry_run=self._dry_run, )
[docs] def admin_reason(self, reason: str, task_id: Optional[str] = None) -> Reason: """Get an administrative Reason instance. Arguments: reason (str): the reason to use to justify an administrative action. See `spicerack.administrative.Reason` for all the details. task_id (str, optional): the task ID to mention in the reason. Returns: spicerack.administrative.Reason: the administrative Reason instance. """ return Reason(reason, self._username, self._current_hostname, task_id=task_id)
[docs] def icinga_hosts(self, target_hosts: TypeHosts, *, verbatim_hosts: bool = False) -> IcingaHosts: """Get an IcingaHosts instance. Arguments: target_hosts (spicerack.typing.TypeHosts): the target hosts either as a NodeSet instance or a sequence of strings. verbatim_hosts (bool, optional): if :py:data:`True` use the hosts passed verbatim as is, if instead :py:data:`False`, the default, consider the given target hosts as FQDNs and extract their hostnames to be used in Icinga. Returns: spicerack.icinga.IcingaHosts: IcingaHosts instance. """ return IcingaHosts(self.icinga_master_host, target_hosts, verbatim_hosts=verbatim_hosts)
[docs] def puppet(self, remote_hosts: RemoteHosts) -> PuppetHosts: """Get a PuppetHosts instance for the given remote hosts. Arguments: remote_hosts (spicerack.remote.RemoteHosts): the instance with the target hosts. Returns: spicerack.puppet.PuppetHosts: the instance to manage Puppet on the target hosts. """ return PuppetHosts(remote_hosts)
[docs] def puppet_master(self) -> PuppetMaster: """Get a PuppetMaster instance to manage hosts and certificates from a Puppet master. Returns: spicerack.puppet.PuppetMaster: the instance to manage Puppet hosts and certificates. """ return PuppetMaster(self.remote().query(get_puppet_ca_hostname()))
[docs] def ipmi(self, mgmt_fqdn: str) -> Ipmi: """Get an Ipmi instance to send remote IPMI commands to management consoles. Arguments: mgmt_fqdn (str): the management console FQDN to target. Returns: spicerack.ipmi.Ipmi: the instance to run ipmitool commands. """ return Ipmi(mgmt_fqdn, self.management_password, dry_run=self._dry_run)
[docs] def phabricator(self, bot_config_file: str, section: str = "phabricator_bot") -> Phabricator: """Get a Phabricator instance to interact with a Phabricator website. Arguments: bot_config_file (str): the path to the configuration file for the Phabricator bot, with the following structure:: [section_name] host = https://phabricator.example.com/api/ username = phab-bot token = api-12345 section (str, optional): the name of the section of the configuration file where to find the required parameters. Returns: wmflib.phabricator.Phabricator: the instance. """ # Allow to specify the configuration file as opposed to other methods so that different clients can use # different Phabricator BOT accounts, potentially with different permissions. return create_phabricator(bot_config_file, section=section, dry_run=self._dry_run)
[docs] def prometheus(self) -> Prometheus: """Get a Prometheus instance. Returns: wmflib.prometheus.Prometheus: Prometheus instance. """ return Prometheus()
[docs] def thanos(self) -> Thanos: """Get a Thanos instance. Returns: wmflib.prometheus.Thanos: Thanos instance. """ return Thanos()
[docs] def debmonitor(self) -> Debmonitor: """Get a Debmonitor instance to interact with a Debmonitor website. Returns: spicerack.debmonitor.Debmonitor: the instance. Raises: KeyError: if any configuration option is missing. """ options = load_ini_config(self._debmonitor_config).defaults() return Debmonitor(options["server"], options["cert"], options["key"], dry_run=self._dry_run)
[docs] def ganeti(self) -> Ganeti: """Get an instance to interact with Ganeti. Returns: spicerack.ganeti.Ganeti: the instance Raises: KeyError: If the configuration file does not contain the correct keys. """ configuration = load_yaml_config(self._spicerack_config_dir / "ganeti" / "config.yaml") return Ganeti( configuration["username"], configuration["password"], configuration["timeout"], self.remote(), )
[docs] def netbox(self, *, read_write: bool = False) -> Netbox: """Get a Netbox instance to interact with Netbox's API. Arguments: read_write (bool, optional): whether to use a read-write token. Returns: spicerack.netbox.Netbox: the instance """ config = load_yaml_config(self._spicerack_config_dir / "netbox" / "config.yaml") if read_write and not self._dry_run: token = config["api_token_rw"] else: token = config["api_token_ro"] return Netbox(config["api_url"], token, dry_run=self._dry_run)
[docs] def netbox_server(self, hostname: str, *, read_write: bool = False) -> NetboxServer: """Get a NetboxServer instance to interact with a server in Netbox, both physical and virtual. Arguments: hostname (str): the hostname (not FQDN) of the server to manage. read_write (bool, optional): whether to use a read-write token. Raises: spicerack.netbox.NetboxError: if unable to find or load the server. Returns: spicerack.netbox.NetboxServer: the NetboxServer instance. """ return self.netbox(read_write=read_write).get_server(hostname)
[docs] def requests_session(self, name: str, **kwargs: Any) -> requests.Session: """Return a new requests Session with timeout and retry logic. Params: according to :py:func:`wmflib.requests.http_session`. Returns: requests.Session: the pre-configured session. """ name = f"Spicerack/{__version__} {name}" return requests.http_session(name, **kwargs)
[docs] def etcdctl(self, *, remote_host: RemoteHosts) -> EtcdctlController: """Add etcdctl control capabilities to the given RemoteHost. Params: remote_host: Single remote host (that should map one single host) to add capabilities to. Returns: A wrapped RemoteHost with the etcdctl control related methods. """ return EtcdctlController(remote_host=remote_host)
[docs] def kafka(self) -> Kafka: """Get an instance to interact with Kafka. Returns: spicerack.kafka.Kafka: the instance Raises: KeyError: If the configuration file does not contain the correct keys. """ configuration = load_yaml_config(self._spicerack_config_dir / "kafka" / "config.yaml") return Kafka(kafka_config=configuration, dry_run=self._dry_run)
[docs] def redfish(self, mgmt_fqdn: str, username: str, password: str = "") -> Redfish: # nosec """Get an instance to talk to the Redfish API of a physical server. Notes: At the moment only Dell hardware is supported. Arguments: mgmt_fqdn (str): the management console FQDN to target. username (str): the username for the management console (usually 'root'). password (str, optional): the password for the management console for the given user. If empty or not provided would use the production management password and ask the user for it if not already in memory. Returns: spicerack.redfish.Redfish: the instance. """ if not password: password = self.management_password # TODO: generalize when support for additional vendors will be added. return RedfishDell(mgmt_fqdn, username, password, dry_run=self._dry_run)
[docs] def alertmanager_hosts(self, target_hosts: TypeHosts, *, verbatim_hosts: bool = False) -> AlertmanagerHosts: """Get an AlertmanagerHosts instance. Arguments: target_hosts (spicerack.typing.TypeHosts): the target hosts either as a NodeSet instance or a sequence of strings. verbatim_hosts (bool, optional): if :py:data:`True` use the hosts passed verbatim as is, if instead :py:data:`False`, the default, consider the given target hosts as FQDNs and extract their hostnames to be used in Icinga. Returns: spicerack.alertmanager.AlertmanagerHosts: AlertmanagerHosts instance. """ return AlertmanagerHosts(target_hosts, verbatim_hosts=verbatim_hosts, dry_run=self._dry_run)
[docs] def alerting_hosts(self, target_hosts: TypeHosts, *, verbatim_hosts: bool = False) -> AlertingHosts: """Get an AlertingHosts instance. Arguments: target_hosts (spicerack.typing.TypeHosts): the target hosts either as a NodeSet instance or a sequence of strings. verbatim_hosts (bool, optional): if :py:data:`True` use the hosts passed verbatim as is, if instead :py:data:`False`, the default, consider the given target hosts as FQDNs and extract their hostnames to be used in Icinga. Returns: spicerack.alerting.AlertingHosts: AlertingHosts instance. """ return AlertingHosts( self.alertmanager_hosts(target_hosts, verbatim_hosts=verbatim_hosts), self.icinga_hosts(target_hosts, verbatim_hosts=verbatim_hosts), )
[docs] def service_catalog(self) -> Catalog: """Get a Catalog instance that reflects Puppet's service::catalog hieradata variable. Returns: spicerack.service.Catalog: the service catalog instance. """ if self._service_catalog is None: config = load_yaml_config(self._spicerack_config_dir / "service" / "service.yaml") self._service_catalog = Catalog(config, self.confctl("discovery"), self.remote(), dry_run=self._dry_run) return self._service_catalog