Merge pull request #269 from planetlabs/skip_quads_with_no_download_access

Skip mosaic quads that a user does not have access to during downloading

Author: sarasafavi

planet/api/client.py

@@ -16,7 +16,7 @@
 import json
 from .dispatch import RequestsDispatcher
 from . import auth
-from .exceptions import (InvalidIdentity, APIException)
+from .exceptions import (InvalidIdentity, APIException, NoPermission)
 from . import models
 from . import filters
 
@@ -357,14 +357,18 @@ def download_quad(self, quad, callback=None):
         be invoked asynchronously.  Otherwise it is up to the caller to handle
         the response Body.
 
-        :param asset dict: A mosaic quad representation from the API
+        :param quad dict: A mosaic quad representation from the API
         :param callback: An optional function to aysnchronsously handle the
                          download. See :py:func:`planet.api.write_to_file`
         :returns: :py:Class:`planet.api.models.Response` containing a
                   :py:Class:`planet.api.models.Body` of the asset.
         :raises planet.api.exceptions.APIException: On API error.
         '''
-        download_url = quad['_links']['download']
+        try:
+            download_url = quad['_links']['download']
+        except KeyError:
+            msg = 'You do not have download permissions for quad {}'
+            raise NoPermission(msg.format(quad['id']))
         return self._get(download_url, models.Body, callback=callback)
 
     def check_analytics_connection(self):

planet/api/downloader.py

@@ -16,7 +16,7 @@
 import threading
 import time
 from .utils import write_to_file
-from planet.api.exceptions import RequestCancelled
+from planet.api.exceptions import (RequestCancelled, NoPermission)
 try:
     import Queue as queue
 except ImportError:
@@ -466,9 +466,12 @@ def _task(self, t):
     def _do(self, task):
         func = self._write_tracker(task, None)
         writer = write_to_file(self._dest, func, overwrite=False)
-        self._downloads += 1
-        self._results.put((task, {'type': 'quad'},
-                           self._client.download_quad(task, writer)))
+        try:
+            self._results.put((task, {'type': 'quad'},
+                               self._client.download_quad(task, writer)))
+            self._downloads += 1
+        except NoPermission:
+            _info('No download permisson for %s, skipping', task['id'])
 
 
 class _MosaicDownloader(_Downloader):