summaryrefslogtreecommitdiff
path: root/copybump.py
blob: 83f0364c05325b65011f5bde26b2f0cb79a4021b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
#! /usr/bin/env python3

import os
import re
import datetime


FSF = 'by the Free Software Foundation, Inc.'
this_year = datetime.date.today().year
pyre = re.compile(r'^# Copyright (C) (?P<start>\d{4}-)?(?P<end>\d{4})')


def do_file(path):
    print('=>', path)
    with open(path) as in_file, open(path + '.out', 'w') as out_file:
        for line in in_file:
            mo = pyre.match(line)
            if mo is None:
                out_file.write(line)
                continue
            start = (mo.group('end')
                     if mo.group('start') is None
                     else mo.group('start'))
            print('# Copyright (C) {}-{} {}'.format(
                  mo.group('end'), this_year, FSF), file=out_file)
            for line in in_file:
                out_file.write(line)
    os.rename(path + '.out', path)


def do_walk():
    for root, dirs, files in os.walk('.'):
        if root == '.':
            dirs.remove('.bzr')
            dirs.remove('bin')
            dirs.remove('contrib')
            dirs.remove('develop-eggs')
            dirs.remove('eggs')
            dirs.remove('parts')
            files.remove('gnu-COPYING-GPL')
            files.remove('.installed.cfg')
            files.remove('.bzrignore')
            files.remove('distribute_setup.py')
        if root == './src':
            dirs.remove('mailman.egg-info')
        if root == './src/mailman':
            dirs.remove('messages')
        for file_name in files:
            if os.path.splitext(file_name)[1] in ('.pyc', '.gz', '.egg'):
                continue
            path = os.path.join(root, file_name)
            if os.path.isfile(path):
                do_file(path)


if __name__ == '__main__':
    do_walk()