1 files changed, 16 insertions, 22 deletions
diff --git a/images/populate_images.py b/images/populate_images.py
index 20fcb7028..adf75ec8b 100755
--- a/images/populate_images.py
+++ b/images/populate_images.py
@@ -1,47 +1,41 @@
 #!/usr/bin/env python
 #
 # Copyright 2014 Ettus Research LLC
+# Copyright 2018 Ettus Research, a National Instruments Company
 #
-# This program is free software: you can redistribute it and/or modify
-# it under the terms of the GNU General Public License as published by
-# the Free Software Foundation, either version 3 of the License, or
-# (at your option) any later version.
-#
-# This program is distributed in the hope that it will be useful,
-# but WITHOUT ANY WARRANTY; without even the implied warranty of
-# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
-# GNU General Public License for more details.
-#
-# You should have received a copy of the GNU General Public License
-# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+# SPDX-License-Identifier: GPL-3.0-or-later
 #
 """
 Populates the current directory with a valid set of binaries for the
 current commit.
 """
 
+from __future__ import print_function
 import os
 import subprocess
 import uhdimgs
 
-def main():
+def download_images(img_root_dir=None):
     " Go, go, go! "
     # Switch to correct dir
-    img_root_dir = os.path.join(uhdimgs.get_images_dir(), 'images')
+    img_root_dir = img_root_dir or os.path.join(uhdimgs.get_images_dir(), 'images')
+    if not os.path.isdir(img_root_dir):
+        print("== Creating images directory...")
+        os.mkdir(img_root_dir)
     os.chdir(uhdimgs.get_images_dir())
-    print "== Starting download..."
+    print("== Starting download...")
     try:
         downloader_cmd = [
-                'python',
-                '../host/utils/uhd_images_downloader.py.in',
-                '-i', img_root_dir,
-                '-m', 'manifest.txt'
+            'python',
+            '../host/utils/uhd_images_downloader.py.in',
+            '-i', img_root_dir,
+            '-m', 'manifest.txt'
         ]
         subprocess.check_call(downloader_cmd)
     except (subprocess.CalledProcessError, OSError):
-        print "[ERROR] Failed to run downloader script."
+        print("[ERROR] Failed to run downloader script.")
         exit(1)
-    print "== Done!"
+    print("== Done!")
 
 if __name__ == "__main__":
-    main()
+    download_images()