python/lilylib.py

   1 # This file is part of LilyPond, the GNU music typesetter.
   2 #
   3 # Copyright (C) 1998--2011 Han-Wen Nienhuys <hanwen@xs4all.nl>
   4 #                Jan Nieuwenhuizen <janneke@gnu.org>
   5 #
   6 # LilyPond is free software: you can redistribute it and/or modify
   7 # it under the terms of the GNU General Public License as published by
   8 # the Free Software Foundation, either version 3 of the License, or
   9 # (at your option) any later version.
  10 #
  11 # LilyPond 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 LilyPond.  If not, see <http://www.gnu.org/licenses/>.
  18
  19 import __main__
  20 import glob
  21 import os
  22 import re
  23 import shutil
  24 import sys
  25 import optparse
  26 import time
  27
  28 ################################################################
  29 # Users of python modules should include this snippet
  30 # and customize variables below.
  31
  32
  33 # Python 2.5 only accepts strings with proper Python internal encoding
  34 # (i.e. ASCII or Unicode) when writing to stdout/stderr, so we must
  35 # use ugettext iso gettext, and encode the string when writing to
  36 # stdout/stderr
  37
  38 localedir = '@localedir@'
  39 try:
  40     import gettext
  41     t = gettext.translation ('lilypond', localedir)
  42     _ = t.ugettext
  43 except:
  44     def _ (s):
  45         return s
  46 underscore = _
  47
  48 # Urg, Python 2.4 does not define stderr/stdout encoding
  49 # Maybe guess encoding from LANG/LC_ALL/LC_CTYPE?
  50
  51 reload (sys)
  52 sys.setdefaultencoding ('utf-8')
  53 import codecs
  54 sys.stdout = codecs.getwriter ('utf8') (sys.stdout)
  55 sys.stderr = codecs.getwriter ('utf8') (sys.stderr)
  56
  57 def encoded_write(f, s):
  58     f.write (s.encode (f.encoding or 'utf-8', 'replace'))
  59
  60 # ugh, Python 2.5 optparse requires Unicode strings in some argument
  61 # functions, and refuse them in some other places
  62 def display_encode (s):
  63     return s.encode (sys.stderr.encoding or 'utf-8', 'replace')
  64
  65 # Lilylib globals.
  66 program_version = '@TOPLEVEL_VERSION@'
  67 program_name = os.path.basename (sys.argv[0])
  68
  69
  70 # Check if program_version contains @ characters. This will be the case if
  71 # the .py file is called directly while building the lilypond documentation.
  72 # If so, try to check for the env var LILYPOND_VERSION, which is set by our
  73 # makefiles and use its value.
  74 at_re = re.compile (r'@')
  75 if at_re.match (program_version):
  76     if os.environ.has_key('LILYPOND_VERSION'):
  77         program_version = os.environ['LILYPOND_VERSION']
  78     else:
  79         program_version = "unknown"
  80
  81
  82 # Logging framework: We have the following output functions:
  83 #    error
  84 #    warning
  85 #    progress
  86 #    debug
  87
  88 loglevels = {"NONE":0, "ERROR":1, "WARN":2, "BASIC":3, "PROGRESS":4, "INFO":5, "DEBUG":6}
  89
  90 loglevel = loglevels["PROGRESS"]
  91
  92 def set_loglevel (l):
  93     global loglevel
  94     newlevel = loglevels.get (l, -1)
  95     if newlevel > 0:
  96         debug_output (_ ("Setting loglevel to %s") % l)
  97         loglevel = newlevel
  98     else:
  99         error (_ ("Unknown or invalid loglevel '%s'") % l)
 100
 101
 102 def handle_loglevel_option (option, opt_str, value, parser, *args):
 103     if value:
 104         set_loglevel (value);
 105     elif args:
 106         set_loglevel (args[0]);
 107
 108 def is_loglevel (l):
 109     global loglevel
 110     return loglevel >= loglevels[l];
 111
 112 def is_verbose ():
 113     return is_loglevel ("DEBUG")
 114
 115 def stderr_write (s):
 116     encoded_write (sys.stderr, s)
 117
 118 def print_logmessage (level, s, fullmessage = True, newline = True):
 119     if (is_loglevel (level)):
 120         if fullmessage:
 121             stderr_write (program_name + ": " + s + '\n')
 122         elif newline:
 123             stderr_write (s + '\n')
 124         else:
 125             stderr_write (s)
 126
 127 def error (s):
 128     print_logmessage ("ERROR", _ ("error: %s") % s);
 129
 130 def warning (s):
 131     print_logmessage ("WARN", _ ("warning: %s") % s);
 132
 133 def basic_progress (s):
 134     print_logmessage ("BASIC", s);
 135
 136 def progress (s, fullmessage = False, newline = True):
 137     print_logmessage ("PROGRESS", s, fullmessage, newline);
 138
 139 def debug_output (s, fullmessage = False, newline = True):
 140     print_logmessage ("DEBUG", s, fullmessage, newline);
 141
 142
 143
 144 def require_python_version ():
 145     if sys.hexversion < 0x02040000:
 146         error ("Python 2.4 or newer is required to run this program.\n\
 147 Please upgrade Python from http://python.org/download/, and if you use MacOS X,\n\
 148 please read 'Setup for MacOS X' in Application Usage.")
 149         os.system ("open http://python.org/download/")
 150         sys.exit (2)
 151
 152 # Modified version of the commands.mkarg(x), which always uses
 153 # double quotes (since Windows can't handle the single quotes:
 154 def mkarg(x):
 155     if os.name == 'nt':
 156         return x
 157     s = ' "'
 158     for c in x:
 159         if c in '\\$"`':
 160             s = s + '\\'
 161         s = s + c
 162     s = s + '"'
 163     return s
 164
 165 def command_name (cmd):
 166     # Strip all stuf after command,
 167     # deal with "((latex ) >& 1 ) .." too
 168     cmd = re.match ('([\(\)]*)([^\\\ ]*)', cmd).group (2)
 169     return os.path.basename (cmd)
 170
 171 def subprocess_system (cmd,
 172                        ignore_error=False,
 173                        progress_p=True,
 174                        be_verbose=False,
 175                        redirect_output=False,
 176                        log_file=None):
 177     import subprocess
 178
 179     show_progress= progress_p
 180     name = command_name (cmd)
 181     error_log_file = ''
 182
 183     if redirect_output:
 184         progress (_ ("Processing %s.ly") % log_file)
 185     else:
 186         if be_verbose:
 187             show_progress = 1
 188             progress (_ ("Invoking `%s\'") % cmd)
 189         else:
 190             progress ( _("Running %s...") % name)
 191
 192     stdout_setting = None
 193     stderr_setting = None
 194     if not show_progress:
 195         stdout_setting = subprocess.PIPE
 196
 197     if redirect_output:
 198         stderr_filename = log_file + '.log'
 199         stderr_setting = open(stderr_filename, 'w')
 200
 201     proc = subprocess.Popen (cmd,
 202                              shell=True,
 203                              universal_newlines=True,
 204                              stdout=stdout_setting,
 205                              stderr=stderr_setting)
 206
 207     log = ''
 208
 209     if redirect_output:
 210         while proc.poll()==None:
 211             time.sleep(0.01)
 212         retval = proc.returncode
 213         stderr_setting.close()
 214     else:
 215         if show_progress:
 216             retval = proc.wait()
 217         else:
 218             log = proc.communicate ()
 219             retval = proc.returncode
 220
 221     if retval:
 222         print >>sys.stderr, 'command failed:', cmd
 223         if retval < 0:
 224             print >>sys.stderr, "Child was terminated by signal", -retval
 225         elif retval > 0:
 226             print >>sys.stderr, "Child returned", retval
 227
 228         if ignore_error:
 229             print >>sys.stderr, "Error ignored by lilylib"
 230         else:
 231             if not show_progress:
 232                 print log[0]
 233                 print log[1]
 234             sys.exit (1)
 235
 236     return abs (retval)
 237
 238 def ossystem_system (cmd,
 239                      ignore_error=False,
 240                      progress_p=True,
 241                      be_verbose=False,
 242                      redirect_output=False,
 243                      log_file=None):
 244
 245
 246     name = command_name (cmd)
 247     if be_verbose:
 248         show_progress = 1
 249         progress (_ ("Invoking `%s\'") % cmd)
 250     else:
 251         progress ( _("Running %s...") % name)
 252
 253     retval = os.system (cmd)
 254     if retval:
 255         print >>sys.stderr, 'command failed:', cmd
 256         if retval < 0:
 257             print >>sys.stderr, "Child was terminated by signal", -retval
 258         elif retval > 0:
 259             print >>sys.stderr, "Child returned", retval
 260
 261         if ignore_error:
 262             print >>sys.stderr, "Error ignored"
 263         else:
 264             sys.exit (1)
 265
 266     return abs (retval)
 267
 268
 269 system = subprocess_system
 270 if sys.platform == 'mingw32':
 271
 272     ## subprocess x-compile doesn't work.
 273     system = ossystem_system
 274
 275 def strip_extension (f, ext):
 276     (p, e) = os.path.splitext (f)
 277     if e == ext:
 278         e = ''
 279     return p + e
 280
 281
 282 def search_exe_path (name):
 283     p = os.environ['PATH']
 284     exe_paths = p.split (':')
 285     for e in exe_paths:
 286         full = os.path.join (e, name)
 287         if os.path.exists (full):
 288             return full
 289     return None
 290
 291
 292 def print_environment ():
 293     for (k,v) in os.environ.items ():
 294         sys.stderr.write ("%s=\"%s\"\n" % (k, v))
 295
 296 class NonDentedHeadingFormatter (optparse.IndentedHelpFormatter):
 297     def format_heading(self, heading):
 298         if heading:
 299             return heading[0].upper() + heading[1:] + ':\n'
 300         return ''
 301     def format_option_strings(self, option):
 302         sep = ' '
 303         if option._short_opts and option._long_opts:
 304             sep = ','
 305
 306         metavar = ''
 307         if option.takes_value():
 308             metavar = '=%s' % option.metavar or option.dest.upper()
 309
 310         return "%3s%s %s%s" % (" ".join (option._short_opts),
 311                                sep,
 312                                " ".join (option._long_opts),
 313                                metavar)
 314
 315     # Only use one level of indentation (even for groups and nested groups),
 316     # since we don't indent the headeings, either
 317     def indent(self):
 318         self.current_indent = self.indent_increment
 319         self.level += 1
 320     def dedent(self):
 321         self.level -= 1
 322         if self.level <= 0:
 323             self.current_indent = ''
 324             self.level = 0;
 325
 326     def format_usage(self, usage):
 327         return _("Usage: %s") % usage + '\n'
 328
 329     def format_description(self, description):
 330         return description
 331
 332 def get_option_parser (*args, **kwargs):
 333     p = optparse.OptionParser (*args, **kwargs)
 334     p.formatter = NonDentedHeadingFormatter ()
 335     p.formatter.set_parser (p)
 336     return p