JIT/jit_cli.py at master · prashant-jayan21/JIT · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
#!/usr/bin/python
from optparse import OptionParser
from jit_interpreter import *
import re

try:
    import readline
except ImportError:
    # no readline on Windows
    pass

def main():
    # This code allows us to run any program file using the CLI.
    option_parser = OptionParser()
    option_parser.add_option("-f", "--file", dest="filename", help="JIT program filename", type="string")
    option_parser.add_option("--debug", action="store_true", dest="debug", default=False)
    (options, args) = option_parser.parse_args()

    if (options.filename):
        # Note: Input file must end with a new line character
        interpreter = Interpreter(re.sub('.txt', '.gen.py', options.filename))

        with open(options.filename, "r") as file:
            data = file.readlines()

        overflow = ""
        for i, line in enumerate(data):
            line = overflow + line

            if line.count('{') == line.count('}'):
                if options.debug:
                    print "=======%s" %line

                interpreter.execute_txt(line.strip(), line=i, debug=options.debug)
                overflow = ""
            else:
                overflow = line

    else:
        code_locals = {}
        from jitlib.node import Node, Keyword, node_get
        from jitlib.jit_nlp import extract_entity_names, extract_tags
        from jitlib.graf import search, pull
        from jitlib.database import engine, Base
        Base.metadata.create_all(engine)

        interpreter = Interpreter()
        while True:
            try:
                line = raw_input("JIT> ")
            except KeyboardInterrupt:
                # Execute, don't crash
                # This is the Small Slant font from
                # http://patorjk.com/software/taag/
                # for the text 'G O O D B Y E !'
                print '''
      _____  ____    ____    ___    ___  __  __  ____  __
     / ___/ / __ \  / __ \  / _ \  / _ ) \ \/ / / __/ / /
    / (_ / / /_/ / / /_/ / / // / / _  |  \  / / _/  /_/
    \___/  \____/  \____/ /____/ /____/   /_/ /___/ (_)
                '''
                exit()

            while line.count('{') != line.count('}'):
                line = line + raw_input("... ")

            try:
                code = interpreter.execute_txt(line, debug=options.debug)
                exec code
            except NameError:
                # Already prints error
                pass
            except AttributeError:
                print "There was a problem parsing that line (AttributeError)."


    if options.debug:
        print '''
               ____  __  __  _____  ____  ____  ____   __ *
              / __/ / / / / / ___/ / __/ / __/ / __/  / /
             _\ \  / /_/ / / /__  / _/  _\ \  _\ \   /_/
            /___/  \____/  \___/ /___/ /___/ /___/  (_)

            * Maybe. We didn't crash. That's good, right?
        '''

if __name__ == "__main__":
    main()