Refactor sg_parser.py

tuura · May 14, 2018 · bf927b5 · bf927b5
1 parent e6c4d54
commit bf927b5
Showing 1 changed file with 7 additions and 20 deletions.
diff --git a/sg_parser.py b/sg_parser.py
@@ -16,26 +16,13 @@ def load_sg(file):
 
         words = line.split()
 
-        if words[0] in ignored:
-            pass
-
-        elif words[0] == ".model":
-            sg["model"] = words[1]
-
-        elif words[0] == ".inputs":
-            sg["inputs"] = words[1:]
-
-        elif words[0] == ".outputs":
-            sg["outputs"] = words[1:]
-
-        elif words[0] == ".marking":
-            sg["initial_state"] = words[1][1:-1]
-
-        elif words[0][0] == ".":
-            raise Exception("Unsupported statement: %s" % words[0])
-
-        else:
-            sg["transitions"].append(words)
+        if words[0] in ignored: pass
+        elif words[0] == ".model": sg["model"] = words[1]
+        elif words[0] == ".inputs": sg["inputs"] = words[1:]
+        elif words[0] == ".outputs": sg["outputs"] = words[1:]
+        elif words[0] == ".marking": sg["initial_state"] = words[1][1:-1]
+        elif words[0][0] == ".": raise Exception("Unsupported statement: %s" % words[0])
+        else: sg["transitions"].append(words)
 
     return sg