Fred Drake | 2ef38a7 | 1999-02-24 17:33:07 +0000 | [diff] [blame] | 1 | #! /usr/bin/env python |
| 2 | # -*- Python -*- |
| 3 | |
Fred Drake | 5f7832d | 1999-03-04 21:19:57 +0000 | [diff] [blame] | 4 | """usage: %(program)s [options] file... |
| 5 | |
| 6 | Supported options: |
| 7 | |
| 8 | --address addr |
| 9 | -a addr Set the address text to include at the end of the generated |
| 10 | HTML; this should be used for contact information. |
| 11 | --columns cols |
| 12 | -c cols Set the number of columns each index section should be |
| 13 | displayed in. The default is 1. |
| 14 | --help |
| 15 | -h Display this help message. |
| 16 | --letters |
| 17 | -l Split the output into sections by letter. |
| 18 | --output file |
| 19 | -o file Write output to 'file' instead of standard out. |
| 20 | --iconserver is Use 'is' as the directory containing icons for the |
| 21 | navigation bar. The default is 'icons'. |
| 22 | --title str Set the page title to 'str'. The default is 'Global |
| 23 | Module Index'. |
| 24 | --uplink url Set the upward link URL. The default is './'. |
| 25 | --uptitle str Set the upward link title. The default is 'Python |
| 26 | Documentation Index'. |
| 27 | """ |
Fred Drake | 2ef38a7 | 1999-02-24 17:33:07 +0000 | [diff] [blame] | 28 | import buildindex |
Fred Drake | 2ef38a7 | 1999-02-24 17:33:07 +0000 | [diff] [blame] | 29 | import os |
| 30 | import re |
| 31 | import string |
Fred Drake | 01a110b | 2000-10-05 05:14:26 +0000 | [diff] [blame] | 32 | import support |
Fred Drake | 2ef38a7 | 1999-02-24 17:33:07 +0000 | [diff] [blame] | 33 | import sys |
| 34 | |
| 35 | |
Fred Drake | 01a110b | 2000-10-05 05:14:26 +0000 | [diff] [blame] | 36 | class IndexOptions(support.Options): |
Fred Drake | e03e1fe | 2002-04-05 17:34:50 +0000 | [diff] [blame] | 37 | aesop_type = "links" |
| 38 | |
Fred Drake | 01a110b | 2000-10-05 05:14:26 +0000 | [diff] [blame] | 39 | def __init__(self): |
| 40 | support.Options.__init__(self) |
| 41 | self.add_args("l", ["letters"]) |
| 42 | self.letters = 0 |
Fred Drake | 5f7832d | 1999-03-04 21:19:57 +0000 | [diff] [blame] | 43 | |
Fred Drake | 01a110b | 2000-10-05 05:14:26 +0000 | [diff] [blame] | 44 | def handle_option(self, opt, val): |
| 45 | if opt in ("-l", "--letters"): |
| 46 | self.letters = 1 |
Fred Drake | 5f7832d | 1999-03-04 21:19:57 +0000 | [diff] [blame] | 47 | |
Fred Drake | 01a110b | 2000-10-05 05:14:26 +0000 | [diff] [blame] | 48 | def usage(self): |
| 49 | program = os.path.basename(sys.argv[0]) |
| 50 | print __doc__ % {"program": program} |
Fred Drake | 5f7832d | 1999-03-04 21:19:57 +0000 | [diff] [blame] | 51 | |
| 52 | |
Fred Drake | 0a0ef86 | 2000-11-28 16:20:50 +0000 | [diff] [blame] | 53 | class Node(buildindex.Node): |
Fred Drake | 9ca78ac | 2001-06-22 17:11:30 +0000 | [diff] [blame] | 54 | def __init__(self, link, str, seqno, platinfo): |
| 55 | self.annotation = platinfo or None |
| 56 | if str[0][-5:] == "</tt>": |
| 57 | str = str[:-5] |
| 58 | self.modname = str |
Fred Drake | 0a0ef86 | 2000-11-28 16:20:50 +0000 | [diff] [blame] | 59 | buildindex.Node.__init__(self, link, self.modname, seqno) |
Fred Drake | 9ca78ac | 2001-06-22 17:11:30 +0000 | [diff] [blame] | 60 | if platinfo: |
| 61 | s = '<tt class="module">%s</tt> %s' \ |
| 62 | % (self.modname, self.annotation) |
| 63 | else: |
| 64 | s = '<tt class="module">%s</tt>' % str |
| 65 | self.text = [s] |
Fred Drake | 0a0ef86 | 2000-11-28 16:20:50 +0000 | [diff] [blame] | 66 | |
| 67 | def __str__(self): |
Fred Drake | 9ca78ac | 2001-06-22 17:11:30 +0000 | [diff] [blame] | 68 | if self.annotation: |
| 69 | return '<tt class="module">%s</tt> %s' \ |
| 70 | % (self.modname, self.annotation) |
| 71 | else: |
| 72 | return '<tt class="module">%s</tt>' % self.modname |
Fred Drake | 0a0ef86 | 2000-11-28 16:20:50 +0000 | [diff] [blame] | 73 | |
Fred Drake | 2ef38a7 | 1999-02-24 17:33:07 +0000 | [diff] [blame] | 74 | _rx = re.compile( |
Fred Drake | 9ca78ac | 2001-06-22 17:11:30 +0000 | [diff] [blame] | 75 | "<dt><a href=['\"](module-.*\.html)(?:#l2h-\d+)?['\"]>" |
| 76 | "<tt class=['\"]module['\"]>([a-zA-Z_][a-zA-Z0-9_.]*)</tt>\s*(<em>" |
| 77 | "\(<span class=['\"]platform['\"]>.*</span>\)</em>)?</a>") |
Fred Drake | 2ef38a7 | 1999-02-24 17:33:07 +0000 | [diff] [blame] | 78 | |
| 79 | def main(): |
Fred Drake | 01a110b | 2000-10-05 05:14:26 +0000 | [diff] [blame] | 80 | options = IndexOptions() |
| 81 | options.variables["title"] = "Global Module Index" |
| 82 | options.parse(sys.argv[1:]) |
| 83 | args = options.args |
Fred Drake | 2ef38a7 | 1999-02-24 17:33:07 +0000 | [diff] [blame] | 84 | if not args: |
| 85 | args = ["-"] |
| 86 | # |
| 87 | # Collect the input data: |
| 88 | # |
| 89 | nodes = [] |
Fred Drake | 7f492ad | 1999-03-02 16:22:56 +0000 | [diff] [blame] | 90 | has_plat_flag = 0 |
Fred Drake | 2ef38a7 | 1999-02-24 17:33:07 +0000 | [diff] [blame] | 91 | for ifn in args: |
| 92 | if ifn == "-": |
| 93 | ifp = sys.stdin |
| 94 | dirname = '' |
| 95 | else: |
| 96 | ifp = open(ifn) |
| 97 | dirname = os.path.dirname(ifn) |
| 98 | while 1: |
| 99 | line = ifp.readline() |
| 100 | if not line: |
| 101 | break |
| 102 | m = _rx.match(line) |
| 103 | if m: |
| 104 | # This line specifies a module! |
Fred Drake | 9ca78ac | 2001-06-22 17:11:30 +0000 | [diff] [blame] | 105 | basename, modname, platinfo = m.group(1, 2, 3) |
| 106 | has_plat_flag = has_plat_flag or platinfo |
Fred Drake | 2ef38a7 | 1999-02-24 17:33:07 +0000 | [diff] [blame] | 107 | linkfile = os.path.join(dirname, basename) |
Fred Drake | 9ca78ac | 2001-06-22 17:11:30 +0000 | [diff] [blame] | 108 | nodes.append(Node('<a href="%s">' % linkfile, modname, |
| 109 | len(nodes), platinfo)) |
Fred Drake | 2ef38a7 | 1999-02-24 17:33:07 +0000 | [diff] [blame] | 110 | ifp.close() |
Fred Drake | 7f492ad | 1999-03-02 16:22:56 +0000 | [diff] [blame] | 111 | # |
| 112 | # Generate all output: |
| 113 | # |
Fred Drake | 2ef38a7 | 1999-02-24 17:33:07 +0000 | [diff] [blame] | 114 | num_nodes = len(nodes) |
Fred Drake | 7f492ad | 1999-03-02 16:22:56 +0000 | [diff] [blame] | 115 | # Here's the HTML generation: |
Fred Drake | 01a110b | 2000-10-05 05:14:26 +0000 | [diff] [blame] | 116 | parts = [options.get_header(), |
| 117 | buildindex.process_nodes(nodes, options.columns, options.letters), |
| 118 | options.get_footer(), |
Fred Drake | 5f7832d | 1999-03-04 21:19:57 +0000 | [diff] [blame] | 119 | ] |
Fred Drake | 7f492ad | 1999-03-02 16:22:56 +0000 | [diff] [blame] | 120 | if has_plat_flag: |
| 121 | parts.insert(1, PLAT_DISCUSS) |
| 122 | html = string.join(parts, '') |
Fred Drake | 2ef38a7 | 1999-02-24 17:33:07 +0000 | [diff] [blame] | 123 | program = os.path.basename(sys.argv[0]) |
Fred Drake | 01a110b | 2000-10-05 05:14:26 +0000 | [diff] [blame] | 124 | fp = options.get_output_file() |
Fred Drake | b258bed | 2001-02-12 15:30:22 +0000 | [diff] [blame] | 125 | fp.write(string.rstrip(html) + "\n") |
Fred Drake | 01a110b | 2000-10-05 05:14:26 +0000 | [diff] [blame] | 126 | if options.outputfile == "-": |
Fred Drake | b258bed | 2001-02-12 15:30:22 +0000 | [diff] [blame] | 127 | sys.stderr.write("%s: %d index nodes\n" % (program, num_nodes)) |
Fred Drake | 2ef38a7 | 1999-02-24 17:33:07 +0000 | [diff] [blame] | 128 | else: |
Fred Drake | 2ef38a7 | 1999-02-24 17:33:07 +0000 | [diff] [blame] | 129 | print |
| 130 | print "%s: %d index nodes" % (program, num_nodes) |
| 131 | |
| 132 | |
Fred Drake | 7f492ad | 1999-03-02 16:22:56 +0000 | [diff] [blame] | 133 | PLAT_DISCUSS = """ |
| 134 | <p> Some module names are followed by an annotation indicating what |
| 135 | platform they are available on.</p> |
| 136 | |
Fred Drake | 2ef38a7 | 1999-02-24 17:33:07 +0000 | [diff] [blame] | 137 | """ |
| 138 | |
Fred Drake | 2ef38a7 | 1999-02-24 17:33:07 +0000 | [diff] [blame] | 139 | |
| 140 | if __name__ == "__main__": |
| 141 | main() |