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
58
59
60
61
62
63
64
65
66
|
#!/usr/bin/env python
import re
import sys
import datetime
import subprocess
import multiprocessing
co_line_matcher = re.compile('^.*Copyright (.*) Ettus Research LLC$')
def command(*args): return subprocess.Popen(args, stdout=subprocess.PIPE).communicate()[0]
def get_co_line(lines):
for i, line in enumerate(lines[:5]):
if co_line_matcher.match(line): return line, i
return None, None
def fix_co_years(files):
for file in files:
print file
lines = open(file).readlines()
line, num = get_co_line(lines)
if line is None: continue
#extract the years from the git history
log_years = map(
lambda l: int(l.split()[-2]),
filter(
lambda l: l.startswith('Date'),
command('git', 'log', file).splitlines(),
),
)
log_years = min(log_years), max(log_years)
#extract years from co string
try:
co_years_str = co_line_matcher.match(line).groups()[0]
co_years = map(int, co_years_str.split('-'))
co_years = min(co_years), max(co_years)
except Exception, e:
print ' format error on line %d: "%s"'%(num, line), e
continue
if log_years != co_years:
print ' log years: %s != copyright years: %s'%(log_years, co_years)
year_now = datetime.datetime.now().year
all_years = min(log_years), max(list(log_years)+[year_now]) #add the current year
all_years_str = '%s-%s'%all_years
if all_years[0] == all_years[1]: all_years_str = str(all_years[0])
new_text = ''.join(lines[:num] + [line.replace(co_years_str, all_years_str)] + lines[num+1:])
open(file, 'w').write(new_text)
if __name__ == "__main__":
if len(sys.argv) < 2: print "Usage: %s path/"%sys.argv[0]; exit()
#get recursive list of files in the repo
files = command('git', 'ls-tree', '--name-only', 'HEAD', '-r', sys.argv[1]).splitlines()
#start n+1 processes to handle the files
num_procs = multiprocessing.cpu_count()
procs = [multiprocessing.Process(
target=lambda *files: fix_co_years(files),
args=files[num::num_procs],
) for num in range(num_procs)]
map(multiprocessing.Process.start, procs)
map(multiprocessing.Process.join, procs)
|
