2 files changed, 190 insertions, 0 deletions

diff --git a/geneticone/helper.py b/geneticone/helper.py
new file mode 100644
index 0000000..8f801ce
--- /dev/null
+++ b/geneticone/helper.py
@@ -0,0 +1,190 @@
+#!/usr/bin/python
+
+from geneticone import *
+import geneticone
+
+import re
+import os
+
+import gentoolkit
+import portage
+from portage_util import unique_array
+
+class BlockedException (Exception):
+	pass
+
+class PackageNotFoundException (Exception):
+	pass
+
+def find_lambda (name):
+	"""Returns the function needed by all the find_all_*-functions. Returns None if no name is given."""
+	if name != None:
+		return lambda x: re.match(".*"+name+".*",x)
+	else:
+		return lambda x: True
+
+def geneticize_list (list_of_packages):
+	'''to convert the output of each gentoolkit helper function is a list of *geneticone* Package objects
+	'''
+	cpvs=[x.get_cpv() for x in list_of_packages]
+	return [geneticone.Package(x) for x in cpvs]
+
+def find_best_match (search_key, only_installed = False):
+	"""Finds the best match in the portage tree."""
+	t = None
+	if not only_installed:
+		t = porttree.dep_bestmatch(search_key)
+	else:
+		t = vartree.dep_bestmatch(search_key)
+	if t:
+		return geneticone.Package(t)
+	return None
+
+def find_packages (search_key, masked=False):
+	"""This returns a list of packages which have to fit exactly. Additionally ranges like '<,>,=,~,!' et. al. are possible."""
+	return geneticize_list(gentoolkit.find_packages(search_key, masked))
+
+def find_installed_packages (search_key, masked=False):
+	"""This returns a list of installed packages which have to fit exactly. Additionally ranges like '<,>,=,~,!' et. al. are possible.""" 
+	return geneticize_list(gentoolkit.find_installed_packages(search_key, masked))
+
+def find_system_packages (name=None):
+	"""Returns a list-tuple (resolved_packages, unresolved_packages) for all system packages."""
+	list = gentoolkit.find_system_packages()
+	return (geneticize_list(list[0]), geneticize_list(list[1]))
+
+def find_world_packages ():
+	"""Returns a list-tuple (resolved_packages, unresolved_packages) for all packages in the world-file."""
+	list = gentoolkit.find_world_packages()
+	return geneticize_list(list[0]),geneticize_list(list[1])
+
+def find_all_installed_packages (name=None):
+	"""Returns a list of all installed packages matching ".*name.*". 
+	Returns ALL installed packages if name is None."""
+	return geneticize_list(gentoolkit.find_all_installed_packages(find_lambda(name)))
+
+def find_all_uninstalled_packages (name=None):
+	"""Returns a list of all uninstalled packages matching ".*name.*". 
+	Returns ALL uninstalled packages if name is None."""
+	return geneticize_list(gentoolkit.find_all_uninstalled_packages(find_lambda(name)))
+
+def find_all_packages (name=None):
+	"""Returns a list of all packages matching ".*name.*". 
+	Returns ALL packages if name is None."""
+	return geneticize_list(gentoolkit.find_all_packages(find_lambda(name)))
+
+def find_all_world_files (name=None):
+	"""Returns a list of all world packages matching ".*name.*". 
+	Returns ALL world packages if name is None."""
+	world = filter(find_lambda(name), [x.get_cpv() for x in find_world_packages()[0]])
+	world = unique_array(world)
+	return [geneticone.Package(x) for x in world]
+
+def find_all_system_files (name=None):
+	"""Returns a list of all system packages matching ".*name.*". 
+	Returns ALL system packages if name is None."""
+	sys = filter(find_lambda(name), [x.get_cpv() for x in find_system_packages()[0]])
+	sys = unique_array(sys)
+	return [geneticone.Package(x) for x in sys]
+
+def list_categories (name=None):
+	"""Returns a list of categories matching ".*name.*" or all categories."""
+	categories = gentoolkit.settings.categories
+	return filter(find_lambda(name), categories)
+
+def split_package_name (name):
+	"""Returns a list in the form [category, name, version, revision]. Revision will
+	be 'r0' if none can be inferred. Category and version will be empty, if none can
+	be inferred."""
+	return gentoolkit.split_package_name(name)
+
+def sort_package_list(pkglist):
+	"""Sorts a package list in the same manner portage does."""
+	return gentoolkit.sort_package_list(pkglist)
+
+def am_i_root ():
+	"""Returns True if the current user is root, False otherwise."""
+	if os.getuid() == 0:
+		return True
+	else:
+		return False
+
+def old_export_to_dictionaries (list_of_packages):
+	'''DEPRECATED:
+	Exports some of the intrinsic data of a list of Package objects to a list of dictionaries.
+	This is meant to transmit data back to the genetic-client, just by eval()-uating the output.'''
+	dictionaries=[]
+	keys=['name','version','category','cpv','runtime_deps','compiletime_deps','postmerge_deps','is_installed', 'is_overlay', 'is_masked','mask_status', 'package_path', 'size','use_flags_when_installed','all_useflags','set_useflags']
+	package_methods=['get_name','get_version','get_category','get_cpv','get_runtime_deps', 'get_compiletime_deps','get_postmerge_deps','is_installed','is_overlay','is_masked','get_mask_status','get_package_path','size','get_use_flags','get_all_useflags','get_set_useflags']
+	
+	for item in list_of_packages:
+		dictionaries.append({})
+		for key,method in zip(keys,package_methods):
+			try:
+				dictionaries[-1][key]=eval('item.'+method+'()')
+			except AttributeError: #this may happen if, for example, package is not installed and I look for the path...
+				dictionaries[-1][key]=None
+	return dictionaries
+
+def export_to_dictionaries (packages):
+	'''Exports some of the intrinsic data of a list of Package objects to a list of dictionaries.
+	This is meant to transmit data back to the genetic-client, just by eval()-uating the output.'''
+	dictionaries=[]
+
+	for item in packages:
+		dictionaries.append({})
+		for method in dir(item):
+			if (method.startswith('get_') or method.startswith('is_'))\
+					and method != 'get_dependants': # bug in gentoolkit.Package.get_dependants --> see bug #137783
+				key = method[method.index('_')+1:] # the key is everything after the first underscore
+				try:
+					dictionaries[-1][key] = eval("item."+method+"()")
+				except AttributeError: # this may happen if, for example, package is not installed and I look for the path...
+					dictionaries[-1][key] = None
+				except TypeError:
+					pass # this method takes an argument - ignore it
+				except NotImplementedError:
+					pass # this method is not implemented - ignore
+				except "Not implemented yet!":
+					pass
+	return dictionaries
+
+use_descs = {}
+local_use_descs = {}
+def get_use_desc (flag, package = None):
+	"""Returns the description of a specific useflag or None if no desc was found. 
+	If a package is given (in the <cat>/<name> format) the local use descriptions are searched too.
+	In the first run the dictionaries 'use_descs' and 'local_use_descs' are filled."""
+	# fill cache if needed
+	if use_descs == {} or local_use_descs == {}:
+		# read use.desc
+		fd = open(gentoolkit.settings["PORTDIR"]+"/profiles/use.desc")
+		for line in fd.readlines():
+			line = line.strip()
+			if line != "" and line[0] != '#':
+				fields = [x.strip() for x in line.split(" - ",1)]
+				if len(fields) == 2:
+					use_descs[fields[0]] = fields[1]
+
+		# read use.local.desc
+		fd = open(gentoolkit.settings["PORTDIR"]+"/profiles/use.local.desc")
+		for line in fd.readlines():
+			line = line.strip()
+			if line != "" and line[0] != '#':
+				fields = [x.strip() for x in line.split(":",1)]
+				if len(fields) == 2:
+					if not fields[0] in local_use_descs: # create
+						local_use_descs[fields[0]] = {}
+					subfields = [x.strip() for x in fields[1].split(" - ",1)]
+					if len(subfields) == 2:
+						local_use_descs[fields[0]][subfields[0]] = subfields[1]
+	
+	# start
+	desc = None
+	if flag in use_descs:
+		desc = use_descs[flag]
+	if package != None:
+		if package in local_use_descs:
+			if flag in local_use_descs[package]:
+				desc = local_use_descs[package][flag]
+	return desc
diff --git a/geneticone/helper.pyc b/geneticone/helper.pyc
new file mode 100644
index 0000000..50fb7e8
--- /dev/null
+++ b/geneticone/helper.pyc