get rid of Enum dependency and import sys only if run as script

Signed-off-by: Frantisek Hrbata <frantisek@hrbata.com>

diff --git a/pbaq.py b/pbaq.py
index 859789f55704d6ee52ef9f69edc0586ea5231cfd..ca8e3034e86668f92fd28ff45adca091967e5e5d 100755 (executable)
--- a/pbaq.py
+++ b/pbaq.py
@@ -81,10 +81,7 @@ $ops  - bugzilla operator
 $str  - string, may be quoted if necessary
 '''
 
 $str  - string, may be quoted if necessary
 '''
 

-import sys
-from enum import Enum
-
-class TokenType(Enum):
+class TokenType:

     UNKNOWN  = 0
     AND      = 1
     FAND     = 2
     UNKNOWN  = 0
     AND      = 1
     FAND     = 2


@@ -550,6 +547,8 @@ class Parser:
         return dst
 
 if __name__ == "__main__":
         return dst
 
 if __name__ == "__main__":

+    import sys
+

     if len(sys.argv) != 2:
         print("usage: pbqa.py 'expression'")
         sys.exit(1)
     if len(sys.argv) != 2:
         print("usage: pbqa.py 'expression'")
         sys.exit(1)