# Copyright:
# Moshe Zadka (c) 2002
# Canonical Ltd. (c) 2005 (DebconfCommunicator)
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions
# are met:
# 1. Redistributions of source code must retain the above copyright
# notice, this list of conditions and the following disclaimer.
# 2. Redistributions in binary form must reproduce the above copyright
# notice, this list of conditions and the following disclaimer in the
# documentation and/or other materials provided with the distribution.
#
# THIS SOFTWARE IS PROVIDED BY AUTHORS AND CONTRIBUTORS ``AS IS'' AND
# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHORS OR CONTRIBUTORS BE LIABLE
# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
# OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
# SUCH DAMAGE.
from __future__ import annotations
import errno
import fcntl
import os
import re
import subprocess
import sys
from types import TracebackType
from typing import IO, Protocol
class Command(Protocol):
def __call__(self, *params: str | int) -> str:
...
class DebconfError(Exception):
pass
LOW, MEDIUM, HIGH, CRITICAL = "low", "medium", "high", "critical"
class Debconf:
"""A class that speaks the debconf protocol.
The simplest way to use this is as a context manager:
import debconf
with debconf.Debconf(run_frontend=True) as db:
print(db.get('debconf/frontend'))
Note that this will send the STOP command on exiting the context
manager, so you shouldn't expect to be able to use the same frontend
again after this. If you need to do that, then you should instantiate
the class directly instead:
import debconf
db = debconf.Debconf(run_frontend=True)
print(db.get('debconf/frontend'))
"""
beginblock: Command
capb: Command
endblock: Command
exist: Command
fset: Command
get: Command
go: Command
info: Command
input: Command
progress: Command
purge: Command
register: Command
reset: Command
set: Command
settitle: Command
subst: Command
title: Command
unregister: Command
version_: Command
visible: Command
def __init__(
self,
title: str | None = None,
read: IO[str] | None = None,
write: IO[str] | None = None,
run_frontend: bool = False,
) -> None:
for command in (
"capb set reset title input beginblock endblock go get"
" register unregister subst fset fget"
" visible purge metaget exist version_ settitle"
" info progress data"
).split():
self.setCommand(command)
self.read = read or sys.stdin
self.write = write or sys.stdout
sys.stdout = sys.stderr
if run_frontend:
runFrontEnd()
self.setUp(title)
def setUp(self, title: str | None) -> None:
self.version = self.version_(2)
if self.version[:2] != "2.":
raise DebconfError(256, "wrong version: %s" % self.version)
self.capabilities = self.capb().split()
if title:
self.title(title)
def setCommand(self, command: str) -> None:
setattr(
self,
command,
lambda *args, **kw: self.command(command, *args, **kw),
)
def command(self, command: str, *params: str | int) -> str:
if command == "version_":
command = "version"
command = command.upper()
self.write.write("{} {}\n".format(command, " ".join(map(str, params))))
self.write.flush()
while True:
try:
resp = self.read.readline().rstrip("\n")
break
except OSError as e:
if e.errno == errno.EINTR:
continue
else:
raise
if " " in resp:
status_, data = resp.split(" ", 1)
else:
status_, data = resp, ""
status = int(status_)
if status == 0:
return data
elif status == 1: # unescaped data
unescaped = ""
for chunk in re.split(r"(\\.)", data):
if chunk.startswith("\\") and len(chunk) == 2:
if chunk[1] == "n":
unescaped += "\n"
else:
unescaped += chunk[1]
else:
unescaped += chunk
return unescaped
else:
raise DebconfError(status, data)
def stop(self) -> None:
self.write.write("STOP\n")
self.write.flush()
def forceInput(self, priority: str, question: str) -> int:
try:
self.input(priority, question)
return 1
except DebconfError as e:
if e.args[0] != 30:
raise
return 0
def getBoolean(self, question: str) -> bool:
result = self.get(question)
return result == "true"
def getString(self, question: str) -> str:
return self.get(question)
def __enter__(self) -> Debconf:
return self
def __exit__(
self,
exc_type: type[BaseException] | None,
exc_value: BaseException | None,
exc_tb: TracebackType | None,
) -> None:
self.stop()
class DebconfCommunicator(Debconf):
def __init__(
self, owner: str, title: str | None = None, cloexec: bool = False
) -> None:
args = ["debconf-communicate", "-fnoninteractive", owner]
self.dccomm: subprocess.Popen[str] | None = subprocess.Popen(
args,
stdin=subprocess.PIPE,
stdout=subprocess.PIPE,
close_fds=True,
universal_newlines=True,
)
super().__init__(
title=title, read=self.dccomm.stdout, write=self.dccomm.stdin
)
if cloexec:
fcntl.fcntl(self.read.fileno(), fcntl.F_SETFD, fcntl.FD_CLOEXEC)
fcntl.fcntl(self.write.fileno(), fcntl.F_SETFD, fcntl.FD_CLOEXEC)
def shutdown(self) -> None:
if self.dccomm is not None:
assert self.dccomm.stdin is not None
assert self.dccomm.stdout is not None
self.dccomm.stdin.close()
self.dccomm.stdout.close()
self.dccomm.wait()
self.dccomm = None
# Don't rely on this; call .shutdown() explicitly.
def __del__(self) -> None:
try:
self.shutdown()
except AttributeError:
pass
if (
"DEBCONF_USE_CDEBCONF" in os.environ
and os.environ["DEBCONF_USE_CDEBCONF"] != ""
):
_frontEndProgram = "/usr/lib/cdebconf/debconf"
else:
_frontEndProgram = "/usr/share/debconf/frontend"
def runFrontEnd(*, pass_sys_executable: bool = False) -> None:
if "DEBIAN_HAS_FRONTEND" not in os.environ:
os.environ["PERL_DL_NONLAZY"] = "1"
args = [_frontEndProgram]
if pass_sys_executable:
args.append(sys.executable)
args.extend(sys.argv)
os.execv(_frontEndProgram, args)
if __name__ == "__main__":
runFrontEnd(pass_sys_executable=True)
db = Debconf()
db.forceInput(CRITICAL, "bsdmainutils/calendar_lib_is_not_empty")
db.go()
less = db.getBoolean("less/add_mime_handler")
aptlc = db.getString("apt-listchanges/email-address")
db.stop()
print(db.version)
print(db.capabilities)
print(less)
print(aptlc)
|