Return local filename after download

yannforget · Nov 27, 2018 · eeeb7ac · eeeb7ac
1 parent 013eeef
commit eeeb7ac
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/landsatxplore/earthexplorer.py b/landsatxplore/earthexplorer.py
@@ -88,6 +88,7 @@ def _download(self, url, output_dir, file_size, chunk_size=1024):
                         if chunk:
                             f.write(chunk)
                             pbar.update(chunk_size)
+        return local_filename
 
     def download(self, scene_id, output_dir):
         """Download a Landsat scene given its identifier and an output
@@ -97,4 +98,5 @@ def download(self, scene_id, output_dir):
         if is_product_id(scene_id):
             scene_id = self.api.lookup(dataset, [scene_id], inverse=True)[0]
         url = EE_DOWNLOAD_URL.format(folder=EE_FOLDER[dataset], sid=scene_id)
-        self._download(url, output_dir, file_size=SIZES[dataset])
+        filename = self._download(url, output_dir, file_size=SIZES[dataset])
+        return filename
diff --git a/setup.py b/setup.py
@@ -10,7 +10,7 @@
 
 setup(
     name='landsatxplore',
-    version='0.0.4',
+    version='0.0.5',
     description='Search and download Landsat scenes from EarthExplorer.',
     long_description=long_description,
     long_description_content_type='text/markdown',