fixing bug with download stream --> stream_to

Author: vsoch

examples/singularity_hub/compare_builds.py

@@ -12,6 +12,8 @@
 
 shub = Client()    # Singularity Hub Client
 
+container_name = 'vsoch/singularity-hello-world'
+
 # Let's keep images in a temporary folder
 base = "/home/vanessa/Documents/Work/singularity/hub"
 storage = "%s/containers" %base

singularity/hub/base.py

@@ -62,7 +62,6 @@ def download_image(manifest,download_folder=None,extract=True,name=None):
     print("Found image %s:%s" %(manifest['name'],manifest['branch']))
     print("Downloading image... %s" %(image_file))
 
-    #TODO: add temporary file here
     if download_folder is not None:
         image_file = "%s/%s" %(download_folder,image_file)
     url = manifest['image']

singularity/hub/utils.py

@@ -10,6 +10,7 @@
 
 import requests
 import os
+import tempfile
 import sys
 
 
@@ -184,7 +185,7 @@ def download_atomically(url,file_name,headers=None):
     try:               # file_name.tmp.XXXXXX
         fd, tmp_file = tempfile.mkstemp(prefix=("%s.tmp." % file_name)) 
         os.close(fd)
-        response = api_get(url,headers=headers,stream=tmp_file)
+        response = api_get(url,headers=headers,stream_to=tmp_file)
         if isinstance(response, HTTPError):
             bot.logger.error("Error downloading %s, exiting.", url)
             sys.exit(1)