apt.py

from subprocess import CalledProcessError, check_call, DEVNULL
from typing import Any, List, Sequence

import dotbot


class Apt(dotbot.Plugin):
    def can_handle(self, directive: str) -> bool:
        return directive == "apt"

    def handle(self, directive: str, packages: List[str]) -> bool:
        success = self._run(["apt", "update"], "Updating APT") \
                  and self._run(["apt", "install", "-y"] + packages,
                                "Installing the APT packages: {}".format(", ".join(packages)))

        if success:
            self._log.info("APT packages installed successfully")

        return success

    def _run(self, command: Sequence[Any], low_info: str) -> bool:
        self._log.lowinfo(low_info)
        try:
            try:
                # Try running the command without sudo first
                check_call(command, stdout=DEVNULL, stderr=DEVNULL)
            except CalledProcessError as e:
                sudo_command = ["sudo"] + command
                check_call(sudo_command, stdout=DEVNULL, stderr=DEVNULL)
            return True
        except CalledProcessError as e:
            self._log.error(e)
            return False