From 00cec97b2da41a916e00961976ce69a9beea551e Mon Sep 17 00:00:00 2001
From: Fabio Erculiani <lxnay@sabayon.org>
Date: Tue, 23 Nov 2010 21:09:39 +0100
Subject: [PATCH] [entropy.dep] add helper function expand_dependencies()

---
 libraries/entropy/dep.py | 52 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 52 insertions(+)

diff --git a/libraries/entropy/dep.py b/libraries/entropy/dep.py
index a96687e6a..419d09f87 100644
--- a/libraries/entropy/dep.py
+++ b/libraries/entropy/dep.py
@@ -940,3 +940,55 @@ class DependencyStringParser(object):
         except DependencyStringParser.MalformedDependency:
             return False, []
         return matched, matched_deps
+
+
+def expand_dependencies(dependencies, entropy_repository):
+    """
+    Expand a list of dependencies resolving conditional ones.
+    NOTE: it automatically handles dependencies metadata extended format:
+        [(dep, type), ...]
+
+    @param dependencies: list of raw package dependencies, as
+        returned by EntropyRepositoryBase.retrieveDependencies{,List}()
+    @type dependencies: iterable
+    @param entropy_repository: EntropyRepositoryBase instance used to execute
+        the actual dependency resolution
+    @type entropy_repository: EntropyRepositoryBase
+    @return: list (keeping the iterable order when possible) of expanded
+        dependencies
+    @rtype: list
+    @raise AttributeError: if dependencies structure is unsupported (this
+        function supports list of strings, or list of tuples of length 2)
+    """
+    pkg_deps = []
+    for dep in dependencies:
+        is_tuple = False
+        dep_type = None
+        if isinstance(dep, tuple):
+            if len(dep) == 2:
+                dep, dep_type = dep
+            elif len(dep) == 1:
+                dep = dep[0]
+            else:
+                raise AttributeError("malformed input dependencies")
+        if dep.startswith("("):
+            try:
+                deps = DependencyStringParser(pkg_dep,
+                    entropy_repository).parse()
+            except DependencyStringParser.MalformedDependency:
+                # wtf! add as-is
+                if dep_type is None:
+                    pkg_deps.append(dep)
+                else:
+                    pkg_deps.append((dep, dep_type))
+                continue
+            if dep_type is None:
+                pkg_deps.extend(deps)
+            else:
+                pkg_deps.extend([(x, dep_type) for x in deps])
+        elif dep_type is None:
+            pkg_deps.append(dep)
+        else:
+            pkg_deps.append((dep, dep_type))
+
+    return pkg_deps