1 #!/usr/bin/env python

     2 # vim: set fileencoding=utf-8 :

     3 

     4 """\

     5 YML 2 compiler version 5.8

     6 Copyleft (c), 2009-2011, Volker Birk  http://fdik.org/yml/

     7 

     8 """

     9 

    10 import sys, os, codecs, locale

    11 import fileinput, unicodedata

    12 from optparse import OptionParser

    13 

    14 from pyPEG import parse, u

    15 from yml2 import ymlCStyle, comment, oldSyntax

    16 import backend

    17 

    18 def printInfo(option, opt_str, value, parser):

    19     sys.stdout.write(__doc__)

    20 

    21 def w(msg):

    22     if isinstance(msg, BaseException):

    23         try:

    24             msg = str(msg) + "\n"

    25         except:

    26             msg = u(msg) + u"\n"

    27     if type(msg) is unicode:

    28         msg = codecs.encode(msg, sys.stderr.encoding)

    29     sys.stderr.write(msg)

    30 

    31 

    32 def main():

    33     optParser = OptionParser()

    34     optParser.add_option("-C", "--old-syntax", action="store_true", dest="old_syntax",

    35             help="syntax of YML 2 version 1.x (compatibility mode)", default=False)

    36     optParser.add_option("-D", "--emit-linenumbers", action="store_true", dest="emitlinenumbers",

    37             help="emit line numbers into the resulting XML for debugging purposes", default=False)

    38     optParser.add_option("-E", "--encoding", dest="encoding", metavar="ENCODING", default=locale.getdefaultlocale()[1],

    39             help="encoding of input files (default to locale)")

    40     optParser.add_option("-I", "--include", dest="includePathText", metavar="INCLUDE_PATH",

    41             help="precede YML_PATH by a colon separated INCLUDE_PATH to search for include files")

    42     optParser.add_option("-m", "--omit-empty-parm-tags", action="store_true", dest="omitemptyparm",

    43             help="does nothing (only there for compatibility reasons)", default=False)

    44     optParser.add_option("-n", "--normalization", dest="normalization", metavar="NORMALIZATION", default="NFC",

    45             help="Unicode normalization (none, NFD, NFKD, NFC, NFKC, FCD, default is NFC)")

    46     optParser.add_option("-o", "--output", dest="outputFile", metavar="FILE",

    47             help="place output in file FILE")

    48     optParser.add_option("-p", "--parse-only", action="store_true", dest="parseonly",

    49             help="parse only, then output pyAST as text to stdout", default=False)

    50     optParser.add_option("-V", "--version", action="callback", callback=printInfo, help="show version info")

    51     (options, args) = optParser.parse_args()

    52 

    53     if options.old_syntax:

    54         oldSyntax()

    55 

    56     if options.emitlinenumbers:

    57         backend.emitlinenumbers = True

    58 

    59     backend.encoding = options.encoding

    60 

    61     try:

    62         if options.includePathText:

    63             backend.includePath = options.includePathText.split(':')

    64 

    65         dirs = os.environ.get('YML_PATH', '.').split(':')

    66         backend.includePath.extend(dirs)

    67 

    68         files = fileinput.input(args, mode="rU", openhook=fileinput.hook_encoded(options.encoding))

    69 

    70         ymlC = ymlCStyle()

    71         result = parse(ymlC, files, True, comment, packrat=True)

    72 

    73         if options.parseonly:

    74             print(result)

    75         else:

    76             result = backend.finish(result)

    77             if options.normalization != "none":

    78                 result = unicodedata.normalize(options.normalization, result)

    79 

    80             if options.outputFile and options.outputFile != "-":

    81                 outfile = open(options.outputFile, "w")

    82                 outfile.write(codecs.encode(result, options.encoding))

    83                 outfile.close()

    84             else:

    85                 print(codecs.encode(result, options.encoding))

    86 

    87     except KeyboardInterrupt:

    88         w("\n")

    89         sys.exit(1)

    90     except KeyError as msg:

    91         w(u"not found: " + u(msg) + u"\n")

    92         sys.exit(4)

    93     except LookupError as msg:

    94         w(u"not found: " + u(msg) + u"\n")

    95         sys.exit(4)

    96     except Exception as msg:

    97         w(msg)

    98         sys.exit(5)

    99 

   100 

   101 if __name__ == "__main__":

   102     sys.exit(main())