| 1 | #!/usr/bin/env python2
|
| 2 | """
|
| 3 | wc_html.py
|
| 4 |
|
| 5 | Filter for HTML
|
| 6 | """
|
| 7 | from __future__ import print_function
|
| 8 |
|
| 9 | import os
|
| 10 | import sys
|
| 11 |
|
| 12 |
|
| 13 | def log(msg, *args):
|
| 14 | if args:
|
| 15 | msg = msg % args
|
| 16 | print(msg, file=sys.stderr)
|
| 17 |
|
| 18 |
|
| 19 | def main(argv):
|
| 20 |
|
| 21 | id_ = argv[1]
|
| 22 | title = argv[2]
|
| 23 | comment = argv[3]
|
| 24 | tmp_dir = argv[4]
|
| 25 |
|
| 26 | total_lines = None
|
| 27 | rows = []
|
| 28 |
|
| 29 | # Parse
|
| 30 | num_files = 0
|
| 31 | for line in sys.stdin:
|
| 32 | line = line.strip()
|
| 33 | count, rel_path = line.split(None, 1)
|
| 34 | count = int(count)
|
| 35 |
|
| 36 | if rel_path == 'total':
|
| 37 | total_lines = count
|
| 38 | else:
|
| 39 | rows.append((count, rel_path))
|
| 40 | num_files += 1
|
| 41 |
|
| 42 | # Append to row
|
| 43 | index_tsv = os.path.join(tmp_dir, 'INDEX.tsv')
|
| 44 |
|
| 45 | anchor = 'i%s' % id_
|
| 46 |
|
| 47 | # Quirk: 'wc' doesn't print total line when there's one file.
|
| 48 | # TODO: Should we get rid of the terminal/text version of this report, and
|
| 49 | # not use 'wc'?
|
| 50 |
|
| 51 | if total_lines is None:
|
| 52 | total_lines = rows[-1][0]
|
| 53 |
|
| 54 | record = {
|
| 55 | 'id': id_,
|
| 56 | 'total_lines': total_lines,
|
| 57 | 'total_lines_str': '{:,}'.format(total_lines),
|
| 58 | 'num_files': num_files,
|
| 59 | 'title': title,
|
| 60 | 'anchor': anchor,
|
| 61 | 'comment': comment,
|
| 62 | }
|
| 63 |
|
| 64 | with open(index_tsv, 'a') as f:
|
| 65 | # The link looks like #i01
|
| 66 | f.write('%(title)s\t#%(anchor)s\t%(total_lines)d\t%(num_files)d\n' % record)
|
| 67 |
|
| 68 | # Write our HTML
|
| 69 | html = os.path.join(tmp_dir, '%02d.html' % int(id_))
|
| 70 |
|
| 71 | with open(html, 'w') as f:
|
| 72 | print('''\
|
| 73 | <a name="%(anchor)s"></a>
|
| 74 | <h2>%(title)s</h2>
|
| 75 | <p>%(comment)s</p>
|
| 76 | ''' % record, file=f)
|
| 77 |
|
| 78 | # Note: not using csv2html.py directly, since it's too many small files.
|
| 79 |
|
| 80 | print('''\
|
| 81 | <div class="wc">
|
| 82 | <pre class="counts">''', file=f)
|
| 83 |
|
| 84 | for count, rel_path in rows:
|
| 85 | count_str = '{:,}'.format(count)
|
| 86 | url = 'https://github.com/oilshell/oil/blob/master/%s' % rel_path
|
| 87 | print('<a href="%s">%-40s</a> %10s' % (url, rel_path, count_str), file=f)
|
| 88 |
|
| 89 | summary = '<span class="summary">%(total_lines_str)s lines in %(num_files)d files</span>' % record
|
| 90 | print('', file=f)
|
| 91 | print(summary, end='', file=f)
|
| 92 | print('''\
|
| 93 | </pre>
|
| 94 | </div>''' % record, file=f)
|
| 95 |
|
| 96 | return 0
|
| 97 |
|
| 98 |
|
| 99 | if __name__ == '__main__':
|
| 100 | try:
|
| 101 | sys.exit(main(sys.argv))
|
| 102 | except RuntimeError as e:
|
| 103 | print('FATAL: %s' % e, file=sys.stderr)
|
| 104 | sys.exit(1)
|