Allow the user to specify the "biggest" section type from the command line;

freddrake · freddrake · commit 473a90e8310d · 1998-03-07T15:34:50.000Z
default is "chapter".  Use 'python toc2bkm.py -c section' to use with
Python HOWTO documents.
diff --git a/Doc/tools/toc2bkm.py b/Doc/tools/toc2bkm.py
@@ -6,6 +6,7 @@
 already uses that extension.  Let's avoid clashing.
 """
 
+import getopt
 import os
 import re
 import string
@@ -38,10 +39,10 @@
     ('subsubsection', 'chapter'): 3,
     }
 
-def parse_toc(fp):
+def parse_toc(fp, bigpart=None):
     toc = top = []
     stack = [toc]
-    level = 'chapter'
+    level = bigpart or 'chapter'
     lineno = 0
     while 1:
 	line = fp.readline()
@@ -105,10 +106,18 @@ def write_toc_entry(entry, fp, layer):
 
 
 def main():
-    base, ext = os.path.splitext(sys.argv[1])
-    ext = ext or ".toc"
-    toc = parse_toc(open(base + ext))
-    write_toc(toc, open(base + ".bkm", "w"))
+    bigpart = None
+    opts, args = getopt.getopt(sys.argv[1:], "c:")
+    if opts:
+	bigpart = opts[0][1]
+    if not args:
+	usage()
+	sys.exit(2)
+    for filename in args:
+	base, ext = os.path.splitext(filename)
+	ext = ext or ".toc"
+	toc = parse_toc(open(base + ext), bigpart)
+	write_toc(toc, open(base + ".bkm", "w"))
 
 
 if __name__ == "__main__":
