None check (#103)

* support limit object

* support limit object

* support limit object

* None check

Author: chkp-edenbr

import_export_package.py

@@ -26,7 +26,7 @@ def get_version(client):
     if sys.version_info < (3, 7):
         raise Exception("Min Python version required is 3.7")
 
-    arg_parser = argparse.ArgumentParser(description="R80.X Policy Package Export/Import Tool, V6.1.0")
+    arg_parser = argparse.ArgumentParser(description="R80.X Policy Package Export/Import Tool, V6.2.0")
     args = process_arguments(arg_parser)
     if args.force:
         args.unsafe_auto_accept = True

menu.py

@@ -21,10 +21,11 @@ def __init__(self, args):
         self.build()
 
     def display(self):
-        Menu.menu_print(self.title, 1)
-        for i in range(1, len(self.options)):
-            self.menu_print(str(i) + ". " + self.options[i - 1], 0)
+        if self.title:
+            Menu.menu_print(self.title, 1)
         if self.options:
+            for i in range(1, len(self.options)):
+                self.menu_print(str(i) + ". " + self.options[i - 1], 0)
             self.menu_print(str(len(self.options)) + ". " + self.options[len(self.options) - 1], 1)
             self.menu_print("99. " + self.last_option, 0)
         self.handle_input()