1 files changed, 41 insertions, 8 deletions
diff --git a/compilador/main.py b/compilador/main.py
index 186ca83..98e23c9 100644
--- a/compilador/main.py
+++ b/compilador/main.py
@@ -1,21 +1,30 @@
-import sys, getopt
+import sys, getopt, os, traceback
+from enum import Enum
 from tkinter import *
 from tkinter import ttk, filedialog
 from lexer import *
 
+class Step(Enum):
+    LEXICO = 0
+    SINTACTICO = 1
+    SEMANTICO = 2
+
 class Main:
     input_file = None
     output_file = None
     output_table = False
+    step = None
 
     def print_help (self, arg0):
         print("Uso: % s -i entrada.es -o salida.es" % arg0)
-        print("     % s -i entrada.es -o salida.es -t")
+        print("     % s -i entrada.es -o salida.es [-l|-p|-s] [-t]" % arg0)
         print("     % s -h" % arg0)
 
     def main(self, argv):
         try:
-            opts, args = getopt.getopt(argv[1:], "hi:o:t", ["input=", "output=", "table"])
+            opts, args = getopt.getopt(argv[1:], "hi:o:lpst", [
+                "input=", "output=", "lex", "parse", "semantic", "table"
+            ])
         except getopt.GetoptError as err:
             print(err)
             print_help(argv[0]);
@@ -30,16 +39,40 @@ class Main:
                 self.output_file = a
             elif o in ("-t", "--table"):
                 self.output_table = True
+            elif o in ("-l", "--lex"):
+                self.step = Step.LEXICO
+            elif o in ("-p", "--parse"):
+                self.step = Step.SINTACTICO
+            elif o in ("-s", "--semantic"):
+                self.step = Step.SEMANTICO
             else:
                 assert False, "opción desconocida"
 
         if self.input_file and self.output_file:
-            with open(self.input_file) as f:
-                data = f.read()
-                if self.output_table:
-                    Lexer(data, self.input_file).inicio()
+            table_file = self.input_file + '.tab'
+            delete_tab = not self.step and not self.output_table and not os.path.exists(table_file)
+            try:
+                if self.step == Step.LEXICO:
+                    sys.exit(Lexer(self.input_file).inicio())
+                elif self.step == Step.SINTACTICO:
+                    sys.exit(Parser(self.input_file).inicio())
+                elif self.step == Step.SEMANTICO:
+                    print("NOT IMPLEMENTED")
+                    sys.exit(1)
                 else:
-                    Lexer(data, None).inicio()
+                    if Lexer(self.input_file).inicio() != 0:
+                        raise Exception("Error léxico")
+                    if Parser(self.input_file).inicio() != 0:
+                        raise Exception("Error sintáctico")
+            except Exception as e:
+                traceback.print_exception(type(e), e, e.__traceback__)
+                sys.exit(1)
+            # Borrar tabla de símbolos
+            if delete_tab:
+                os.remove(table_file)
+        else:
+            self.print_help(argv[0])
+            sys.exit(2)
 
 if __name__ == "__main__":
     Main().main(sys.argv)