update code

This commit is contained in:
Dimitris Zlatanidis 2014-10-24 08:00:32 +03:00
parent a835733460
commit 31ad7ef03f

View file

@ -50,8 +50,9 @@ def slack_install(slack_pkg, version):
CYAN, slack_pkg, ENDC)) CYAN, slack_pkg, ENDC))
sys.stdout.write("{0}Reading package lists ...{1}".format(GREY, ENDC)) sys.stdout.write("{0}Reading package lists ...{1}".format(GREY, ENDC))
sys.stdout.flush() sys.stdout.flush()
PACKAGES_TXT = _data(version) data = _greps(_packages(version), slack_pkg, version)
dwn_list, install_all, comp_sum, uncomp_sum = _greps(PACKAGES_TXT, dwn_list, install_all, comp_sum, uncomp_sum = _store(data[0], data[1],
data[2], data[3],
slack_pkg, version) slack_pkg, version)
sys.stdout.write("{0}Done{1}\n\n".format(GREY, ENDC)) sys.stdout.write("{0}Done{1}\n\n".format(GREY, ENDC))
if install_all: if install_all:
@ -101,9 +102,9 @@ def _init(tmp_path):
os.mkdir(tmp_path) os.mkdir(tmp_path)
def _data(version): def _packages(version):
''' '''
Collects and return data Collects and return packages
''' '''
PACKAGES = url_read(mirrors("PACKAGES.TXT", "", version)) PACKAGES = url_read(mirrors("PACKAGES.TXT", "", version))
EXTRA = url_read(mirrors("PACKAGES.TXT", "extra/", version)) EXTRA = url_read(mirrors("PACKAGES.TXT", "extra/", version))
@ -127,29 +128,33 @@ def _greps(PACKAGES_TXT, slack_pkg, version):
''' '''
Grap data packages Grap data packages
''' '''
(pkg_name, pkg_location, size, unsize, dwn, (name, location, size, unsize) = ([] for i in range(4))
install, comp_sum, uncomp_sum) = ([] for i in range(8))
toolbar_width, index = 800, 0 toolbar_width, index = 800, 0
for line in PACKAGES_TXT.splitlines(): for line in PACKAGES_TXT.splitlines():
index += 1 index += 1
toolbar_width = _toolbar(index, toolbar_width) toolbar_width = _toolbar(index, toolbar_width)
if line.startswith("PACKAGE NAME"): if line.startswith("PACKAGE NAME"):
pkg_name.append(line[15:].strip()) name.append(line[15:].strip())
if line.startswith("PACKAGE LOCATION"): if line.startswith("PACKAGE LOCATION"):
pkg_location.append(line[21:].strip()) location.append(line[21:].strip())
if line.startswith("PACKAGE SIZE (compressed): "): if line.startswith("PACKAGE SIZE (compressed): "):
size.append(line[28:-2].strip()) size.append(line[28:-2].strip())
if line.startswith("PACKAGE SIZE (uncompressed): "): if line.startswith("PACKAGE SIZE (uncompressed): "):
unsize.append(line[30:-2].strip()) unsize.append(line[30:-2].strip())
for name, loc, comp, uncomp in zip(pkg_name, pkg_location, return [name, location, size, unsize]
size, unsize):
if slack_pkg in name and slack_pkg not in BlackList().packages():
dwn.append("{0}{1}/{2}".format(mirrors("", "", version), def _store(*args):
loc, name)) '''
Store data
'''
dwn, install, comp_sum, uncomp_sum = ([] for i in range(4))
for name, loc, comp, uncomp in zip(args[0], args[1], args[2], args[3]):
if args[4] in name and args[4] not in BlackList().packages():
dwn.append("{0}{1}/{2}".format(mirrors("", "", args[5]), loc, name))
install.append(name) install.append(name)
comp_sum.append(comp) comp_sum.append(comp)
uncomp_sum.append(uncomp) uncomp_sum.append(uncomp)
print dwn, install
return [dwn, install, comp_sum, uncomp_sum] return [dwn, install, comp_sum, uncomp_sum]
@ -202,6 +207,9 @@ def _units(comp_sum, uncomp_sum):
def _msgs(install_all, uni_sum): def _msgs(install_all, uni_sum):
'''
Print singular plural
'''
msg_pkg = "package" msg_pkg = "package"
msg_2_pkg = msg_pkg msg_2_pkg = msg_pkg
if len(install_all) > 1: if len(install_all) > 1: