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())