fix: remove Python 2 compat in qltui

qilingframework · Feb 9, 2024 · 7e7655e · 7e7655e
1 parent 89123fd
commit 7e7655e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/qltool b/qltool
@@ -186,7 +186,7 @@ def handle_examples(parser: argparse.ArgumentParser):
     parser.exit(0, __ql_examples)
 
 
-if __name__ == '__main__':
+def run():
     parser = argparse.ArgumentParser()
     parser.add_argument('--version', action='version', version=f'qltool for Qiling {ql_ver}, using Unicorn {uc_ver}')
 
@@ -319,3 +319,6 @@ if __name__ == '__main__':
             pprint(rep)
 
     exit(ql.os.exit_code)
+
+if __name__ == '__main__':
+    run()
diff --git a/qltui.py b/qltui.py
@@ -2,7 +2,6 @@
 import ast
 import pickle
 import re
-import six
 import argparse
 import json
 
@@ -131,13 +130,13 @@ def log(string, color):
         """
         Function to beautify terminal output
         """
-        six.print_(colored(string, color))
+        print(colored(string, color))
 else:
     def log(string, color):
         """
         Function to beautify terminal output
         """
-        six.print_(string)
+        print(string)
 
 
 class IntValidator(Validator):