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

Last change on this file since 445 was 445, checked in by kbut, 4 years ago

wip

File size: 18.2 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        if options.recursive:
190            mapped_seeds.extend(structure.inner_seeds(seed_name))
191            mapped_seeds = list(set(mapped_seeds))
192
193        for mapped_seed in mapped_seeds:
194            packages.extend(germinator_method(structure, mapped_seed))
195        return packages
196
197    def metapackage_name(structure, seed_name):
198        if config.has_option(dist, "metapackage_map/%s" % seed_name):
199            return config.get(dist, "metapackage_map/%s" % seed_name)
200        else:
201            task_meta_re = re.compile('^Task-Metapackage:\s*(.*)', re.I)
202            with structure[seed_name] as seed:
203                for line in seed:
204                    task_meta_match = task_meta_re.match(line)
205                    if task_meta_match is not None:
206                        return task_meta_match.group(1)
207            return "%s-%s" % (metapackage, seed_name)
208
209    debootstrap_version_file = 'debootstrap-version'
210
211    def get_debootstrap_version():
212        version_cmd = subprocess.Popen(
213            ['dpkg-query', '-W', '--showformat', '${Version}', 'debootstrap'],
214            stdout=subprocess.PIPE, universal_newlines=True)
215        version, _ = version_cmd.communicate()
216        if not version:
217            error_exit('debootstrap does not appear to be installed')
218
219        return version
220
221    def debootstrap_packages(arch):
222        env = dict(os.environ)
223        if 'PATH' in env:
224            env['PATH'] = '/usr/sbin:/sbin:%s' % env['PATH']
225        else:
226            env['PATH'] = '/usr/sbin:/sbin:/usr/bin:/bin'
227        debootstrap = subprocess.Popen(
228            ['debootstrap', '--arch', arch,
229             '--no-check-gpg',
230             '--components', ','.join(components),
231             '--print-debs', dist, 'debootstrap-dir', archive_base[arch][0]],
232            stdout=subprocess.PIPE, env=env, stderr=subprocess.PIPE,
233            universal_newlines=True)
234        (debootstrap_stdout, debootstrap_stderr) = debootstrap.communicate()
235        if debootstrap.returncode != 0:
236            error_exit('Unable to retrieve package list from debootstrap; '
237                       'stdout: %s\nstderr: %s' %
238                       (debootstrap_stdout, debootstrap_stderr))
239
240        # sometimes debootstrap gives empty packages / multiple separators
241        packages = [pkg for pkg in debootstrap_stdout.split() if pkg]
242
243        return sorted(packages)
244
245    def check_debootstrap_version():
246        if os.path.exists(debootstrap_version_file):
247            with open(debootstrap_version_file) as debootstrap:
248                old_debootstrap_version = debootstrap.read().strip()
249            debootstrap_version = get_debootstrap_version()
250            failed = subprocess.call(
251                ['dpkg', '--compare-versions',
252                 debootstrap_version, 'ge', old_debootstrap_version])
253            if failed:
254                error_exit('Installed debootstrap is older than in the '
255                           'previous version! (%s < %s)' %
256                           (debootstrap_version, old_debootstrap_version))
257
258    def update_debootstrap_version():
259        with open(debootstrap_version_file, 'w') as debootstrap:
260            debootstrap.write(get_debootstrap_version() + '\n')
261
262    def format_changes(items):
263        by_arch = defaultdict(set)
264        for pkg, arch in items:
265            by_arch[pkg].add(arch)
266        all_pkgs = sorted(by_arch)
267        chunks = []
268        for pkg in all_pkgs:
269            arches = by_arch[pkg]
270            if set(architectures) - arches:
271                # only some architectures
272                chunks.append('%s [%s]' % (pkg, ' '.join(sorted(arches))))
273            else:
274                # all architectures
275                chunks.append(pkg)
276        return ', '.join(chunks)
277
278    germinate_logging(logging.DEBUG)
279
280    check_debootstrap_version()
281
282    additions = defaultdict(list)
283    removals = defaultdict(list)
284    moves = defaultdict(list)
285    metapackage_map = {}
286    for architecture in architectures:
287        print("[%s] Downloading available package lists..." % architecture)
288        germinator = Germinator(architecture)
289        archive = germinate.archive.TagFile(
290            dists, components, architecture,
291            archive_base[architecture], source_mirrors=archive_base_default,
292            cleanup=True)
293        germinator.parse_archive(archive)
294        debootstrap_base = set(debootstrap_packages(architecture))
295
296        print("[%s] Loading seed lists..." % architecture)
297        try:
298            structure = SeedStructure(seed_dist, seed_base, options.vcs)
299            germinator.plant_seeds(structure, seeds=seeds)
300        except SeedError:
301            sys.exit(1)
302
303        print("[%s] Merging seeds with available package lists..." %
304              architecture)
305        for seed_name in output_seeds:
306            meta_name = metapackage_name(structure, seed_name)
307            metapackage_map[seed_name] = meta_name
308
309            output_filename = os.path.join(
310                options.outdir, '%s-%s' % (seed_name, architecture))
311            old_list = None
312            if os.path.exists(output_filename):
313                with open(output_filename) as output:
314                    old_list = set(map(str.strip, output.readlines()))
315                os.rename(output_filename, output_filename + '.old')
316
317            # work on the depends
318            new_list = []
319            packages = seed_packages(germinator.get_seed_entries,
320                                     structure, seed_name)
321            for package in packages:
322                if package == meta_name:
323                    print("%s/%s: Skipping package %s (metapackage)" %
324                          (seed_name, architecture, package))
325                elif (seed_name == 'minimal' and
326                      package not in debootstrap_base):
327                    print("%s/%s: Skipping package %s (package not in "
328                          "debootstrap)" % (seed_name, architecture, package))
329                elif germinator.is_essential(package):
330                    print("%s/%s: Skipping package %s (essential)" %
331                          (seed_name, architecture, package))
332                else:
333                    new_list.append(package)
334
335            new_list.sort()
336            with open(output_filename, 'w') as output:
337                for package in new_list:
338                    output.write(package)
339                    output.write('\n')
340
341            # work on the recommends
342            old_recommends_list = None
343            new_recommends_list = []
344            packages = seed_packages(germinator.get_seed_recommends_entries,
345                                     structure, seed_name)
346            for package in packages:
347                if package == meta_name:
348                    print("%s/%s: Skipping package %s (metapackage)" %
349                          (seed_name, architecture, package))
350                    continue
351                if seed_name == 'minimal' and package not in debootstrap_base:
352                    print("%s/%s: Skipping package %s (package not in "
353                          "debootstrap)" % (seed_name, architecture, package))
354                else:
355                    new_recommends_list.append(package)
356
357            new_recommends_list.sort()
358            seed_name_recommends = '%s-recommends' % seed_name
359            output_recommends_filename = os.path.join(
360                options.outdir, '%s-%s' % (seed_name_recommends, architecture))
361            if os.path.exists(output_recommends_filename):
362                with open(output_recommends_filename) as output:
363                    old_recommends_list = set(
364                        map(str.strip, output.readlines()))
365                os.rename(
366                    output_recommends_filename,
367                    output_recommends_filename + '.old')
368
369            with open(output_recommends_filename, 'w') as output:
370                for package in new_recommends_list:
371                    output.write(package)
372                    output.write('\n')
373
374            # Calculate deltas
375            merged = defaultdict(int)
376            recommends_merged = defaultdict(int)
377            if old_list is not None:
378                for package in new_list:
379                    merged[package] += 1
380                for package in old_list:
381                    merged[package] -= 1
382            if old_recommends_list is not None:
383                for package in new_recommends_list:
384                    recommends_merged[package] += 1
385                for package in old_recommends_list:
386                    recommends_merged[package] -= 1
387
388            mergeditems = sorted(merged.items())
389            for package, value in mergeditems:
390                #print(package, value)
391                if value == 1:
392                    if recommends_merged.get(package, 0) == -1:
393                        moves[package].append([seed_name, architecture])
394                        recommends_merged[package] += 1
395                    else:
396                        additions[package].append([seed_name, architecture])
397                elif value == -1:
398                    if recommends_merged.get(package, 0) == 1:
399                        moves[package].append([seed_name_recommends,
400                                               architecture])
401                        recommends_merged[package] -= 1
402                    else:
403                        removals[package].append([seed_name, architecture])
404
405            mergedrecitems = sorted(recommends_merged.items())
406            for package, value in mergedrecitems:
407                #print(package, value)
408                if value == 1:
409                    additions[package].append([seed_name_recommends,
410                                               architecture])
411                elif value == -1:
412                    removals[package].append([seed_name_recommends,
413                                              architecture])
414
415    with open('metapackage-map', 'w') as metapackage_map_file:
416        for seed_name in output_seeds:
417            print(seed_name, metapackage_map[seed_name],
418                  file=metapackage_map_file)
419
420    if not options.nodch and (additions or removals or moves):
421        dch_help = subprocess.Popen(['dch', '--help'], stdout=subprocess.PIPE,
422                                    universal_newlines=True)
423        try:
424            have_U = '-U' in dch_help.stdout.read()
425        finally:
426            if dch_help.stdout:
427                dch_help.stdout.close()
428            dch_help.wait()
429        if have_U:
430            subprocess.check_call(['dch', '-iU', 'Refreshed dependencies'])
431        else:
432            subprocess.check_call(['dch', '-i', 'Refreshed dependencies'])
433        changes = []
434        for package in sorted(additions):
435            changes.append('Added %s to %s' %
436                           (package, format_changes(additions[package])))
437        for package in sorted(removals):
438            changes.append('Removed %s from %s' %
439                           (package, format_changes(removals[package])))
440        for package in sorted(moves):
441            # TODO: We should really list where it moved from as well, but
442            # that gets wordy very quickly, and at the moment this is only
443            # implemented for depends->recommends or vice versa. In future,
444            # using this for moves between seeds might also be useful.
445            changes.append('Moved %s to %s' %
446                           (package, format_changes(moves[package])))
447        for change in changes:
448            print(change)
449            subprocess.check_call(['dch', '-a', change])
450        update_debootstrap_version()
451    else:
452        if not options.nodch:
453            print("No changes found")
454
455    return 0
Note: See TracBrowser for help on using the repository browser.