tools/generate-notice-files.py - platform/build - Git at Google

 #!/usr/bin/env python
 #
 # Copyright (C) 2012 The Android Open Source Project
 #
 # 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.
 """
 Usage: generate-notice-files [plain text output file] [html output file] [file title] [directory of notices]

 Generate the Android notice files, including both text and html files.

 -h to display this usage message and exit.
 """
 from collections import defaultdict
 import getopt
 import hashlib
 import itertools
 import os
 import os.path
 import re
 import sys

 MD5_BLOCKSIZE = 1024 * 1024
 HTML_ESCAPE_TABLE = {
     "&": "&amp;",
     '"': "&quot;",
     "'": "&apos;",
     ">": "&gt;",
     "<": "&lt;",
     }

 try:
   opts, args = getopt.getopt(sys.argv[1:], "h")
 except getopt.GetoptError, err:
     print str(err)
     print __doc__
     sys.exit(2)

 for o, a in opts:
   if o == "-h":
     print __doc__
     sys.exit(2)
   else:
     print >> sys.stderr, "unhandled option %s" % (o,)

 if len(args) != 4:
     print """need exactly four arguments, the two output files, the file title
              and the directory containing notices, not %d""" % (len(args),)
     print __doc__
     sys.exit(1)

 def hexify(s):
     return ("%02x"*len(s)) % tuple(map(ord, s))

 def md5sum(filename):
     """Calculate an MD5 of the file given by FILENAME,
     and return hex digest as a string.
     Output should be compatible with md5sum command"""

     f = open(filename, "rb")
     sum = hashlib.md5()
     while 1:
         block = f.read(MD5_BLOCKSIZE)
         if not block:
             break
         sum.update(block)
     f.close()
     return hexify(sum.digest())


 def html_escape(text):
     """Produce entities within text."""
     return "".join(HTML_ESCAPE_TABLE.get(c,c) for c in text)

 HTML_OUTPUT_CSS="""
 <style type="text/css">
 body { padding: 0; font-family: sans-serif; }
 .same-license { background-color: #eeeeee; border-top: 20px solid white; padding: 10px; }
 .label { font-weight: bold; }
 .file-list { margin-left: 1em; color: blue; }
 </style>
 """

 def combine_notice_files_html(file_hash, input_dir, output_filename):
     """Combine notice files in FILE_HASH and output a HTML version to OUTPUT_FILENAME."""

     SRC_DIR_STRIP_RE = re.compile(input_dir + "(/.*).txt")

     # Set up a filename to row id table (anchors inside tables don't work in
     # most browsers, but href's to table row ids do)
     id_table = {}
     id_count = 0
     for value in file_hash.values():
         for filename in value:
              id_table[filename] = id_count
         id_count += 1

     # Open the output file, and output the header pieces
     output_file = open(output_filename, "wb")

     print >> output_file, "<html><head>"
     print >> output_file, HTML_OUTPUT_CSS
     print >> output_file, '</head><body topmargin="0" leftmargin="0" rightmargin="0" bottommargin="0">'

     # Output our table of contents
     print >> output_file, '<div class="toc">'
     print >> output_file, "<ul>"

     # Flatten the list of lists into a single list of filenames
     sorted_filenames = sorted(itertools.chain.from_iterable(file_hash.values()))

     # Print out a nice table of contents
     for filename in sorted_filenames:
         stripped_filename = SRC_DIR_STRIP_RE.sub(r"\1", filename)
         print >> output_file, '<li><a href="#id%d">%s</a></li>' % (id_table.get(filename), stripped_filename)

     print >> output_file, "</ul>"
     print >> output_file, "</div><!-- table of contents -->"
     # Output the individual notice file lists
     print >>output_file, '<table cellpadding="0" cellspacing="0" border="0">'
     for value in file_hash.values():
         print >> output_file, '<tr id="id%d"><td class="same-license">' % id_table.get(value[0])
         print >> output_file, '<div class="label">Notices for file(s):</div>'
         print >> output_file, '<div class="file-list">'
         for filename in sorted(value):
             print >> output_file, "%s <br/>" % (SRC_DIR_STRIP_RE.sub(r"\1", filename))
         print >> output_file, "</div><!-- file-list -->"
         print >> output_file
         print >> output_file, '<pre class="license-text">'
         print >> output_file, html_escape(open(value[0]).read())
         print >> output_file, "</pre><!-- license-text -->"
         print >> output_file, "</td></tr><!-- same-license -->"
         print >> output_file
         print >> output_file
         print >> output_file

     # Finish off the file output
     print >> output_file, "</table>"
     print >> output_file, "</body></html>"
     output_file.close()

 def combine_notice_files_text(file_hash, input_dir, output_filename, file_title):
     """Combine notice files in FILE_HASH and output a text version to OUTPUT_FILENAME."""

     SRC_DIR_STRIP_RE = re.compile(input_dir + "(/.*).txt")
     output_file = open(output_filename, "wb")
     print >> output_file, file_title
     for value in file_hash.values():
       print >> output_file, "============================================================"
       print >> output_file, "Notices for file(s):"
       for filename in sorted(value):
         print >> output_file, SRC_DIR_STRIP_RE.sub(r"\1", filename)
       print >> output_file, "------------------------------------------------------------"
       print >> output_file, open(value[0]).read()
     output_file.close()

 def main(args):
     txt_output_file = args[0]
     html_output_file = args[1]
     file_title = args[2]

     # Find all the notice files and md5 them
     input_dir = os.path.normpath(args[3])
     files_with_same_hash = defaultdict(list)
     for root, dir, files in os.walk(input_dir):
         for file in files:
             if file.endswith(".txt"):
                 filename = os.path.join(root, file)
                 file_md5sum = md5sum(filename)
                 files_with_same_hash[file_md5sum].append(filename)


     print "Combining NOTICE files into HTML"
     combine_notice_files_html(files_with_same_hash, input_dir, html_output_file)
     print "Combining NOTICE files into text"
     combine_notice_files_text(files_with_same_hash, input_dir, txt_output_file, file_title)

 if __name__ == "__main__":
     main(args)
	#!/usr/bin/env python
	#
	# Copyright (C) 2012 The Android Open Source Project
	#
	# 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.
	"""
	Usage: generate-notice-files [plain text output file] [html output file] [file title] [directory of notices]

	Generate the Android notice files, including both text and html files.

	-h to display this usage message and exit.
	"""
	from collections import defaultdict
	import getopt
	import hashlib
	import itertools
	import os
	import os.path
	import re
	import sys

	MD5_BLOCKSIZE = 1024 * 1024
	HTML_ESCAPE_TABLE = {
	"&": "&",
	'"': """,
	"'": "'",
	">": ">",
	"<": "<",
	}

	try:
	opts, args = getopt.getopt(sys.argv[1:], "h")
	except getopt.GetoptError, err:
	print str(err)
	print __doc__
	sys.exit(2)

	for o, a in opts:
	if o == "-h":
	print __doc__
	sys.exit(2)
	else:
	print >> sys.stderr, "unhandled option %s" % (o,)

	if len(args) != 4:
	print """need exactly four arguments, the two output files, the file title
	and the directory containing notices, not %d""" % (len(args),)
	print __doc__
	sys.exit(1)

	def hexify(s):
	return ("%02x"*len(s)) % tuple(map(ord, s))

	def md5sum(filename):
	"""Calculate an MD5 of the file given by FILENAME,
	and return hex digest as a string.
	Output should be compatible with md5sum command"""

	f = open(filename, "rb")
	sum = hashlib.md5()
	while 1:
	block = f.read(MD5_BLOCKSIZE)
	if not block:
	break
	sum.update(block)
	f.close()
	return hexify(sum.digest())


	def html_escape(text):
	"""Produce entities within text."""
	return "".join(HTML_ESCAPE_TABLE.get(c,c) for c in text)

	HTML_OUTPUT_CSS="""
	<style type="text/css">
	body { padding: 0; font-family: sans-serif; }
	.same-license { background-color: #eeeeee; border-top: 20px solid white; padding: 10px; }
	.label { font-weight: bold; }
	.file-list { margin-left: 1em; color: blue; }
	</style>
	"""

	def combine_notice_files_html(file_hash, input_dir, output_filename):
	"""Combine notice files in FILE_HASH and output a HTML version to OUTPUT_FILENAME."""

	SRC_DIR_STRIP_RE = re.compile(input_dir + "(/.*).txt")

	# Set up a filename to row id table (anchors inside tables don't work in
	# most browsers, but href's to table row ids do)
	id_table = {}
	id_count = 0
	for value in file_hash.values():
	for filename in value:
	id_table[filename] = id_count
	id_count += 1

	# Open the output file, and output the header pieces
	output_file = open(output_filename, "wb")

	print >> output_file, "<html><head>"
	print >> output_file, HTML_OUTPUT_CSS
	print >> output_file, '</head><body topmargin="0" leftmargin="0" rightmargin="0" bottommargin="0">'

	# Output our table of contents
	print >> output_file, '<div class="toc">'
	print >> output_file, "<ul>"

	# Flatten the list of lists into a single list of filenames
	sorted_filenames = sorted(itertools.chain.from_iterable(file_hash.values()))

	# Print out a nice table of contents
	for filename in sorted_filenames:
	stripped_filename = SRC_DIR_STRIP_RE.sub(r"\1", filename)
	print >> output_file, '<li><a href="#id%d">%s</a></li>' % (id_table.get(filename), stripped_filename)

	print >> output_file, "</ul>"
	print >> output_file, "</div><!-- table of contents -->"
	# Output the individual notice file lists
	print >>output_file, '<table cellpadding="0" cellspacing="0" border="0">'
	for value in file_hash.values():
	print >> output_file, '<tr id="id%d"><td class="same-license">' % id_table.get(value[0])
	print >> output_file, '<div class="label">Notices for file(s):</div>'
	print >> output_file, '<div class="file-list">'
	for filename in sorted(value):
	print >> output_file, "%s <br/>" % (SRC_DIR_STRIP_RE.sub(r"\1", filename))
	print >> output_file, "</div><!-- file-list -->"
	print >> output_file
	print >> output_file, '<pre class="license-text">'
	print >> output_file, html_escape(open(value[0]).read())
	print >> output_file, "</pre><!-- license-text -->"
	print >> output_file, "</td></tr><!-- same-license -->"
	print >> output_file
	print >> output_file
	print >> output_file

	# Finish off the file output
	print >> output_file, "</table>"
	print >> output_file, "</body></html>"
	output_file.close()

	def combine_notice_files_text(file_hash, input_dir, output_filename, file_title):
	"""Combine notice files in FILE_HASH and output a text version to OUTPUT_FILENAME."""

	SRC_DIR_STRIP_RE = re.compile(input_dir + "(/.*).txt")
	output_file = open(output_filename, "wb")
	print >> output_file, file_title
	for value in file_hash.values():
	print >> output_file, "============================================================"
	print >> output_file, "Notices for file(s):"
	for filename in sorted(value):
	print >> output_file, SRC_DIR_STRIP_RE.sub(r"\1", filename)
	print >> output_file, "------------------------------------------------------------"
	print >> output_file, open(value[0]).read()
	output_file.close()

	def main(args):
	txt_output_file = args[0]
	html_output_file = args[1]
	file_title = args[2]

	# Find all the notice files and md5 them
	input_dir = os.path.normpath(args[3])
	files_with_same_hash = defaultdict(list)
	for root, dir, files in os.walk(input_dir):
	for file in files:
	if file.endswith(".txt"):
	filename = os.path.join(root, file)
	file_md5sum = md5sum(filename)
	files_with_same_hash[file_md5sum].append(filename)


	print "Combining NOTICE files into HTML"
	combine_notice_files_html(files_with_same_hash, input_dir, html_output_file)
	print "Combining NOTICE files into text"
	combine_notice_files_text(files_with_same_hash, input_dir, txt_output_file, file_title)

	if __name__ == "__main__":
	main(args)