cpython/Doc/modindex.py

63 lines
1.5 KiB
Python
Executable File

#! /usr/bin/env python
"""Script to convert raw module index data to module index."""
import os
import string
import sys
def parse_line(input):
lineno = string.split(input)[-1]
module = string.strip(input[:len(input)-(len(lineno)+1)])
return module, lineno
def cmp_items((s1, line1), (s2, line2)):
rc = cmp(string.lower(s1), string.lower(s2))
if rc == 0:
# Make the lower-case version come first since the upper-case
# version is usually a helper module for constants and such.
rc = cmp(s2, s1)
return rc
def main():
if sys.argv[1:]:
infile = sys.argv[1]
else:
infile = "-"
if infile == "-":
ifp = sys.stdin
ofp = sys.stdout
sys.stdout = sys.stderr
else:
ifp = open(infile)
base, ext = os.path.splitext(infile)
outfile = base + ".ind"
ofp = open(outfile, "w")
ofp.write("\\begin{theindex}\n\label{modindex}\n\n")
lines = ifp.readlines()
for i in range(len(lines)):
if lines[i][0] == '\\':
lines[i] = '\1' + lines[i]
lines = map(parse_line, lines)
lines.sort(cmp_items)
prev_letter = lines[0][0][0]
if prev_letter == '\1':
prev_letter = lines[0][0][1]
prev_letter = string.lower(prev_letter)
for module, lineno in lines:
if module[0] == '\1':
module = module[1:]
if string.lower(module[0]) != prev_letter:
ofp.write("\n \\indexspace\n\n")
prev_letter = string.lower(module[0])
ofp.write(" \\item {\\tt %s}, %s\n" % (module, lineno))
ofp.write("\n\\end{theindex}\n")
if __name__ == "__main__":
main()