Files
entropy/client/solo/utils.py
2012-09-12 16:40:11 +02:00

496 lines
17 KiB
Python

# -*- coding: utf-8 -*-
"""
@author: Fabio Erculiani <lxnay@sabayon.org>
@contact: lxnay@sabayon.org
@copyright: Fabio Erculiani
@license: GPL-2
B{Entropy Command Line Client}.
"""
import os
import codecs
import subprocess
from entropy.const import etpConst, const_convert_to_unicode
from entropy.i18n import _
from entropy.output import print_generic, darkgreen, purple, blue, \
brown, bold, red, darkred, teal, decolorize
import entropy.dep
def read_client_release():
"""
Read Entropy Command Line Client release.
@rtype: None
@return: None
"""
# handle Entropy Version
revision_file = "../client/revision"
if not os.path.isfile(revision_file):
revision_file = os.path.join(etpConst['installdir'],
'client/revision')
if os.path.isfile(revision_file) and \
os.access(revision_file, os.R_OK):
enc = etpConst['conf_encoding']
with codecs.open(revision_file, "r", encoding=enc) \
as rev_f:
myrev = rev_f.readline().strip()
return myrev
return "0"
def cleanup(entropy_client, directories):
"""
Temporary files cleaner.
@param directories: list of directory paths
@type directories: list
@return: exit status
@rtype: int
"""
counter = 0
for xdir in directories:
if not os.path.isdir(xdir):
continue
entropy_client.output(
"%s %s %s..." % (
_("Cleaning"), darkgreen(xdir),
_("directory"),),
back = True)
for data in os.listdir(xdir):
subprocess.call(["rm", "-rf", os.path.join(xdir, data)])
counter += 1
entropy_client.output(
"%s: %s %s" % (
_("Cleaned"),
counter,
_("files and directories"),)
)
return 0
def print_table(lines_data, cell_spacing = 2, cell_padding = 0,
side_color = darkgreen):
"""
Print a table composed by len(lines_data[i]) columns and len(lines_data)
rows.
@param lines_data: list of row data
@type lines_data: list
@keyword cell_spacing: cell spacing
@type cell_spacing: int
@keyword cell_padding: cell padding
@type cell_padding: int
@keyword side_color: colorization callback function
@type side_color: callable
"""
column_sizes = {}
padding_side = int((cell_padding / 2))
col_n = 0
for cols in lines_data:
if not isinstance(cols, (list, tuple)):
# can be a plain string
continue
col_n = 0
for cell in cols:
cell_len = len(" "*padding_side + decolorize(cell.split("\n")[0]) \
+ " "*padding_side)
cur_len = column_sizes.get(col_n)
if cur_len is None:
column_sizes[col_n] = cell_len
elif cur_len < cell_len:
column_sizes[col_n] = cell_len
col_n += 1
# now actually print
if col_n > 0:
column_sizes[col_n - 1] = 0
for cols in lines_data:
print_generic(side_color(">>") + " ", end = " ")
if isinstance(cols, (list, tuple)):
col_n = 0
for cell in cols:
max_len = column_sizes[col_n]
cell = " "*padding_side + cell + " "*padding_side
delta_len = max_len - len(decolorize(cell.split("\n")[0])) + \
cell_spacing
if col_n == (len(cols) - 1):
print_generic(cell)
else:
print_generic(cell, end = " "*delta_len)
col_n += 1
else:
print_generic(cols)
def enlightenatom(atom):
"""
Colorize package atoms with standard colors.
@param atom: atom string
@type atom: string
@return: colorized string
@rtype: string
"""
entropy_rev = entropy.dep.dep_get_entropy_revision(atom)
if entropy_rev is None:
entropy_rev = ''
else:
entropy_rev = '~%s' % (str(entropy_rev),)
entropy_tag = entropy.dep.dep_gettag(atom)
if entropy_tag is None:
entropy_tag = ''
else:
entropy_tag = '#%s' % (entropy_tag,)
clean_atom = entropy.dep.remove_entropy_revision(atom)
clean_atom = entropy.dep.remove_tag(clean_atom)
only_cpv = entropy.dep.dep_getcpv(clean_atom)
operator = clean_atom[:len(clean_atom)-len(only_cpv)]
cat, name, pv, rev = entropy.dep.catpkgsplit(only_cpv)
if rev == "r0":
rev = ''
else:
rev = '-%s' % (rev,)
return "%s%s%s%s%s%s%s" % (purple(operator), teal(cat + "/"),
darkgreen(name), purple("-"+pv), purple(rev), brown(entropy_tag),
teal(entropy_rev),)
def show_dependencies_legend(entropy_client, indent = '',
get_data = False):
data = []
dep_type_ids = etpConst['dependency_type_ids']
for dep_id, dep_val in sorted(dep_type_ids.items(),
key = lambda x: x[0], reverse = True):
dep_desc = etpConst['dependency_type_ids_desc'].get(
dep_id, _("N/A"))
txt = '%s%s%s%s %s' % (
indent, teal("{"), dep_val, teal("}"), dep_desc,)
if get_data:
data.append(txt)
else:
entropy_client.output(txt)
if get_data:
return data
def _formatted_print(entropy_client, data, header, reset_columns,
min_chars = 25, color = None, get_data = False):
out_data = []
if isinstance(data, (frozenset, set)):
mydata = sorted(data)
elif not isinstance(data, (list, tuple)):
mydata = data.split()
else:
mydata = data
fcount = 0
desc_text = header
for item in mydata:
fcount += len(item)
if color:
desc_text += color(item)+" "
else:
desc_text += item+" "
if fcount > min_chars:
fcount = 0
if get_data:
out_data.append(desc_text)
else:
entropy_client.output(desc_text)
desc_text = reset_columns
if fcount > 0:
if get_data:
out_data.append(desc_text)
else:
entropy_client.output(desc_text)
if get_data:
return out_data
def print_package_info(package_id, entropy_client, entropy_repository,
installed_search = False, strict_output = False, extended = False,
quiet = False, show_repo_if_quiet = False, show_desc_if_quiet = False):
"""
Print Entropy Package Metadata in a pretty and uniform way.
"""
corrupted_str = _("corrupted")
pkgatom = entropy_repository.retrieveAtom(package_id) or corrupted_str
if quiet:
repoinfo = ''
desc = ''
if show_repo_if_quiet:
repoinfo = "[%s] " % (entropy_repository.repository_id(),)
if show_desc_if_quiet:
desc = ' %s' % (
entropy_repository.retrieveDescription(package_id),)
if not extended:
pkgatom = entropy.dep.dep_getkey(pkgatom)
entropy_client.output(
"%s%s%s" % (repoinfo, pkgatom, desc,),
level="generic")
return
pkghome = entropy_repository.retrieveHomepage(package_id)
if pkghome is None:
pkghome = corrupted_str
pkgslot = entropy_repository.retrieveSlot(package_id) \
or corrupted_str
pkgver = entropy_repository.retrieveVersion(package_id) \
or corrupted_str
pkgtag = entropy_repository.retrieveTag(package_id)
if pkgtag is None:
pkgtag = corrupted_str
pkgrev = entropy_repository.retrieveRevision(package_id)
if pkgrev is None:
pkgrev = 0
pkgdesc = entropy_repository.retrieveDescription(package_id)
if pkgdesc is None:
pkgdesc = corrupted_str
pkgbranch = entropy_repository.retrieveBranch(package_id) \
or corrupted_str
if not pkgtag:
pkgtag = "NoTag"
installed_ver = _("Not installed")
installed_tag = _("N/A")
installed_rev = _("N/A")
if not installed_search:
# client info
pkginstalled = entropy_client.installed_repository().atomMatch(
entropy.dep.dep_getkey(pkgatom), matchSlot = pkgslot)
if pkginstalled[1] == 0:
idx = pkginstalled[0]
# found
installed_ver = entropy_client.installed_repository(
).retrieveVersion(idx) or corrupted_str
installed_tag = entropy_client.installed_repository(
).retrieveTag(idx)
if not installed_tag:
installed_tag = "NoTag"
installed_rev = entropy_client.installed_repository(
).retrieveRevision(idx)
if installed_rev is None:
installed_rev = const_convert_to_unicode("0")
else:
installed_rev = const_convert_to_unicode(installed_rev)
toc = []
entropy_client.output(red(" @@ %s: " % (_("Package"),) ) + \
bold(pkgatom) + \
" "+ blue("%s: " % (_("branch"),)) + bold(pkgbranch) + \
", [" + purple(str(entropy_repository.repository_id())) + "] ")
if not strict_output and extended:
pkgname = entropy_repository.retrieveName(package_id) \
or corrupted_str
pkgcat = entropy_repository.retrieveCategory(package_id) \
or corrupted_str
toc.append((darkgreen(" %s:" % (_("Category"),)),
blue(pkgcat)))
toc.append((darkgreen(" %s:" % (_("Name"),)),
blue(pkgname)))
if extended:
pkgmasked = False
masking_reason = ''
# check if it's masked
package_id_masked, idmasking_reason = \
entropy_repository.maskFilter(package_id)
if package_id_masked == -1:
pkgmasked = True
masking_reason = ", %s" % (
entropy_client.Settings()['pkg_masking_reasons'].get(
idmasking_reason),)
toc.append((darkgreen(" %s:" % (_("Masked"),)),
blue(str(pkgmasked)) + masking_reason,))
avail_str = _("Available")
if installed_search:
avail_str = _("Installed")
toc.append((
darkgreen(" %s:" % (avail_str,)),
blue("%s: " % (_("version"),) ) + bold(pkgver) + blue(" ~ tag: ") + \
bold(pkgtag) + blue(" ~ %s: " % (_("revision"),) ) + bold(str(pkgrev)),)
)
if not installed_search:
toc.append((darkgreen(" %s:" % (_("Installed"),) ),
blue("%s: " % (_("version"),) ) + bold(installed_ver) + \
blue(" ~ tag: ") + bold(installed_tag) + \
blue(" ~ %s: " % (_("revision"),) ) + bold(installed_rev),))
if not strict_output:
toc.append((darkgreen(" %s:" % (_("Slot"),) ),
blue(str(pkgslot)),))
if extended:
pkgsize = entropy_repository.retrieveSize(package_id)
pkgsize = entropy.tools.bytes_into_human(pkgsize)
pkgbin = entropy_repository.retrieveDownloadURL(package_id)
if pkgbin is None:
pkgbin = corrupted_str
pkgdigest = entropy_repository.retrieveDigest(package_id) or \
corrupted_str
pkgdeps = entropy_repository.retrieveDependencies(package_id,
extended = True, resolve_conditional_deps = False)
pkgconflicts = entropy_repository.retrieveConflicts(package_id)
depsorter = lambda x: entropy.dep.dep_getcpv(x[0])
toc.append((darkgreen(" %s:" % (_("Size"),) ),
blue(str(pkgsize)),))
toc.append((darkgreen(" %s:" % (_("Download"),) ),
brown(str(pkgbin)),))
toc.append((darkgreen(" %s:" % (_("Checksum"),) ),
brown(str(pkgdigest)),))
if pkgdeps:
toc.append(darkred(" ##") + " " + \
darkgreen("%s:" % (_("Dependencies"),) ))
for pdep, p_id in sorted(pkgdeps, key = depsorter):
toc.append((" %s " % (brown("##"),),
"%s%s%s %s" % (blue("["), p_id, blue("]"),
brown(pdep),)))
# show legend
len_txt = " %s" % (brown("##"),)
toc.append((len_txt, "%s:" % (blue(_("Legend")),),))
dep_leg = show_dependencies_legend(entropy_client,
indent = "", get_data = True)
toc.extend([(len_txt, x) for x in dep_leg])
if pkgconflicts:
toc.append(darkred(" ##") + " " + \
darkgreen("%s:" % (_("Conflicts"),) ))
for conflict in sorted(pkgconflicts, key = depsorter):
toc.append((" %s" % (darkred("##"),),
brown(conflict),))
home_txt = " %s:" % (_("Homepage"),)
home_lines = _formatted_print(
entropy_client, pkghome, "", "", color = brown,
min_chars = 15, get_data = True)
for home_line in home_lines:
toc.append((darkgreen(home_txt), home_line,))
home_txt = " "
if not strict_output:
desc_txt = " %s:" % (_("Description"),)
desc_lines = _formatted_print(
entropy_client, pkgdesc, "", "", get_data = True)
for desc_line in desc_lines:
toc.append((darkgreen(desc_txt), purple(desc_line)))
desc_txt = " "
if extended:
pkguseflags = entropy_repository.retrieveUseflags(package_id)
use_txt = " %s:" % (_("USE flags"),)
use_lines = _formatted_print(
entropy_client,
pkguseflags, "", "", color = teal,
get_data = True)
for use_line in use_lines:
toc.append((darkgreen(use_txt), use_line))
use_txt = " "
if not strict_output:
if extended:
chost, cflags, cxxflags = \
entropy_repository.retrieveCompileFlags(package_id)
sources = entropy_repository.retrieveSources(package_id)
etpapi = entropy_repository.retrieveApi(package_id)
if etpapi is None:
etpapi = corrupted_str
toc.append((darkgreen(" %s:" % (_("CHOST"),)),
blue(chost)))
toc.append((darkgreen(" %s:" % (_("CFLAGS"),)),
blue(cflags)))
toc.append((darkgreen(" %s:" % (_("CXXFLAGS"),)),
blue(cxxflags)))
if sources:
sources_txt = " %s:" % (_("Sources"),)
toc.append(darkgreen(sources_txt))
for source in sources:
toc.append((" ", source,))
toc.append((darkgreen(" %s:" % (_("Entry API"),)),
purple(str(etpapi))))
toc.append((darkgreen(" %s:" % (_("Compiled with"),)),
blue(cflags)))
pkgkeywords = ' '.join(
sorted(entropy_repository.retrieveKeywords(package_id)))
keyword_txt = " %s:" % (_("Keywords"),)
keyword_lines = _formatted_print(
entropy_client, pkgkeywords, "", "",
color = brown, get_data = True)
for keyword_line in keyword_lines:
toc.append((darkgreen(keyword_txt), brown(keyword_line)))
keyword_txt = " "
mydate = entropy_repository.retrieveCreationDate(package_id)
pkgcreatedate = "N/A"
if mydate:
pkgcreatedate = \
entropy.tools.convert_unix_time_to_human_time(
float(mydate))
toc.append((darkgreen(" %s:" % (_("Created"),)),
purple(pkgcreatedate)))
pkglic = entropy_repository.retrieveLicense(package_id)
if pkglic is None:
pkglic = corrupted_str
toc.append((darkgreen(" %s:" % (_("License"),)),
teal(pkglic)))
print_table(toc, cell_spacing = 3)
def show_you_meant(entropy_client, package, from_installed):
"""
Print Package "did you mean"-like message to stdout.
"""
items = entropy_client.get_meant_packages(
package, from_installed = from_installed)
if not items:
return
items_cache = set()
mytxt = "%s %s %s %s %s" % (
bold(" ?"),
red(_("When you wrote")),
bold(package),
darkgreen(_("You Meant(tm)")),
red(_("one of these below?")),
)
entropy_client.output(mytxt)
for match in items:
if from_installed:
dbconn = entropy_client.installed_repository()
idpackage = match[0]
else:
dbconn = entropy_client.open_repository(match[1])
idpackage = match[0]
key, slot = dbconn.retrieveKeySlot(idpackage)
if (key, slot) not in items_cache:
entropy_client.output(
red(" # ")+blue(key)+":" + \
brown(str(slot))+red(" ?"))
items_cache.add((key, slot))