Add parser for Package-List field.

[dak.git] / daklib / packagelist.py

diff --git a/daklib/packagelist.py b/daklib/packagelist.py
new file mode 100644 (file)
index 0000000..82653a0
--- /dev/null
+++ b/daklib/packagelist.py
@@ -0,0 +1,128 @@
+"""parse Package-List field
+
+@copyright: 2014, Ansgar Burchardt <ansgar@debian.org>
+@license: GPL-2+
+"""
+
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program; if not, write to the Free Software
+# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+
+from daklib.architecture import match_architecture
+from daklib.utils import extract_component_from_section
+
+class InvalidSource(Exception):
+    pass
+
+class PackageListEntry(object):
+    def __init__(self, name, package_type, section, component, priority, **other):
+        self.name = name
+        self.package_type = package_type
+        self.section = section
+        self.component = component
+        self.priority = priority
+        self.other = other
+    @property
+    def architectures(self):
+        archs = self.other.get("arch", None)
+        if archs is None:
+            return None
+        return archs.split(',')
+    def built_on_architecture(self, architecture):
+        archs = self.architectures
+        if archs is None:
+            return None
+        for arch in archs:
+            if match_architecture(architecture, arch):
+                return True
+        return False
+    def built_in_suite(self, suite):
+        built = False
+        for arch in suite.architectures:
+            built_on_arch = self.built_on_architecture(arch.arch_string)
+            if built_on_arch:
+                return True
+            if built_on_arch is None:
+                built = None
+        return built
+
+class PackageList(object):
+    def __init__(self, source):
+        self._source = source
+        if 'Package-List' in self._source:
+            self._parse()
+        elif 'Binary' in self._source:
+            self._parse_fallback()
+        else:
+            raise InvalidSource('Source package has neither Package-List nor Binary field.')
+    def _parse(self):
+        self.package_list = {}
+
+        for line in self._source['Package-List'].split("\n"):
+            if not line:
+                continue
+            fields = line.split()
+            if len(fields) < 4:
+                raise InvalidSource("Package-List entry has less than four fields.")
+
+            # <name> <type> <component/section> <priority> [arch=<arch>[,<arch>]...]
+            name = fields[0]
+            package_type = fields[1]
+            component, section = extract_component_from_section(fields[2])
+            priority = fields[3]
+            other = dict(kv.split('=', 1) for kv in fields[4:])
+
+            entry = PackageListEntry(name, package_type, section, component, priority, **other)
+            self.package_list[name] = entry
+
+    def _parse_fallback(self):
+        self.package_list = {}
+
+        for binary in self._source['Binary'].split():
+            name = binary
+            package_type = None
+            component = None
+            section = None
+            priority = None
+            other = dict()
+
+            entry = PackageListEntry(name, package_type, section, component, priority, **other)
+            self.package_list[name] = entry
+
+    def packages_for_suite(self, suite):
+        packages = []
+        for entry in self.package_list.values():
+            built = entry.built_in_suite(suite)
+            if built or built is None:
+                packages.append(entry)
+        return packages
+
+    def has_arch_indep_packages(self):
+        has_arch_indep = False
+        for entry in self.package_list.values():
+            built = entry.built_on_architecture('all')
+            if built:
+                return True
+            if built is None:
+                has_arch_indep = None
+        return has_arch_indep
+
+    def has_arch_dep_packages(self):
+        has_arch_dep = False
+        for entry in self.package_list.values():
+            built_on_all = entry.built_on_architecture('all')
+            if built_on_all == False:
+                return True
+            if built_on_all is None:
+                has_arch_dep = None
+        return has_arch_dep