From 200377bef14daf8931be28e06411dfdef22298b2 Mon Sep 17 00:00:00 2001
From: Colin Watson <cjwatson@canonical.com>
Date: Sun, 4 Nov 2012 00:03:26 +0000
Subject: [PATCH] Optionally merge packages from testing to unstable

To cope with a partial unstable suite like Ubuntu's -proposed pocket.
---
 britney.py | 65 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 65 insertions(+)

diff --git a/britney.py b/britney.py
index fd8f2b7..3edf89b 100755
--- a/britney.py
+++ b/britney.py
@@ -319,6 +319,8 @@ class Britney(object):
         # read the source and binary packages for the involved distributions
         self.sources['testing'] = self.read_sources(self.options.testing)
         self.sources['unstable'] = self.read_sources(self.options.unstable)
+        if hasattr(self.options, 'partial_unstable'):
+            self.merge_sources('testing', 'unstable')
         for suite in ('tpu', 'pu'):
             if hasattr(self.options, suite):
                 self.sources[suite] = self.read_sources(getattr(self.options, suite))
@@ -343,6 +345,9 @@ class Britney(object):
         # Load testing last as some live-data tests have more complete information in
         # unstable
         self.binaries['testing'] = self.read_binaries(self.options.testing, "testing", self.options.architectures)
+        if hasattr(self.options, 'partial_unstable'):
+            for arch in self.options.architectures:
+                self.merge_binaries('testing', 'unstable', arch)
 
         try:
             constraints_file = os.path.join(self.options.static_input_dir, 'constraints')
@@ -989,6 +994,64 @@ class Britney(object):
 
         return packages
 
+    def merge_sources(self, source, target):
+        """Merge sources from `source' into partial suite `target'."""
+        source_sources = self.sources[source]
+        target_sources = self.sources[target]
+        for pkg, value in source_sources.items():
+            if pkg not in target_sources:
+                target_sources[pkg] = value
+
+    def merge_binaries(self, source, target, arch):
+        """Merge `arch' binaries from `source' into partial suite `target'."""
+        source_sources = self.sources[source]
+        source_binaries, _ = self.binaries[source][arch]
+        target_sources = self.sources[target]
+        target_binaries, target_provides = self.binaries[target][arch]
+        oodsrcs = set()
+        for pkg, value in source_binaries.items():
+            if pkg in target_binaries:
+                continue
+
+            # Don't merge binaries rendered stale by new sources in target
+            # that have built on this architecture.
+            if value.source not in oodsrcs:
+                source_version = source_sources[value.source].version
+                try:
+                    target_version = target_sources[value.source].version
+                except KeyError:
+                    self.log("merge_binaries: pkg %s has no source, NBS?" % pkg)
+                    continue
+                if source_version != target_version:
+                    current_arch = value.architecture
+                    built = False
+                    for b in target_sources[value.source].binaries:
+                        if b.architecture == arch:
+                            target_value = target_binaries[b.package_name]
+                            if current_arch in (
+                                target_value.architecture, 'all'):
+                                built = True
+                                break
+                    if built:
+                        continue
+                oodsrcs.add(value.source)
+
+            if pkg in target_binaries:
+                for p in target_binaries[pkg].provides:
+                    target_provides[p].remove(pkg)
+                    if not target_provides[p]:
+                        del target_provides[p]
+
+            target_binaries[pkg] = value
+
+            if value.pkg_id not in target_sources[value.source].binaries:
+                target_sources[value.source].binaries.append(value.pkg_id)
+
+            for p in value.provides:
+                if p not in target_provides:
+                    target_provides[p] = []
+                target_provides[p].append(pkg)
+
     def read_binaries(self, basedir, distribution, architectures):
         """Read the list of binary packages from the specified directory
 
@@ -1323,6 +1386,8 @@ class Britney(object):
 
             # retrieve the testing (if present) and unstable corresponding binary packages
             binary_t = pkg_name in packages_t_a and packages_t_a[pkg_name] or None
+            if hasattr(self.options, 'partial_unstable') and binary_t is not None and binary_t[ARCHITECTURE] == 'all' and pkg_name not in self.binaries[suite][arch][0]:
+                excuse.addhtml("Ignoring %s %s (from %s) as it is arch: all and not yet built in unstable" % (pkg_name, binary_t.version, binary_t.source_version))
             binary_u = packages_s_a[pkg_name]
 
             # this is the source version for the new binary package