Source code for mac_maker.utilities.sudo

"""SUDO password representation."""

import getpass
import os
import shlex
import subprocess

from mac_maker import config


class SUDO:
  """SUDO password representation."""

  def __init__(self) -> None:
    self.sudo_password = os.getenv(config.ENV_ANSIBLE_BECOME_PASSWORD, None)

  def prompt_for_sudo(self) -> None:
    """Prompt the user to enter the system's SUDO password."""

    if not self.sudo_password:

      while True:

        entered_password = getpass.getpass(config.SUDO_PROMPT)

        with subprocess.Popen(
            shlex.split(config.SUDO_CHECK_COMMAND),
            stdin=subprocess.PIPE,
            stderr=subprocess.PIPE,
        ) as check_sudo:
          check_sudo.communicate((entered_password + '\n').encode('utf-8'))

          if not check_sudo.returncode:
            self.sudo_password = entered_password
            os.environ[config.ENV_ANSIBLE_BECOME_PASSWORD] = entered_password
            return