header_copyright.py

# | Copyright 2017 Karlsruhe Institute of Technology
# |
# | Licensed under the Apache License, Version 2.0 (the "License");
# | you may not use this file except in compliance with the License.
# | You may obtain a copy of the License at
# |
# |     http://www.apache.org/licenses/LICENSE-2.0
# |
# | Unless required by applicable law or agreed to in writing, software
# | distributed under the License is distributed on an "AS IS" BASIS,
# | WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# | See the License for the specific language governing permissions and
# | limitations under the License.

import os, sys, logging, subprocess
from python_compat import imap


HEADER_TEMPLATE = """
~ | Copyright %s Karlsruhe Institute of Technology
~ |
~ | Licensed under the Apache License, Version 2.0 (the "License");
~ | you may not use this file except in compliance with the License.
~ | You may obtain a copy of the License at
~ |
~ |     http://www.apache.org/licenses/LICENSE-2.0
~ |
~ | Unless required by applicable law or agreed to in writing, software
~ | distributed under the License is distributed on an "AS IS" BASIS,
~ | WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
~ | See the License for the specific language governing permissions and
~ | limitations under the License.

""".lstrip().replace('~', '#')


def main():
	import get_file_list
	tab_count = 0
	for (fn, _) in get_file_list.get_file_list(show_type_list=['py', 'sh', 'lib'],
			show_external=False, show_aux=True, show_testsuite=False):
		if 'fragmentForCMSSW.py' in fn:  # don't add header to config fragment
			continue
		sys.stdout.write(fn + ' ')
		sys.stdout.flush()
		tab_count += _update_header(fn)
		sys.stdout.write('OK\n')


def _update_header(fn):
	old_dir = os.getcwd()
	os.chdir(os.path.dirname(fn))
	cmd = 'git log --follow --format="format:%ai %an <%ae> %H" ' + os.path.basename(fn)
	gitlog = subprocess.check_output(cmd, shell=True).decode('utf-8')
	os.chdir(old_dir)

	def _skip_licence_updates(lines):
		licence_update_revs = [
			'421bfb25b9ba81c68225f77bc5142a9c9b68e183',
			'a5cbd19838ca2fe1f30857c19882e008214c11eb',
		]
		for line in lines:
			if any(imap(lambda rev: rev in line, licence_update_revs)):
				continue
			yield line

	log_lines = list(_skip_licence_updates(gitlog.splitlines()))
	try:
		year_a = log_lines[-1].split('-')[0]
		year_b = log_lines[0].split('-')[0]
		if year_a == year_b:
			year = year_a
		else:
			year = year_a + '-' + year_b
		header = HEADER_TEMPLATE % year
	except:
		logging.warning('no version: %s', fn)
		raise

	do_insert = True
	data = open(fn).readlines()
	fp = open(fn, 'w')
	tab_count = 0
	for line in data:
		tab_count += line.count('\t')
		if ('pri' + 'nt "%s"') % fn in line:
			continue
		if line.startswith('#-#') or line.startswith('# |'):
			continue
		elif line.startswith('#!'):
			fp.write(line)
		elif do_insert and (line.strip() != ''):
			do_insert = False
			fp.write(header)
			fp.write(line)
		elif not do_insert:
			fp.write(line)
	fp.close()
	if data != open(fn).readlines():
		logging.warning('changed %s', fn)
	return tab_count


if __name__ == '__main__':
	main()