lib/scripts/lyxpak.py

   1 #! /usr/bin/env python
   2 # -*- coding: utf-8 -*-
   3
   4 # file lyxpak.py
   5 # This file is part of LyX, the document processor.
   6 # Licence details can be found in the file COPYING.
   7
   8 # author Enrico Forestieri
   9 # author Richard Heck
  10
  11 # Full author contact details are available in file CREDITS
  12
  13 # This script creates a tar or zip archive with a lyx file and all included
  14 # files (graphics and so on). By default, the created archive is the standard
  15 # type on a given platform, such that a zip archive is created on Windows and
  16 # a gzip compressed tar archive on *nix. This can be controlled by command
  17 # line options, however.
  18
  19 import os, re, string, sys
  20 if sys.version_info < (2, 4, 0):
  21     from sets import Set as set
  22 from getopt import getopt
  23
  24 # Pre-compiled regular expressions.
  25 re_lyxfile = re.compile("\.lyx$")
  26 re_input = re.compile(r'^(.*)\\(input|include){(\s*)(\S+)(\s*)}.*$')
  27 re_package = re.compile(r'^(.*)\\(usepackage){(\s*)(\S+)(\s*)}.*$')
  28 re_class = re.compile(r'^(\\)(textclass)(\s+)(\S+)$')
  29 re_norecur = re.compile(r'^(.*)\\(verbatiminput|lstinputlisting|includegraphics\[*.*\]*){(\s*)(\S+)(\s*)}.*$')
  30 re_filename = re.compile(r'^(\s*)(filename)(\s+)(\S+)$')
  31 re_options = re.compile(r'^(\s*)options(\s+)(\S+)$')
  32 re_bibfiles = re.compile(r'^(\s*)bibfiles(\s+)(\S+)$')
  33
  34
  35 def usage(prog_name):
  36     msg = '''
  37 Usage: %s [-t] [-z] [-l path] [-o output_dir] file.lyx
  38 Options:
  39 -l: Path to lyx2lyx script
  40 -o: Directory for output
  41 -t: Create gzipped tar file
  42 -z: Create zip file
  43 By default, we create file.zip on Windows and file.tar.gz on *nix,
  44 with the file output to where file.lyx is, and we look for lyx2lyx
  45 in the known locations, querying LyX itself if necessary.
  46 '''
  47     return msg % prog_name
  48
  49
  50 def error(message):
  51     sys.stderr.write(message + '\n')
  52     sys.exit(1)
  53
  54
  55 def run_cmd(cmd):
  56     handle = os.popen(cmd, 'r')
  57     cmd_stdout = handle.read()
  58     cmd_status = handle.close()
  59     return cmd_status, cmd_stdout
  60
  61
  62 def find_exe(candidates, extlist, path):
  63     for prog in candidates:
  64         for directory in path:
  65             for ext in extlist:
  66                 full_path = os.path.join(directory, prog + ext)
  67                 if os.access(full_path, os.X_OK):
  68                     return prog, full_path
  69     return None, None
  70
  71
  72 def abspath(name):
  73     " Resolve symlinks and returns the absolute normalized name."
  74     newname = os.path.normpath(os.path.abspath(name))
  75     if os.name != 'nt':
  76         newname = os.path.realpath(newname)
  77     return newname
  78
  79
  80 def gather_files(curfile, incfiles, lyx2lyx):
  81     " Recursively gather files."
  82     curdir = os.path.dirname(abspath(curfile))
  83     is_lyxfile = re_lyxfile.search(curfile)
  84     if is_lyxfile:
  85         lyx2lyx_cmd = 'python "%s" "%s"' % (lyx2lyx, curfile)
  86         l2l_status, l2l_stdout = run_cmd(lyx2lyx_cmd)
  87         if l2l_status != None:
  88             error('%s failed to convert "%s"' % (lyx2lyx, curfile))
  89         lines = l2l_stdout.splitlines()
  90     else:
  91         input = open(curfile, 'rU')
  92         lines = input.readlines()
  93         input.close()
  94
  95     i = 0
  96     while i < len(lines):
  97         # Gather used files.
  98         recursive = True
  99         extlist = ['']
 100         match = re_filename.match(lines[i])
 101         if not match:
 102             match = re_input.match(lines[i])
 103             if not match:
 104                 match = re_package.match(lines[i])
 105                 extlist = ['.sty']
 106                 if not match:
 107                     match = re_class.match(lines[i])
 108                     extlist = ['.cls']
 109                     if not match:
 110                         match = re_norecur.match(lines[i])
 111                         extlist = ['', '.eps', '.pdf', '.png', '.jpg']
 112                         recursive = False
 113         if match:
 114             file = match.group(4).strip('"')
 115             if not os.path.isabs(file):
 116                 file = os.path.join(curdir, file)
 117             file_exists = False
 118             for ext in extlist:
 119                 if os.path.exists(file + ext):
 120                     file = file + ext
 121                     file_exists = True
 122                     break
 123             if file_exists:
 124                 incfiles.append(abspath(file))
 125                 if recursive:
 126                     gather_files(file, incfiles)
 127             i += 1
 128             continue
 129
 130         if not is_lyxfile:
 131             i += 1
 132             continue
 133
 134         # Gather bibtex *.bst files.
 135         match = re_options.match(lines[i])
 136         if match:
 137             file = match.group(3).strip('"')
 138             if not os.path.isabs(file):
 139                 file = os.path.join(curdir, file + '.bst')
 140             if os.path.exists(file):
 141                 incfiles.append(abspath(file))
 142             i += 1
 143             continue
 144
 145         # Gather bibtex *.bib files.
 146         match = re_bibfiles.match(lines[i])
 147         if match:
 148             bibfiles = match.group(3).strip('"').split(',')
 149             j = 0
 150             while j < len(bibfiles):
 151                 if os.path.isabs(bibfiles[j]):
 152                     file = bibfiles[j]
 153                 else:
 154                     file = os.path.join(curdir, bibfiles[j] + '.bib')
 155                 if os.path.exists(file):
 156                     incfiles.append(abspath(file))
 157                 j += 1
 158             i += 1
 159             continue
 160
 161         i += 1
 162
 163     return 0
 164
 165
 166 def find_lyx2lyx(progloc):
 167     # first we will see if the script is roughly where we are
 168     # i.e., we will assume we are in $SOMEDIR/scripts and look
 169     # for $SOMEDIR/lyx2lyx/lyx2lyx.
 170     ourpath = os.path.dirname(abspath(progloc))
 171     (upone, discard) = os.path.split(ourpath)
 172     tryit = os.path.join(upone, "lyx2lyx", "lyx2lyx")
 173     if os.access(tryit, os.X_OK):
 174         return tryit
 175
 176     # now we will try to query LyX itself to find the path.
 177     extlist = ['']
 178     if os.environ.has_key("PATHEXT"):
 179         extlist = extlist + os.environ["PATHEXT"].split(os.pathsep)
 180     lyx_exe, full_path = find_exe(["lyxc", "lyx"], extlist, path)
 181     if lyx_exe == None:
 182         error('Cannot find the LyX executable in the path.')
 183     cmd_status, cmd_stdout = run_cmd("%s -version 2>&1" % lyx_exe)
 184     if cmd_status != None:
 185         error('Cannot query LyX about the lyx2lyx script.')
 186     re_msvc = re.compile(r'^(\s*)(Host type:)(\s+)(win32)$')
 187     re_sysdir = re.compile(r'^(\s*)(LyX files dir:)(\s+)(\S+)$')
 188     lines = cmd_stdout.splitlines()
 189     for line in lines:
 190         match = re_msvc.match(line)
 191         if match:
 192             # The LyX executable was built with MSVC, so the
 193             # "LyX files dir:" line is unusable
 194             basedir = os.path.dirname(os.path.dirname(full_path))
 195             tryit = os.path.join(basedir, 'Resources', 'lyx2lyx', 'lyx2lyx')
 196             break
 197         match = re_sysdir.match(line)
 198         if match:
 199             tryit = os.path.join(match.group(4), 'lyx2lyx', 'lyx2lyx')
 200             break
 201
 202     if not os.access(tryit, os.X_OK):
 203         error('Unable to find the lyx2lyx script.')
 204     return tryit
 205
 206
 207 def main(args):
 208
 209     ourprog = args[0]
 210
 211     try:
 212       (options, argv) = getopt(args[1:], "htzl:o:")
 213     except:
 214       error(usage(ourprog))
 215
 216     # we expect the filename to be left
 217     if len(argv) != 1:
 218         error(usage(ourprog))
 219
 220     makezip = (os.name == 'nt')
 221     outdir = ""
 222     lyx2lyx = None
 223
 224     for (opt, param) in options:
 225       if opt == "-h":
 226         print usage(ourprog)
 227         sys.exit(0)
 228       elif opt == "-t":
 229         makezip = False
 230       elif opt == "-z":
 231         makezip = True
 232       elif opt == "-l":
 233         lyx2lyx = param
 234       elif opt == "-o":
 235         outdir = param
 236         if not os.path.isdir(outdir):
 237           error('Error: "%s" is not a directory.' % outdir)
 238
 239     lyxfile = argv[0]
 240     if not os.path.exists(lyxfile):
 241         error('File "%s" not found.' % lyxfile)
 242
 243     # Check that it actually is a LyX document
 244     input = open(lyxfile, 'rU')
 245     line = input.readline()
 246     input.close()
 247     if not (line and line.startswith('#LyX')):
 248         error('File "%s" is not a LyX document.' % lyxfile)
 249
 250     if makezip:
 251         import zipfile
 252     else:
 253         import tarfile
 254
 255     # Create a tar archive on *nix and a zip archive on Windows
 256     ar_ext = ".tar.gz"
 257     if makezip:
 258         ar_ext = ".zip"
 259
 260     ar_name = re_lyxfile.sub(ar_ext, abspath(lyxfile))
 261     if outdir:
 262         ar_name = os.path.join(abspath(outdir), os.path.basename(ar_name))
 263
 264     path = string.split(os.environ["PATH"], os.pathsep)
 265
 266     # Try to find the location of the lyx2lyx script
 267     if lyx2lyx == None:
 268         lyx2lyx = find_lyx2lyx(ourprog)
 269
 270     # Initialize the list with the specified LyX file and recursively
 271     # gather all required files (also from child documents).
 272     incfiles = [abspath(lyxfile)]
 273     gather_files(lyxfile, incfiles, lyx2lyx)
 274
 275     # Find the topmost dir common to all files
 276     if len(incfiles) > 1:
 277         topdir = os.path.commonprefix(incfiles)
 278     else:
 279         topdir = os.path.dirname(incfiles[0]) + os.path.sep
 280
 281     # Remove the prefix common to all paths in the list
 282     i = 0
 283     while i < len(incfiles):
 284         incfiles[i] = string.replace(incfiles[i], topdir, '', 1)
 285         i += 1
 286
 287     # Remove duplicates and sort the list
 288     incfiles = list(set(incfiles))
 289     incfiles.sort()
 290
 291     if topdir != '':
 292         os.chdir(topdir)
 293
 294     # Create the archive
 295     try:
 296         if os.name == 'nt':
 297             zip = zipfile.ZipFile(ar_name, "w", zipfile.ZIP_DEFLATED)
 298             for file in incfiles:
 299                 zip.write(file)
 300             zip.close()
 301         else:
 302             tar = tarfile.open(ar_name, "w:gz")
 303             for file in incfiles:
 304                 tar.add(file)
 305             tar.close()
 306     except:
 307         error('Failed to create LyX archive "%s"' % ar_name)
 308
 309     print 'LyX archive "%s" created successfully.' % ar_name
 310     return 0
 311
 312
 313 if __name__ == "__main__":
 314     main(sys.argv)