Current File : //proc/self/root/var/opt/nydus/ops/primordial/service/base_daemon.py
# -*- coding: utf-8 -*-
import logging
from subprocess import PIPE, call, CalledProcessError
from typing import Any, Dict # pylint: disable=W0611 # noqa: F401
LOGGER = logging.getLogger(__name__)
class BaseDaemon:
"""A base implementation of a daemon"""
def __init__(self, name: str, command_fmt: str) -> None:
self.name = name
self.command_fmt = command_fmt
def _run(self, action: str, check_return_code: bool = True, pipes: bool = True) -> Any:
"""Run an action for the daemon.
:param action: The action to be run
:param check_return_code: Whether to error on non-zero return codes
:param pipes:
:raises CalledProcessError: If `check_return_code` is True, and a non-zero return code occurs
"""
LOGGER.info('%sing %s', action, self.name)
kwargs = {'universal_newlines': True, 'timeout': 10} # type: Dict[str, Any]
if pipes:
kwargs.update({'stdout': PIPE, 'stderr': PIPE})
cmd = self.command_fmt.format(name=self.name, action=action)
result = call(cmd.split(), **kwargs)
LOGGER.info(result)
if check_return_code:
if result != 0:
raise CalledProcessError(returncode=result, cmd=cmd)
return result
def status(self) -> Any:
"""Check the status of the daemon."""
return self._run('status', check_return_code=False)
def start(self) -> Any:
"""Start the daemon."""
# Must not connect pipes to stdin/out, or subprocess will never
# complete, hanging on communicate() call.
return self._run('start', pipes=False)
def stop(self) -> Any:
"""Stop the daemon."""
return self._run('stop')
def restart(self) -> Any:
"""Restart the daemon."""
# Must not connect pipes to stdin/out, or subprocess will never
# complete, hanging on communicate() call.
return self._run('restart', pipes=False)
Mini Shell