forked from pupil-labs/pupil
-
Notifications
You must be signed in to change notification settings - Fork 1
/
update_license_header.py
87 lines (71 loc) · 3.07 KB
/
update_license_header.py
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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
'''
(*)~----------------------------------------------------------------------------------
Pupil - eye tracking platform
Copyright (C) 2012-2016 Pupil Labs
Distributed under the terms of the GNU Lesser General Public License (LGPL v3.0).
License details are in the file license.txt, distributed as part of this software.
----------------------------------------------------------------------------------~(*)
'''
import fnmatch
import os
import re
license_txt = """\
(*)~----------------------------------------------------------------------------------
Pupil - eye tracking platform
Copyright (C) 2012-2016 Pupil Labs
Distributed under the terms of the GNU Lesser General Public License (LGPL v3.0).
License details are in the file license.txt, distributed as part of this software.
----------------------------------------------------------------------------------~(*)\
"""
# find out the cwd and change to the top level Pupil folder
cwd = os.getcwd()
pupil_dir = os.path.join(cwd,'pupil_src')
pattern = re.compile('(\'{3}|[/][*])\n\([*]\)~(.+?)~\([*]\)\n(\'{3}|[*][/])', re.DOTALL|re.MULTILINE)
# choose files types to include
# choose directories to exclude from search
includes = ['*.py', '*.c']
excludes = ['.git', '*.md', 'src_video', 'recordings', 'License', 'shader.py', 'vertex_buffer.py',
'gprof2dot.py','git_version.py','libuvcc*', '*.pstats', '*.png',
'*.svg', '*.ico', '*.sh', '*.icns', '*.spec', '.gitignore','glfw.py', 'version_utils.py']
# transform glob patterns to regular expressions
includes = r'|'.join([fnmatch.translate(x) for x in includes])
excludes = r'|'.join([fnmatch.translate(x) for x in excludes]) or r'$.'
def get_files(start_dir, includes, excludes):
# use os.walk to recursively dig down into the Pupil directory
match_files = []
for root, dirs, files in os.walk(start_dir):
if not re.search(excludes, root):
files = [f for f in files if re.search(includes, f) and not re.search(excludes, f)]
files = [os.path.join(root, f) for f in files]
match_files += files
return match_files
def write_header(file_name, license_txt):
# find and replace license header
# or add new header if not existing
c_comment = ['/*\n', '\n*/\n']
py_comment = ["'''\n","\n'''\n"]
file_type = os.path.splitext(file_name)[-1]
if file_type == '.py':
license_txt = py_comment[0] + license_txt + py_comment[1]
if file_type == '.c':
license_txt = c_comment[0] + license_txt + c_comment[1]
with file(file_name, 'r') as original:
data = original.read()
with file(file_name, 'w') as modified:
if re.findall(pattern, data):
# if header already exists, then update, but dont add the last newline.
modified.write(re.sub(pattern, license_txt[:-1], data))
modified.close()
else:
# else write the license header
modified.write(license_txt + data)
modified.close()
def update_header():
# Add a license/docstring header to selected files
match_files = get_files(pupil_dir, includes, excludes)
print match_files
for f in match_files:
write_header(f, license_txt)
if __name__ == '__main__':
# run update_header() to add headers to find files
update_header()