source: germinate/trunk/fuentes/germinate/scripts/germinate_update_metapackage.py @ 169

Last change on this file since 169 was 169, checked in by mabarracus, 5 years ago

wip

File size: 17.9 KB
Line 
1# -*- coding: UTF-8 -*-
2
3# Copyright (C) 2004, 2005, 2006, 2007, 2008, 2009, 2011, 2012 Canonical Ltd.
4# Copyright (C) 2006 Gustavo Franco
5#
6# This file is part of Germinate.
7#
8# Germinate is free software; you can redistribute it and/or modify it
9# under the terms of the GNU General Public License as published by the
10# Free Software Foundation; either version 2, or (at your option) any
11# later version.
12#
13# Germinate is distributed in the hope that it will be useful, but
14# WITHOUT ANY WARRANTY; without even the implied warranty of
15# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
16# General Public License for more details.
17#
18# You should have received a copy of the GNU General Public License
19# along with Germinate; see the file COPYING.  If not, write to the Free
20# Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
21# 02110-1301, USA.
22
23# TODO:
24# - Exclude essential packages from dependencies
25
26from __future__ import print_function
27
28from collections import defaultdict
29import logging
30import optparse
31import os
32import re
33import subprocess
34import sys
35
36try:
37    # >= 3.0
38    from configparser import NoOptionError, NoSectionError
39    if (sys.version_info[0] < 3 or
40        (sys.version_info[0] == 3 and sys.version_info[1] < 2)):
41        # < 3.2
42        from configparser import SafeConfigParser
43    else:
44        # >= 3.2
45        from configparser import ConfigParser as SafeConfigParser
46except ImportError:
47    # < 3.0
48    from ConfigParser import NoOptionError, NoSectionError, SafeConfigParser
49
50import germinate.archive
51from germinate.germinator import Germinator
52from germinate.log import germinate_logging
53from germinate.seeds import SeedError, SeedStructure, SeedVcs
54import germinate.version
55
56
57__pychecker__ = 'maxlocals=80'
58
59
60def error_exit(message):
61    print("%s: %s" % (sys.argv[0], message), file=sys.stderr)
62    sys.exit(1)
63
64
65def parse_options(argv):
66    description = '''\
67Update metapackage lists for distribution 'dist' as defined in
68update.cfg.'''
69
70    parser = optparse.OptionParser(
71        prog='germinate-update-metapackage',
72        usage='%prog [options] [dist]',
73        version='%prog ' + germinate.version.VERSION,
74        description=description)
75    parser.add_option('-o', '--output-directory', dest='outdir',
76                      default='.', metavar='DIR',
77                      help='output in specific directory')
78    parser.add_option('--nodch', dest='nodch', action='store_true',
79                      default=False,
80                      help="don't modify debian/changelog")
81    parser.add_option('--vcs', dest='vcs',
82                      action='store_const', const=SeedVcs.AUTO,
83                      help='fetch seeds using a version control system')
84    parser.add_option('--bzr', dest='vcs', action='store_true',
85                      help='fetch seeds using bzr (requires bzr to be '
86                           'installed; use --vcs instead)')
87    return parser.parse_args(argv[1:])
88
89
90def main(argv):
91    options, args = parse_options(argv)
92
93    if not os.path.exists('debian/control'):
94        error_exit('must be run from the top level of a source package')
95    this_source = None
96    with open('debian/control') as control:
97        for line in control:
98            if line.startswith('Source:'):
99                this_source = line[7:].strip()
100                break
101            elif line == '':
102                break
103    if this_source is None:
104        error_exit('cannot find Source: in debian/control')
105    if not this_source.endswith('-meta'):
106        error_exit('source package name must be *-meta')
107    metapackage = this_source[:-5]
108
109    print("[info] Initialising %s-* package lists update..." % metapackage)
110
111    config = SafeConfigParser()
112    with open('update.cfg') as config_file:
113        try:
114            # >= 3.2
115            config.read_file(config_file)
116        except AttributeError:
117            # < 3.2
118            config.readfp(config_file)
119
120    if len(args) > 0:
121        dist = args[0]
122    else:
123        dist = config.get('DEFAULT', 'dist')
124
125    seeds = config.get(dist, 'seeds').split()
126    try:
127        output_seeds = config.get(dist, 'output_seeds').split()
128    except NoOptionError:
129        output_seeds = list(seeds)
130    architectures = config.get(dist, 'architectures').split()
131    try:
132        archive_base_default = config.get(dist, 'archive_base/default')
133        archive_base_default = re.split(r'[, ]+', archive_base_default)
134    except (NoSectionError, NoOptionError):
135        archive_base_default = None
136
137    archive_base = {}
138    for arch in architectures:
139        try:
140            archive_base[arch] = config.get(dist, 'archive_base/%s' % arch)
141            archive_base[arch] = re.split(r'[, ]+', archive_base[arch])
142        except (NoSectionError, NoOptionError):
143            if archive_base_default is not None:
144                archive_base[arch] = archive_base_default
145            else:
146                error_exit('no archive_base configured for %s' % arch)
147
148    if options.vcs and config.has_option("%s/vcs" % dist, 'seed_base'):
149        seed_base = config.get("%s/vcs" % dist, 'seed_base')
150    elif options.vcs and config.has_option("%s/bzr" % dist, 'seed_base'):
151        # Backward compatibility.
152        seed_base = config.get("%s/bzr" % dist, 'seed_base')
153    else:
154        seed_base = config.get(dist, 'seed_base')
155    seed_base = re.split(r'[, ]+', seed_base)
156    if options.vcs and config.has_option("%s/vcs" % dist, 'seed_dist'):
157        seed_dist = config.get("%s/vcs" % dist, 'seed_dist')
158    elif options.vcs and config.has_option("%s/bzr" % dist, 'seed_dist'):
159        # Backward compatibility.
160        seed_dist = config.get("%s/bzr" % dist, 'seed_dist')
161    elif config.has_option(dist, 'seed_dist'):
162        seed_dist = config.get(dist, 'seed_dist')
163    else:
164        seed_dist = dist
165    if config.has_option(dist, 'dists'):
166        dists = config.get(dist, 'dists').split()
167    else:
168        dists = [dist]
169    components = config.get(dist, 'components').split()
170
171    def seed_packages(germinator_method, structure, seed_name):
172        if config.has_option(dist, "seed_map/%s" % seed_name):
173            mapped_seeds = config.get(dist, "seed_map/%s" % seed_name).split()
174        else:
175            mapped_seeds = []
176            task_seeds_re = re.compile('^Task-Seeds:\s*(.*)', re.I)
177            with structure[seed_name] as seed:
178                for line in seed:
179                    task_seeds_match = task_seeds_re.match(line)
180                    if task_seeds_match is not None:
181                        mapped_seeds = task_seeds_match.group(1).split()
182                        break
183            if seed_name not in mapped_seeds:
184                mapped_seeds.append(seed_name)
185        packages = []
186        for mapped_seed in mapped_seeds:
187            packages.extend(germinator_method(structure, mapped_seed))
188        return packages
189
190    def metapackage_name(structure, seed_name):
191        if config.has_option(dist, "metapackage_map/%s" % seed_name):
192            return config.get(dist, "metapackage_map/%s" % seed_name)
193        else:
194            task_meta_re = re.compile('^Task-Metapackage:\s*(.*)', re.I)
195            with structure[seed_name] as seed:
196                for line in seed:
197                    task_meta_match = task_meta_re.match(line)
198                    if task_meta_match is not None:
199                        return task_meta_match.group(1)
200            return "%s-%s" % (metapackage, seed_name)
201
202    debootstrap_version_file = 'debootstrap-version'
203
204    def get_debootstrap_version():
205        version_cmd = subprocess.Popen(
206            ['dpkg-query', '-W', '--showformat', '${Version}', 'debootstrap'],
207            stdout=subprocess.PIPE, universal_newlines=True)
208        version, _ = version_cmd.communicate()
209        if not version:
210            error_exit('debootstrap does not appear to be installed')
211
212        return version
213
214    def debootstrap_packages(arch):
215        env = dict(os.environ)
216        if 'PATH' in env:
217            env['PATH'] = '/usr/sbin:/sbin:%s' % env['PATH']
218        else:
219            env['PATH'] = '/usr/sbin:/sbin:/usr/bin:/bin'
220        debootstrap = subprocess.Popen(
221            ['debootstrap', '--arch', arch,
222             '--no-check-gpg',
223             '--components', ','.join(components),
224             '--print-debs', dist, 'debootstrap-dir', archive_base[arch][0]],
225            stdout=subprocess.PIPE, env=env, stderr=subprocess.PIPE,
226            universal_newlines=True)
227        (debootstrap_stdout, debootstrap_stderr) = debootstrap.communicate()
228        if debootstrap.returncode != 0:
229            error_exit('Unable to retrieve package list from debootstrap; '
230                       'stdout: %s\nstderr: %s' %
231                       (debootstrap_stdout, debootstrap_stderr))
232
233        # sometimes debootstrap gives empty packages / multiple separators
234        packages = [pkg for pkg in debootstrap_stdout.split() if pkg]
235
236        return sorted(packages)
237
238    def check_debootstrap_version():
239        if os.path.exists(debootstrap_version_file):
240            with open(debootstrap_version_file) as debootstrap:
241                old_debootstrap_version = debootstrap.read().strip()
242            debootstrap_version = get_debootstrap_version()
243            failed = subprocess.call(
244                ['dpkg', '--compare-versions',
245                 debootstrap_version, 'ge', old_debootstrap_version])
246            if failed:
247                error_exit('Installed debootstrap is older than in the '
248                           'previous version! (%s < %s)' %
249                           (debootstrap_version, old_debootstrap_version))
250
251    def update_debootstrap_version():
252        with open(debootstrap_version_file, 'w') as debootstrap:
253            debootstrap.write(get_debootstrap_version() + '\n')
254
255    def format_changes(items):
256        by_arch = defaultdict(set)
257        for pkg, arch in items:
258            by_arch[pkg].add(arch)
259        all_pkgs = sorted(by_arch)
260        chunks = []
261        for pkg in all_pkgs:
262            arches = by_arch[pkg]
263            if set(architectures) - arches:
264                # only some architectures
265                chunks.append('%s [%s]' % (pkg, ' '.join(sorted(arches))))
266            else:
267                # all architectures
268                chunks.append(pkg)
269        return ', '.join(chunks)
270
271    germinate_logging(logging.DEBUG)
272
273    check_debootstrap_version()
274
275    additions = defaultdict(list)
276    removals = defaultdict(list)
277    moves = defaultdict(list)
278    metapackage_map = {}
279    for architecture in architectures:
280        print("[%s] Downloading available package lists..." % architecture)
281        germinator = Germinator(architecture)
282        archive = germinate.archive.TagFile(
283            dists, components, architecture,
284            archive_base[architecture], source_mirrors=archive_base_default,
285            cleanup=True)
286        germinator.parse_archive(archive)
287        debootstrap_base = set(debootstrap_packages(architecture))
288
289        print("[%s] Loading seed lists..." % architecture)
290        try:
291            structure = SeedStructure(seed_dist, seed_base, options.vcs)
292            germinator.plant_seeds(structure, seeds=seeds)
293        except SeedError:
294            sys.exit(1)
295
296        print("[%s] Merging seeds with available package lists..." %
297              architecture)
298        for seed_name in output_seeds:
299            meta_name = metapackage_name(structure, seed_name)
300            metapackage_map[seed_name] = meta_name
301
302            output_filename = os.path.join(
303                options.outdir, '%s-%s' % (seed_name, architecture))
304            old_list = None
305            if os.path.exists(output_filename):
306                with open(output_filename) as output:
307                    old_list = set(map(str.strip, output.readlines()))
308                os.rename(output_filename, output_filename + '.old')
309
310            # work on the depends
311            new_list = []
312            packages = seed_packages(germinator.get_seed_entries,
313                                     structure, seed_name)
314            for package in packages:
315                if package == meta_name:
316                    print("%s/%s: Skipping package %s (metapackage)" %
317                          (seed_name, architecture, package))
318                elif (seed_name == 'minimal' and
319                      package not in debootstrap_base):
320                    print("%s/%s: Skipping package %s (package not in "
321                          "debootstrap)" % (seed_name, architecture, package))
322                elif germinator.is_essential(package):
323                    print("%s/%s: Skipping package %s (essential)" %
324                          (seed_name, architecture, package))
325                else:
326                    new_list.append(package)
327
328            new_list.sort()
329            with open(output_filename, 'w') as output:
330                for package in new_list:
331                    output.write(package)
332                    output.write('\n')
333
334            # work on the recommends
335            old_recommends_list = None
336            new_recommends_list = []
337            packages = seed_packages(germinator.get_seed_recommends_entries,
338                                     structure, seed_name)
339            for package in packages:
340                if package == meta_name:
341                    print("%s/%s: Skipping package %s (metapackage)" %
342                          (seed_name, architecture, package))
343                    continue
344                if seed_name == 'minimal' and package not in debootstrap_base:
345                    print("%s/%s: Skipping package %s (package not in "
346                          "debootstrap)" % (seed_name, architecture, package))
347                else:
348                    new_recommends_list.append(package)
349
350            new_recommends_list.sort()
351            seed_name_recommends = '%s-recommends' % seed_name
352            output_recommends_filename = os.path.join(
353                options.outdir, '%s-%s' % (seed_name_recommends, architecture))
354            if os.path.exists(output_recommends_filename):
355                with open(output_recommends_filename) as output:
356                    old_recommends_list = set(
357                        map(str.strip, output.readlines()))
358                os.rename(
359                    output_recommends_filename,
360                    output_recommends_filename + '.old')
361
362            with open(output_recommends_filename, 'w') as output:
363                for package in new_recommends_list:
364                    output.write(package)
365                    output.write('\n')
366
367            # Calculate deltas
368            merged = defaultdict(int)
369            recommends_merged = defaultdict(int)
370            if old_list is not None:
371                for package in new_list:
372                    merged[package] += 1
373                for package in old_list:
374                    merged[package] -= 1
375            if old_recommends_list is not None:
376                for package in new_recommends_list:
377                    recommends_merged[package] += 1
378                for package in old_recommends_list:
379                    recommends_merged[package] -= 1
380
381            mergeditems = sorted(merged.items())
382            for package, value in mergeditems:
383                #print(package, value)
384                if value == 1:
385                    if recommends_merged.get(package, 0) == -1:
386                        moves[package].append([seed_name, architecture])
387                        recommends_merged[package] += 1
388                    else:
389                        additions[package].append([seed_name, architecture])
390                elif value == -1:
391                    if recommends_merged.get(package, 0) == 1:
392                        moves[package].append([seed_name_recommends,
393                                               architecture])
394                        recommends_merged[package] -= 1
395                    else:
396                        removals[package].append([seed_name, architecture])
397
398            mergedrecitems = sorted(recommends_merged.items())
399            for package, value in mergedrecitems:
400                #print(package, value)
401                if value == 1:
402                    additions[package].append([seed_name_recommends,
403                                               architecture])
404                elif value == -1:
405                    removals[package].append([seed_name_recommends,
406                                              architecture])
407
408    with open('metapackage-map', 'w') as metapackage_map_file:
409        for seed_name in output_seeds:
410            print(seed_name, metapackage_map[seed_name],
411                  file=metapackage_map_file)
412
413    if not options.nodch and (additions or removals or moves):
414        dch_help = subprocess.Popen(['dch', '--help'], stdout=subprocess.PIPE,
415                                    universal_newlines=True)
416        try:
417            have_U = '-U' in dch_help.stdout.read()
418        finally:
419            if dch_help.stdout:
420                dch_help.stdout.close()
421            dch_help.wait()
422        if have_U:
423            subprocess.check_call(['dch', '-iU', 'Refreshed dependencies'])
424        else:
425            subprocess.check_call(['dch', '-i', 'Refreshed dependencies'])
426        changes = []
427        for package in sorted(additions):
428            changes.append('Added %s to %s' %
429                           (package, format_changes(additions[package])))
430        for package in sorted(removals):
431            changes.append('Removed %s from %s' %
432                           (package, format_changes(removals[package])))
433        for package in sorted(moves):
434            # TODO: We should really list where it moved from as well, but
435            # that gets wordy very quickly, and at the moment this is only
436            # implemented for depends->recommends or vice versa. In future,
437            # using this for moves between seeds might also be useful.
438            changes.append('Moved %s to %s' %
439                           (package, format_changes(moves[package])))
440        for change in changes:
441            print(change)
442            subprocess.check_call(['dch', '-a', change])
443        update_debootstrap_version()
444    else:
445        if not options.nodch:
446            print("No changes found")
447
448    return 0
Note: See TracBrowser for help on using the repository browser.