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

Last change on this file since 733 was 733, checked in by kbut, 3 years ago

recursive enhanced

File size: 18.5 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    parser.add_option('-r','--recursive', dest='recursive', action='store_true',
88                      default=False,
89                      help="expand all seeds dependencies from STRUCTURE")
90    return parser.parse_args(argv[1:])
91
92
93def main(argv):
94    options, args = parse_options(argv)
95
96    if not os.path.exists('debian/control'):
97        error_exit('must be run from the top level of a source package')
98    this_source = None
99    with open('debian/control') as control:
100        for line in control:
101            if line.startswith('Source:'):
102                this_source = line[7:].strip()
103                break
104            elif line == '':
105                break
106    if this_source is None:
107        error_exit('cannot find Source: in debian/control')
108    if not this_source.endswith('-meta'):
109        error_exit('source package name must be *-meta')
110    metapackage = this_source[:-5]
111
112    print("[info] Initialising %s-* package lists update..." % metapackage)
113
114    config = SafeConfigParser()
115    with open('update.cfg') as config_file:
116        try:
117            # >= 3.2
118            config.read_file(config_file)
119        except AttributeError:
120            # < 3.2
121            config.readfp(config_file)
122
123    if len(args) > 0:
124        dist = args[0]
125    else:
126        dist = config.get('DEFAULT', 'dist')
127
128    seeds = config.get(dist, 'seeds').split()
129    try:
130        output_seeds = config.get(dist, 'output_seeds').split()
131    except NoOptionError:
132        output_seeds = list(seeds)
133    architectures = config.get(dist, 'architectures').split()
134    try:
135        archive_base_default = config.get(dist, 'archive_base/default')
136        archive_base_default = re.split(r'[, ]+', archive_base_default)
137    except (NoSectionError, NoOptionError):
138        archive_base_default = None
139
140    archive_base = {}
141    for arch in architectures:
142        try:
143            archive_base[arch] = config.get(dist, 'archive_base/%s' % arch)
144            archive_base[arch] = re.split(r'[, ]+', archive_base[arch])
145        except (NoSectionError, NoOptionError):
146            if archive_base_default is not None:
147                archive_base[arch] = archive_base_default
148            else:
149                error_exit('no archive_base configured for %s' % arch)
150
151    if options.vcs and config.has_option("%s/vcs" % dist, 'seed_base'):
152        seed_base = config.get("%s/vcs" % dist, 'seed_base')
153    elif options.vcs and config.has_option("%s/bzr" % dist, 'seed_base'):
154        # Backward compatibility.
155        seed_base = config.get("%s/bzr" % dist, 'seed_base')
156    else:
157        seed_base = config.get(dist, 'seed_base')
158    seed_base = re.split(r'[, ]+', seed_base)
159    if options.vcs and config.has_option("%s/vcs" % dist, 'seed_dist'):
160        seed_dist = config.get("%s/vcs" % dist, 'seed_dist')
161    elif options.vcs and config.has_option("%s/bzr" % dist, 'seed_dist'):
162        # Backward compatibility.
163        seed_dist = config.get("%s/bzr" % dist, 'seed_dist')
164    elif config.has_option(dist, 'seed_dist'):
165        seed_dist = config.get(dist, 'seed_dist')
166    else:
167        seed_dist = dist
168    if config.has_option(dist, 'dists'):
169        dists = config.get(dist, 'dists').split()
170    else:
171        dists = [dist]
172    components = config.get(dist, 'components').split()
173
174    def seed_packages(germinator_method, structure, seed_name):
175        if config.has_option(dist, "seed_map/%s" % seed_name):
176            mapped_seeds = config.get(dist, "seed_map/%s" % seed_name).split()
177        else:
178            mapped_seeds = []
179            task_seeds_re = re.compile('^Task-Seeds:\s*(.*)', re.I)
180            with structure[seed_name] as seed:
181                for line in seed:
182                    task_seeds_match = task_seeds_re.match(line)
183                    if task_seeds_match is not None:
184                        mapped_seeds = task_seeds_match.group(1).split()
185                        break
186            if seed_name not in mapped_seeds:
187                mapped_seeds.append(seed_name)
188        packages = []
189        task_seeds_re = re.compile('^Task-Recursive:\s*(.*)', re.I)
190        with structure[seed_name] as seed:
191            for line in seed:
192                task_seeds_match = task_seeds_re.match(line)
193                if task_seeds_match is not None:
194                    mapped_seeds.extend(structure.inner_seeds(seed_name))
195                    mapped_seeds = list(set(mapped_seeds))
196                    break
197        for mapped_seed in mapped_seeds:
198            packages.extend(germinator_method(structure, mapped_seed))
199        return packages
200
201    def metapackage_name(structure, seed_name):
202        if config.has_option(dist, "metapackage_map/%s" % seed_name):
203            return config.get(dist, "metapackage_map/%s" % seed_name)
204        else:
205            task_meta_re = re.compile('^Task-Metapackage:\s*(.*)', re.I)
206            with structure[seed_name] as seed:
207                for line in seed:
208                    task_meta_match = task_meta_re.match(line)
209                    if task_meta_match is not None:
210                        return task_meta_match.group(1)
211            return "%s-%s" % (metapackage, seed_name)
212
213    debootstrap_version_file = 'debootstrap-version'
214
215    def get_debootstrap_version():
216        version_cmd = subprocess.Popen(
217            ['dpkg-query', '-W', '--showformat', '${Version}', 'debootstrap'],
218            stdout=subprocess.PIPE, universal_newlines=True)
219        version, _ = version_cmd.communicate()
220        if not version:
221            error_exit('debootstrap does not appear to be installed')
222
223        return version
224
225    def debootstrap_packages(arch):
226        env = dict(os.environ)
227        if 'PATH' in env:
228            env['PATH'] = '/usr/sbin:/sbin:%s' % env['PATH']
229        else:
230            env['PATH'] = '/usr/sbin:/sbin:/usr/bin:/bin'
231        debootstrap = subprocess.Popen(
232            ['debootstrap', '--arch', arch,
233             '--no-check-gpg',
234             '--components', ','.join(components),
235             '--print-debs', dist, 'debootstrap-dir', archive_base[arch][0]],
236            stdout=subprocess.PIPE, env=env, stderr=subprocess.PIPE,
237            universal_newlines=True)
238        (debootstrap_stdout, debootstrap_stderr) = debootstrap.communicate()
239        if debootstrap.returncode != 0:
240            error_exit('Unable to retrieve package list from debootstrap; '
241                       'stdout: %s\nstderr: %s' %
242                       (debootstrap_stdout, debootstrap_stderr))
243
244        # sometimes debootstrap gives empty packages / multiple separators
245        packages = [pkg for pkg in debootstrap_stdout.split() if pkg]
246
247        return sorted(packages)
248
249    def check_debootstrap_version():
250        if os.path.exists(debootstrap_version_file):
251            with open(debootstrap_version_file) as debootstrap:
252                old_debootstrap_version = debootstrap.read().strip()
253            debootstrap_version = get_debootstrap_version()
254            failed = subprocess.call(
255                ['dpkg', '--compare-versions',
256                 debootstrap_version, 'ge', old_debootstrap_version])
257            if failed:
258                error_exit('Installed debootstrap is older than in the '
259                           'previous version! (%s < %s)' %
260                           (debootstrap_version, old_debootstrap_version))
261
262    def update_debootstrap_version():
263        with open(debootstrap_version_file, 'w') as debootstrap:
264            debootstrap.write(get_debootstrap_version() + '\n')
265
266    def format_changes(items):
267        by_arch = defaultdict(set)
268        for pkg, arch in items:
269            by_arch[pkg].add(arch)
270        all_pkgs = sorted(by_arch)
271        chunks = []
272        for pkg in all_pkgs:
273            arches = by_arch[pkg]
274            if set(architectures) - arches:
275                # only some architectures
276                chunks.append('%s [%s]' % (pkg, ' '.join(sorted(arches))))
277            else:
278                # all architectures
279                chunks.append(pkg)
280        return ', '.join(chunks)
281
282    germinate_logging(logging.DEBUG)
283
284    check_debootstrap_version()
285
286    additions = defaultdict(list)
287    removals = defaultdict(list)
288    moves = defaultdict(list)
289    metapackage_map = {}
290    for architecture in architectures:
291        print("[%s] Downloading available package lists..." % architecture)
292        germinator = Germinator(architecture)
293        archive = germinate.archive.TagFile(
294            dists, components, architecture,
295            archive_base[architecture], source_mirrors=archive_base_default,
296            cleanup=True)
297        germinator.parse_archive(archive)
298        debootstrap_base = set(debootstrap_packages(architecture))
299
300        print("[%s] Loading seed lists..." % architecture)
301        try:
302            structure = SeedStructure(seed_dist, seed_base, options.vcs)
303            germinator.plant_seeds(structure, seeds=seeds)
304        except SeedError:
305            sys.exit(1)
306
307        print("[%s] Merging seeds with available package lists..." %
308              architecture)
309        for seed_name in output_seeds:
310            meta_name = metapackage_name(structure, seed_name)
311            metapackage_map[seed_name] = meta_name
312
313            output_filename = os.path.join(
314                options.outdir, '%s-%s' % (seed_name, architecture))
315            old_list = None
316            if os.path.exists(output_filename):
317                with open(output_filename) as output:
318                    old_list = set(map(str.strip, output.readlines()))
319                os.rename(output_filename, output_filename + '.old')
320
321            # work on the depends
322            new_list = []
323            packages = seed_packages(germinator.get_seed_entries,
324                                     structure, seed_name)
325            for package in packages:
326                if package == meta_name:
327                    print("%s/%s: Skipping package %s (metapackage)" %
328                          (seed_name, architecture, package))
329                elif (seed_name == 'minimal' and
330                      package not in debootstrap_base):
331                    print("%s/%s: Skipping package %s (package not in "
332                          "debootstrap)" % (seed_name, architecture, package))
333                elif germinator.is_essential(package):
334                    print("%s/%s: Skipping package %s (essential)" %
335                          (seed_name, architecture, package))
336                else:
337                    new_list.append(package)
338
339            new_list.sort()
340            with open(output_filename, 'w') as output:
341                for package in new_list:
342                    output.write(package)
343                    output.write('\n')
344
345            # work on the recommends
346            old_recommends_list = None
347            new_recommends_list = []
348            packages = seed_packages(germinator.get_seed_recommends_entries,
349                                     structure, seed_name)
350            for package in packages:
351                if package == meta_name:
352                    print("%s/%s: Skipping package %s (metapackage)" %
353                          (seed_name, architecture, package))
354                    continue
355                if seed_name == 'minimal' and package not in debootstrap_base:
356                    print("%s/%s: Skipping package %s (package not in "
357                          "debootstrap)" % (seed_name, architecture, package))
358                else:
359                    new_recommends_list.append(package)
360
361            new_recommends_list.sort()
362            seed_name_recommends = '%s-recommends' % seed_name
363            output_recommends_filename = os.path.join(
364                options.outdir, '%s-%s' % (seed_name_recommends, architecture))
365            if os.path.exists(output_recommends_filename):
366                with open(output_recommends_filename) as output:
367                    old_recommends_list = set(
368                        map(str.strip, output.readlines()))
369                os.rename(
370                    output_recommends_filename,
371                    output_recommends_filename + '.old')
372
373            with open(output_recommends_filename, 'w') as output:
374                for package in new_recommends_list:
375                    output.write(package)
376                    output.write('\n')
377
378            # Calculate deltas
379            merged = defaultdict(int)
380            recommends_merged = defaultdict(int)
381            if old_list is not None:
382                for package in new_list:
383                    merged[package] += 1
384                for package in old_list:
385                    merged[package] -= 1
386            if old_recommends_list is not None:
387                for package in new_recommends_list:
388                    recommends_merged[package] += 1
389                for package in old_recommends_list:
390                    recommends_merged[package] -= 1
391
392            mergeditems = sorted(merged.items())
393            for package, value in mergeditems:
394                #print(package, value)
395                if value == 1:
396                    if recommends_merged.get(package, 0) == -1:
397                        moves[package].append([seed_name, architecture])
398                        recommends_merged[package] += 1
399                    else:
400                        additions[package].append([seed_name, architecture])
401                elif value == -1:
402                    if recommends_merged.get(package, 0) == 1:
403                        moves[package].append([seed_name_recommends,
404                                               architecture])
405                        recommends_merged[package] -= 1
406                    else:
407                        removals[package].append([seed_name, architecture])
408
409            mergedrecitems = sorted(recommends_merged.items())
410            for package, value in mergedrecitems:
411                #print(package, value)
412                if value == 1:
413                    additions[package].append([seed_name_recommends,
414                                               architecture])
415                elif value == -1:
416                    removals[package].append([seed_name_recommends,
417                                              architecture])
418
419    with open('metapackage-map', 'w') as metapackage_map_file:
420        for seed_name in output_seeds:
421            print(seed_name, metapackage_map[seed_name],
422                  file=metapackage_map_file)
423
424    if not options.nodch and (additions or removals or moves):
425        dch_help = subprocess.Popen(['dch', '--help'], stdout=subprocess.PIPE,
426                                    universal_newlines=True)
427        try:
428            have_U = '-U' in dch_help.stdout.read()
429        finally:
430            if dch_help.stdout:
431                dch_help.stdout.close()
432            dch_help.wait()
433        if have_U:
434            subprocess.check_call(['dch', '-iU', 'Refreshed dependencies'])
435        else:
436            subprocess.check_call(['dch', '-i', 'Refreshed dependencies'])
437        changes = []
438        for package in sorted(additions):
439            changes.append('Added %s to %s' %
440                           (package, format_changes(additions[package])))
441        for package in sorted(removals):
442            changes.append('Removed %s from %s' %
443                           (package, format_changes(removals[package])))
444        for package in sorted(moves):
445            # TODO: We should really list where it moved from as well, but
446            # that gets wordy very quickly, and at the moment this is only
447            # implemented for depends->recommends or vice versa. In future,
448            # using this for moves between seeds might also be useful.
449            changes.append('Moved %s to %s' %
450                           (package, format_changes(moves[package])))
451        for change in changes:
452            print(change)
453            subprocess.check_call(['dch', '-a', change])
454        update_debootstrap_version()
455    else:
456        if not options.nodch:
457            print("No changes found")
458
459    return 0
Note: See TracBrowser for help on using the repository browser.