Viewing file:  CommandNotFound.py (6.77 KB) -rw-r--r--
Select action/file-type:
 (+) |  (+) |  (+) | Code (+) | Session (+) |  (+) | SDB (+) |  (+) |  (+) |  (+) |  (+) |  (+) |
# (c) Zygmunt Krynicki 2005, 2006
# Licensed under GPL, see COPYING for the whole text
import sys, os, os.path, gdbm, posix, grp
from gettext import lgettext as _
def _guessUserLocale():
msg = os.getenv("LC_MESSAGES") or os.getenv("LANG")
if msg:
return msg.split("_", 1)[0]
else:
return None
class BinaryDatabase:
def __init__(self, filename):
self.db = None
if filename.endswith(".db"):
try:
self.db = gdbm.open(filename, "r")
except gdbm.error, err:
print >>sys.stderr, "Unable to open binary database %s: %s" % (filename, err)
def lookup(self, key):
if self.db and self.db.has_key(key):
return self.db[key]
else:
return None
class FlatDatabase:
def __init__(self, filename):
self.rows = []
dbfile = file(filename)
for line in (line.strip() for line in dbfile):
self.rows.append(line.split("|"))
dbfile.close()
def lookup(self, column, text):
result = []
for row in self.rows:
if row[column] == text:
result.append(row)
return result
def createColumnByCallback(self, cb, column):
for row in self.rows:
row.append(cb(row[column]))
def lookupWithCallback(self, column, cb, text):
result = []
for row in self.rows:
if cb(row[column],text):
result.append(row)
return result
class ProgramDatabase:
(PACKAGE, BASENAME_PATH) = range(2)
def __init__(self, filename):
basename = os.path.basename(filename)
(self.arch, self.component) = basename.split(".")[0].split("-")
self.db = BinaryDatabase(filename)
def lookup(self, command):
result = self.db.lookup(command)
if result:
return result.split("|")
else:
return []
class CommandNotFound:
programs_dir = "programs.d"
prefixes = ("/bin", "/usr/bin", "/usr/local/bin", "/sbin", "/usr/sbin", "/usr/local/sbin", "/usr/games")
def __init__(self, data_dir=os.sep.join(('/','usr','share','command-not-found'))):
self.programs = []
self.sources_list = self._getSourcesList()
for filename in os.listdir(os.path.sep.join([data_dir, self.programs_dir])):
self.programs.append(ProgramDatabase(os.path.sep.join([data_dir, self.programs_dir, filename])))
try:
self.user_can_sudo = grp.getgrnam("admin")[2] in posix.getgroups()
except KeyError:
self.user_can_sudo = False
def getPackages(self, command):
result = set()
for db in self.programs:
result.update([(pkg,db.component) for pkg in db.lookup(command)])
return list(result)
def getBlacklist(self):
try:
blacklist = file(os.sep.join((os.getenv("HOME", "/root"), ".command-not-found.blacklist")))
return [line.strip() for line in blacklist if line.strip() != ""]
except IOError:
return []
else:
blacklist.close()
def _getSourcesList(self):
try:
import apt_pkg
from aptsources.sourceslist import SourcesList
apt_pkg.init()
except (SystemError, ImportError), e:
return []
sources_list = set([])
for source in SourcesList():
if not source.disabled and not source.invalid:
for component in source.comps:
sources_list.add(component)
return sources_list
def advise(self, command, ignore_installed=False):
def _in_prefix(prefix, command):
" helper that returns if a command is found in the given prefix "
return (os.path.exists(os.path.join(prefix, command)) and
not os.path.isdir(os.path.join(prefix, command)))
if command.startswith("/"):
if os.path.exists(command):
prefixes = [os.path.dirname(command)]
else:
prefixes = []
else:
prefixes = [prefix for prefix in self.prefixes if _in_prefix(prefix, command)]
if prefixes and not ignore_installed:
if len(prefixes) == 1:
print _("Command '%(command)s' is available in '%(place)s'") % {"command": command, "place": os.path.join(prefixes[0], command)}
else:
print _("Command '%(command)s' is available in the following places") % {"command": command}
for prefix in prefixes:
print " * %s" % os.path.join(prefix, command)
missing = list(set(prefixes) - set(os.getenv("PATH").split(":")))
if len(missing) > 0:
print _("The command could not be located because '%s' is not included in the PATH environment variable.") % ":".join(missing)
if "sbin" in ":".join(missing):
print _("This is most likely caused by the lack of administrative priviledges associated with your user account.")
return False
if command in self.getBlacklist():
return False
packages = self.getPackages(command)
if len(packages) == 1:
print >>sys.stderr, _("The program '%s' is currently not installed. ") % command,
if posix.geteuid() == 0:
print >>sys.stderr, _("You can install it by typing:")
print >>sys.stderr, "apt-get install %s" % packages[0][0]
elif self.user_can_sudo:
print >>sys.stderr, _("You can install it by typing:")
print >>sys.stderr, "sudo apt-get install %s" % packages[0][0]
else:
print >>sys.stderr, _("To run '%(command)s' please ask your administrator to install the package '%(package)s'") % {'command': command, 'package': packages[0][0]}
if not packages[0][1] in self.sources_list:
print >>sys.stderr, _("You will have to enable the component called '%s'") % packages[0][1]
elif len(packages) > 1:
print >>sys.stderr, _("The program '%s' can be found in the following packages:") % command
for package in packages:
if package[1] in self.sources_list:
print >>sys.stderr, " * %s" % package[0]
else:
print >>sys.stderr, " * %s" % package[0] + " (" + _("You will have to enable component called '%s'") % package[1] + ")"
if posix.geteuid() == 0:
print >>sys.stderr, _("Try: %s <selected package>") % "apt-get install"
elif self.user_can_sudo:
print >>sys.stderr, _("Try: %s <selected package>") % "sudo apt-get install"
else:
print >>sys.stderr, _("Ask your administrator to install one of them")
return len(packages) > 0
|