From 49f9e44966566bfe8608351363434cb343f1df84 Mon Sep 17 00:00:00 2001
From: Martin Braun <martin.braun@ettus.com>
Date: Mon, 29 Oct 2018 17:06:41 -0700
Subject: utils: uhd_images_downloader: Remove archive type

Since the uhd_images_downloader is so tightly integrated into our
infrastructure, we've dropped support for .tar.gz and .tar.xz file for
this particular purpose. Note that UHD releases still receive images in
all three formats.
---
 host/utils/uhd_images_downloader.py.in | 50 +++++++++++++++-------------------
 1 file changed, 22 insertions(+), 28 deletions(-)

(limited to 'host')

diff --git a/host/utils/uhd_images_downloader.py.in b/host/utils/uhd_images_downloader.py.in
index ff75a7b85..0c19e9167 100644
--- a/host/utils/uhd_images_downloader.py.in
+++ b/host/utils/uhd_images_downloader.py.in
@@ -38,7 +38,6 @@ _INVENTORY_FILENAME       = "inventory.json"
 _CONTACT                  = "support@ettus.com"
 _DEFAULT_BUFFER_SIZE      = 8192
 _DEFAULT_DOWNLOAD_LIMIT   = 100 * 1024 * 1024 # Bytes
-_ARCHIVE_ALGS             = ["zip", "targz", "tarxz"]
 _ARCHIVE_DEFAULT_TYPE     = "zip"
 _UHD_VERSION              = "@UHD_VERSION@"
 # Note: _MANIFEST_CONTENTS are placed at the bottom of this file for aesthetic reasons
@@ -87,9 +86,6 @@ def parse_args():
                              "interactively, or by providing --yes.")
     parser.add_argument("-b", "--base-url", type=str, default=_DEFAULT_BASE_URL,
                         help="Set base URL for images download location")
-    parser.add_argument("-z", "--archive-type", type=str, default=_ARCHIVE_DEFAULT_TYPE,
-                        help=("Select archiving function (options: {})"
-                              .format(",".join(_ARCHIVE_ALGS))))
     parser.add_argument("-k", "--keep", action="store_true", default=False,
                         help="Keep the downloaded images archives in the image directory")
     parser.add_argument("-T", "--test", action="store_true", default=False,
@@ -112,10 +108,6 @@ def parse_args():
     if args.yes:
         global _YES
         _YES = True
-    archive_type = args.archive_type
-    if archive_type not in _ARCHIVE_ALGS:
-        log("ERROR", "Selected archive type not supported: {}".format(archive_type))
-        return 1
     # Set the verbosity
     global _LOG_LEVEL
     log("TRACE", "Default log level: {}".format(_LOG_LEVEL))
@@ -383,28 +375,31 @@ def delete_from_inv(target_info, inventory, images_dir):
     return True
 
 
-def extract(archive_path, images_dir, archive_type, test_zip=False):
-    """Extract the contents of the archive into `images_dir`"""
-    if archive_type == "zip":
-        log("TRACE", "Attempting to extracted files from {}".format(archive_path))
-        with zipfile.ZipFile(archive_path) as images_zip:
-            # Check that the Zip file is valid, in which case `testzip()` returns None.
-            # If its bad, that function will return a list of bad files
-            try:
-                if test_zip and images_zip.testzip():
-                    log("ERROR", "Could not extract the following invalid Zip file:"
-                                 " {}".format(archive_path))
-                    return []
-            except OSError:
+def extract(archive_path, images_dir, test_zip=False):
+    """
+    Extract the contents of `archive_path` into `images_dir`
+
+    Returns a list of files that were extracted.
+
+    If test_zip is set, it will verify the zip file before extracting.
+    """
+    log("TRACE", "Attempting to extracted files from {}".format(archive_path))
+    with zipfile.ZipFile(archive_path) as images_zip:
+        # Check that the Zip file is valid, in which case `testzip()` returns
+        # None.  If it's bad, that function will return a list of bad files
+        try:
+            if test_zip and images_zip.testzip():
                 log("ERROR", "Could not extract the following invalid Zip file:"
                              " {}".format(archive_path))
                 return []
-            images_zip.extractall(images_dir)
-            archive_namelist = images_zip.namelist()
-            log("TRACE", "Extracted files: {}".format(archive_namelist))
-            return archive_namelist
-    else:
-        raise NotImplementedError("Archive type {} not implemented".format(archive_type))
+        except OSError:
+            log("ERROR", "Could not extract the following invalid Zip file:"
+                         " {}".format(archive_path))
+            return []
+        images_zip.extractall(images_dir)
+        archive_namelist = images_zip.namelist()
+        log("TRACE", "Extracted files: {}".format(archive_namelist))
+        return archive_namelist
 
 
 def update_target(target_info, temp_dir, images_dir, inventory, args):
@@ -445,7 +440,6 @@ def update_target(target_info, temp_dir, images_dir, inventory, args):
         archive_namelist = extract(
             temp_path,
             images_dir,
-            'zip',
             args.test)
         if args.keep:
             # If the user wants to keep the downloaded archive,
-- 
cgit v1.2.3