lib/doc/doc_toc.py

   1 #! /usr/bin/env python
   2 # -*- coding: utf-8 -*-
   3 # This file is part of the LyX Documentation
   4 # Copyright (C) 2004 José Matos <jamatos@lyx.org>
   5 #
   6 # This program is free software; you can redistribute it and/or
   7 # modify it under the terms of the GNU General Public License
   8 # as published by the Free Software Foundation; either version 2
   9 # of the License, or (at your option) any later version.
  10 #
  11 # This program is distributed in the hope that it will be useful,
  12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 # GNU General Public License for more details.
  15 #
  16 # You should have received a copy of the GNU General Public License
  17 # along with this program; if not, write to the Free Software
  18 # Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  19
  20 # This script is called by the script depend.py to create a "master table of contents"
  21 # for a set of LyX docs.
  22 # It does so by going through the files and printing out all of the
  23 # chapter, section, and sub(sub)section headings out. (It numbers the
  24 # sections sequentially; hopefully noone's using Section* in the docs.)
  25 # It is called using this syntax:
  26 # depend.py doc_toc.py SetOfDocuments
  27 # where SetOfDocuments is a set of documents
  28
  29 import sys
  30 import os
  31
  32 if __name__ == "__main__":
  33     srcdir = os.path.dirname(sys.argv[0])
  34     if srcdir == '':
  35         srcdir = '.'
  36     sys.path.insert(0, srcdir + "/../lyx2lyx")
  37
  38 # when doc_toc is imported by scons, sys.path is set over there
  39 import parser_tools
  40 import LyX
  41 import depend
  42
  43 # Specific language information
  44 # info["isoname"] = (language, language_quotes, TOC_translated)
  45 info = { 'cs' : ('czech', 'german', u"Obsah dokumentace LyXu"),
  46          'da' : ('danish', 'german', u"Indholdsfortegnelse over LyX's dokumentation"),
  47          'de' : ('german', 'german', u"Inhaltsverzeichnis LyX Dokumentation"),
  48          'es' : ('spanish', 'spanish', u"Índice general LyX documentation"),
  49          'fr' : ('french', 'french', u"Plan de la documentation LyX"),
  50          'ru' : ('russian', 'english', u"LyX Documentation Table of Contents"),
  51          'sl' : ('slovene', 'german', u"Kazalo dokumentacije LyXa"),
  52          'en' : ('english', 'english', u"LyX Documentation Table of Contents")}
  53
  54 def usage(pname):
  55     print """Usage: %s lang output
  56
  57     lang is the language to build the TOC file,
  58 """ % pname
  59
  60
  61 transform_table = {'Title' : 'Section*', 'Chapter': 'Enumerate',
  62                    'Section':'Enumerate', 'Subsection': 'Enumerate',
  63                    'Subsubsection' : 'Enumerate'}
  64
  65 def build_from_toc(par_list):
  66     if not par_list:
  67         return []
  68
  69     if par_list[0].name == 'Title':
  70         par = par_list[0]
  71         par.name = transform_table[par.name]
  72
  73         if len(par_list) == 1:
  74             return par_list
  75
  76         for i in range(1, len(par_list)):
  77             if par_list[i].name == 'Title':
  78                 return [par] + \
  79                         build_from_toc(par_list[1:i]) + \
  80                         build_from_toc(par_list[i:])
  81
  82         return [par] + build_from_toc(par_list[1:])
  83
  84     if par_list[0].name in ('Chapter', 'Section', 'Subsection', 'Subsubsection'):
  85         return nest_struct(par_list[0].name, par_list)
  86
  87
  88 def nest_struct(name, par_list):
  89     if par_list[0].name == name:
  90         par = par_list[0]
  91         par.name = transform_table[par.name]
  92
  93         if len(par_list) == 1:
  94             return par_list
  95
  96         for i in range(1, len(par_list)):
  97             if par_list[i].name == name:
  98                 par.child = build_from_toc(par_list[1:i])
  99                 return [par] + build_from_toc(par_list[i:])
 100         par.child = build_from_toc(par_list[1:])
 101         return [ par ]
 102
 103
 104 def build_toc(output, documents, lang=None):
 105     # Determine existing translated documents for that language.
 106     toc_general = []
 107     for file in documents:
 108         file = LyX.File(input= file)
 109         file.convert()
 110         toc_general.extend(file.get_toc())
 111
 112     # if lang is not given, guess from document names. (Used in scons build)
 113     if lang is None:
 114         lang = 'en'
 115         for file in documents:
 116             dir = file.split(os.sep)[-2]
 117             if dir in info.keys():
 118                 lang = dir
 119     file = LyX.NewFile(output = output)
 120     data = info[lang]
 121     file.set_header(language = data[0], language_quotes = data[1], inputencoding = "auto")
 122     file.language = data[0]
 123     file.encoding = "utf-8"
 124     body = [ LyX.Paragraph('Title', [data[2]])]
 125     body.extend(build_from_toc(toc_general))
 126     file.set_body(body)
 127     file.write()
 128
 129
 130 def main(argv):
 131     if len(argv) != 3:
 132         usage(argv[0])
 133         sys.exit(1)
 134
 135     lang = argv[1]
 136     if not os.path.isdir(os.path.join(argv[2], lang)):
 137         # need to create lang dir if build dir != src dir
 138         os.mkdir(os.path.join(argv[2], lang))
 139
 140     # choose language files
 141     if lang == 'en':
 142         output = os.path.join(argv[2], 'TOC.lyx')
 143     else:
 144         output = os.path.join(argv[2], lang, 'TOC.lyx')
 145         # fallback
 146         if lang not in info:
 147             lang = 'en'
 148
 149     build_toc(output, depend.documents(srcdir, lang), lang)
 150
 151
 152 if __name__ == "__main__":
 153     main(sys.argv)