[nnpackage] nnpackager -c will create compressed nnpackage (#6278)

Use "nnpackage -c path_to_nnpackage_root" to get compressed nnpackage.
Also it removed redundant parenthesis in if condition.

Signed-off-by: Sanggyu Lee <sg5.lee@samsung.com>

diff --git a/tools/nnpackager/nnpackager.py b/tools/nnpackager/nnpackager.py
index 5d835ac..8e2805a 100755 (executable)
--- a/tools/nnpackager/nnpackager.py
+++ b/tools/nnpackager/nnpackager.py
@@ -5,6 +5,7 @@ import os
 from collections import OrderedDict
 import sys
 import argparse
+import shutil
 
 
 def verify(path):
@@ -38,11 +39,27 @@ def verify(path):
             print("models attribute does not exist.")
 
 
+def compress(path):
+    nnpackage_name = os.path.basename(os.path.normpath(path))
+    shutil.make_archive(nnpackage_name, 'zip', path)
+    print("nnpackage compression is done")
+
+
 if __name__ == '__main__':
     parser = argparse.ArgumentParser()
     parser.add_argument('path', type=str, help='the path to nnpackage')
     parser.add_argument(
         '-v', '--verify', action='store_true', help="verify nnpackage (default: false)")
+    parser.add_argument(
+        '-c',
+        '--compress',
+        action='store_true',
+        help="compress nnpackage (default: false)")
+
     args = parser.parse_args()
-    if (args.verify):
+
+    if args.verify:
         verify(args.path)
+
+    if args.compress:
+        compress(args.path)