Viewing file:  debconf.py (6.61 KB) -rw-r--r--
Select action/file-type:
 (+) |  (+) |  (+) | Code (+) | Session (+) |  (+) | SDB (+) |  (+) |  (+) |  (+) |  (+) |  (+) |
# 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 print_function
import sys, os
import errno
import re
import subprocess
import fcntl
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'))
"""
def __init__(self, title=None, read=None, write=None, run_frontend=False):
for command in ('capb set reset title input beginblock endblock go get'
' register unregister subst fset fget previous_module'
' 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):
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):
setattr(self, command,
lambda *args, **kw: self.command(command, *args, **kw))
def command(self, command, *params):
command = command.upper()
self.write.write("%s %s\n" % (command, ' '.join(map(str, params))))
self.write.flush()
while True:
try:
resp = self.read.readline().rstrip('\n')
break
except IOError 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):
self.write.write('STOP\n')
self.write.flush()
def forceInput(self, priority, question):
try:
self.input(priority, question)
return 1
except DebconfError as e:
if e.args[0] != 30:
raise
return 0
def getBoolean(self, question):
result = self.get(question)
return result == 'true'
def getString(self, question):
return self.get(question)
def __enter__(self):
return self
def __exit__(self, exc_type, exc_value, exc_tb):
self.stop()
class DebconfCommunicator(Debconf, object):
def __init__(self, owner, title=None, cloexec=False):
args = ['debconf-communicate', '-fnoninteractive', owner]
self.dccomm = subprocess.Popen(
args, stdin=subprocess.PIPE, stdout=subprocess.PIPE,
close_fds=True, universal_newlines=True)
super(DebconfCommunicator, self).__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):
if self.dccomm 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):
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():
if 'DEBIAN_HAS_FRONTEND' not in os.environ:
os.environ['PERL_DL_NONLAZY']='1'
os.execv(_frontEndProgram, [_frontEndProgram, sys.executable]+sys.argv)
if __name__ == '__main__':
runFrontEnd()
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)
|