Source code for sentinelsat.sentinel

import hashlib
import logging
import re
import threading
import xml.etree.ElementTree as ET
from collections import OrderedDict, defaultdict, namedtuple
from copy import copy
from datetime import date, datetime, timedelta
from pathlib import Path
from typing import Dict
from urllib.parse import quote_plus, urljoin

import geojson
import geomet.wkt
import html2text
import requests
from import tqdm

from import DownloadStatus, Downloader
from sentinelsat.exceptions import (
from . import __version__ as sentinelsat_version

[docs]class SentinelAPI: """Class to connect to Copernicus Open Access Hub, search and download imagery. Parameters ---------- user : string username for DataHub set to None to use ~/.netrc password : string password for DataHub set to None to use ~/.netrc api_url : string, optional URL of the DataHub defaults to '' show_progressbars : bool Whether progressbars should be shown or not, e.g. during download. Defaults to True. timeout : float or tuple, default 60 How long to wait for DataHub response (in seconds). Tuple (connect, read) allowed. Set to None to wait indefinitely. Attributes ---------- session : requests.Session Session to connect to DataHub api_url : str URL to the DataHub page_size : int Number of results per query page. Current value: 100 (maximum allowed on ApiHub) timeout : float or tuple How long to wait for DataHub response (in seconds). """ logger = logging.getLogger("sentinelsat.SentinelAPI") def __init__( self, user, password, api_url="", show_progressbars=True, timeout=60, ): self.session = requests.Session() if user and password: self.session.auth = (user, password) self.api_url = api_url if api_url.endswith("/") else api_url + "/" self.page_size = 100 self.user_agent = "sentinelsat/" + sentinelsat_version self.session.headers["User-Agent"] = self.user_agent self.session.timeout = timeout self.show_progressbars = show_progressbars self._dhus_version = None # For unit tests self._last_query = None self._last_response = None self._online_attribute_used = True self._concurrent_dl_limit = 4 self._concurrent_lta_trigger_limit = 10 # The number of allowed concurrent GET requests is limited on the server side. # We use a bounded semaphore to ensure we stay within that limit. # Notably, LTA trigger requests also count against that limit. self._dl_limit_semaphore = threading.BoundedSemaphore(self._concurrent_dl_limit) self._lta_limit_semaphore = threading.BoundedSemaphore(self._concurrent_lta_trigger_limit) self.downloader = Downloader(self) @property def concurrent_dl_limit(self): """int: Maximum number of concurrent downloads allowed by the server.""" return self._concurrent_dl_limit @concurrent_dl_limit.setter def concurrent_dl_limit(self, value): self._concurrent_dl_limit = value self._lta_limit_semaphore = threading.BoundedSemaphore(self._concurrent_dl_limit) @property def concurrent_lta_trigger_limit(self): """int: Maximum number of concurrent Long Term Archive retrievals allowed.""" return self._concurrent_lta_trigger_limit @concurrent_lta_trigger_limit.setter def concurrent_lta_trigger_limit(self, value): self._concurrent_lta_trigger_limit = value self._dl_limit_semaphore = threading.BoundedSemaphore(self._concurrent_lta_trigger_limit) @property def dl_retry_delay(self): """float, default 10: Number of seconds to wait between retrying of failed downloads.""" return self.downloader.dl_retry_delay @dl_retry_delay.setter def dl_retry_delay(self, value): self.downloader.dl_retry_delay = value @property def lta_retry_delay(self): """float, default 60: Number of seconds to wait between requests to the Long Term Archive.""" return self.downloader.lta_retry_delay @lta_retry_delay.setter def lta_retry_delay(self, value): self.downloader.lta_retry_delay = value @property def lta_timeout(self): """float, optional: Maximum number of seconds to wait for triggered products to come online. Defaults to no timeout.""" return self.downloader.lta_timeout @lta_timeout.setter def lta_timeout(self, value): self.downloader.lta_timeout = value @property def dl_limit_semaphore(self): return self._dl_limit_semaphore @property def lta_limit_semaphore(self): return self._lta_limit_semaphore @staticmethod def _api2dhus_url(api_url): url = re.sub("apihub/$", "dhus/", api_url) url = re.sub("", "", url) return url def _req_dhus_stub(self): try: with self.dl_limit_semaphore: resp = self.session.get(self.api_url + "api/stub/version") resp.raise_for_status() except requests.exceptions.HTTPError as err: self.logger.error("HTTPError: %s", err) self.logger.error("Are you trying to get the DHuS version of APIHub?") self.logger.error("Trying again after conversion to DHuS URL") with self.dl_limit_semaphore: resp = self.session.get(self._api2dhus_url(self.api_url) + "api/stub/version") resp.raise_for_status() return resp.json()["value"] @property def dhus_version(self): if self._dhus_version is None: self._dhus_version = self._req_dhus_stub() return self._dhus_version
[docs] def query( self, area=None, date=None, raw=None, area_relation="Intersects", order_by=None, limit=None, offset=0, **keywords ): """Query the OpenSearch API with the coordinates of an area, a date interval and any other search keywords accepted by the API. Parameters ---------- area : str, optional The area of interest formatted as a Well-Known Text string. date : tuple of (str or datetime) or str, optional A time interval filter based on the Sensing Start Time of the products. Expects a tuple of (start, end), e.g. ("NOW-1DAY", "NOW"). The timestamps can be either a Python datetime or a string in one of the following formats: - yyyyMMdd - yyyy-MM-ddThh:mm:ss.SSSZ (ISO-8601) - yyyy-MM-ddThh:mm:ssZ - NOW - NOW-<n>DAY(S) (or HOUR(S), MONTH(S), etc.) - NOW+<n>DAY(S) - yyyy-MM-ddThh:mm:ssZ-<n>DAY(S) - NOW/DAY (or HOUR, MONTH etc.) - rounds the value to the given unit Alternatively, an already fully formatted string such as "[NOW-1DAY TO NOW]" can be used as well. raw : str, optional Additional query text that will be appended to the query. area_relation : {'Intersects', 'Contains', 'IsWithin'}, optional What relation to use for testing the AOI. Case insensitive. - Intersects: true if the AOI and the footprint intersect (default) - Contains: true if the AOI is inside the footprint - IsWithin: true if the footprint is inside the AOI order_by: str, optional A comma-separated list of fields to order by (on server side). Prefix the field name by '+' or '-' to sort in ascending or descending order, respectively. Ascending order is used if prefix is omitted. Example: "cloudcoverpercentage, -beginposition". limit: int, optional Maximum number of products returned. Defaults to no limit. offset: int, optional The number of results to skip. Defaults to 0. **keywords Additional keywords can be used to specify other query parameters, e.g. `relativeorbitnumber=70`. See for a full list. Range values can be passed as two-element tuples, e.g. `cloudcoverpercentage=(0, 30)`. `None` can be used in range values for one-sided ranges, e.g. `orbitnumber=(16302, None)`. Ranges with no bounds (`orbitnumber=(None, None)`) will not be included in the query. Multiple values for the same query parameter can be provided as sets and will be handled as logical OR, e.g. `orbitnumber={16302, 1206}`. The time interval formats accepted by the `date` parameter can also be used with any other parameters that expect time intervals (that is: 'beginposition', 'endposition', 'date', 'creationdate', and 'ingestiondate'). Returns ------- dict[string, dict] Products returned by the query as a dictionary with the product ID as the key and the product's attributes (a dictionary) as the value. """ query = self.format_query(area, date, raw, area_relation, **keywords) if query.strip() == "": # An empty query should return the full set of products on the server, which is a bit unreasonable. # The server actually raises an error instead and it's better to fail early in the client. raise ValueError("Empty query.") # check query length - often caused by complex polygons if self.check_query_length(query) > 1.0: self.logger.warning( "The query string is too long and will likely cause a bad DHuS response." ) self.logger.debug( "Running query: order_by=%s, limit=%s, offset=%s, query=%s", order_by, limit, offset, query, ) formatted_order_by = _format_order_by(order_by) response, count = self._load_query(query, formatted_order_by, limit, offset)"Found {count:,} products") return _parse_opensearch_response(response)
[docs] @staticmethod def format_query(area=None, date=None, raw=None, area_relation="Intersects", **keywords): """Create a OpenSearch API query string.""" if area_relation.lower() not in {"intersects", "contains", "iswithin"}: raise ValueError("Incorrect AOI relation provided ({})".format(area_relation)) # Check for duplicate keywords kw_lower = {x.lower() for x in keywords} if ( len(kw_lower) != len(keywords) or (date is not None and "beginposition" in kw_lower) or (area is not None and "footprint" in kw_lower) ): raise ValueError( "Query contains duplicate keywords. Note that query keywords are case-insensitive." ) query_parts = [] if date is not None: keywords["beginPosition"] = date for attr, value in sorted(keywords.items()): if isinstance(value, set): if len(value) == 0: continue sub_parts = [] for sub_value in value: sub_value = _format_query_value(attr, sub_value) if sub_value is not None: sub_parts.append(f"{attr}:{sub_value}") sub_parts = sorted(sub_parts) query_parts.append("({})".format(" OR ".join(sub_parts))) else: value = _format_query_value(attr, value) if value is not None: query_parts.append(f"{attr}:{value}") if raw: query_parts.append(raw) if area is not None: query_parts.append('footprint:"{}({})"'.format(area_relation, area)) return " ".join(query_parts)
[docs] def count(self, area=None, date=None, raw=None, area_relation="Intersects", **keywords): """Get the number of products matching a query. Accepted parameters are identical to :meth:`SentinelAPI.query()`. This is a significantly more efficient alternative to doing `len(api.query())`, which can take minutes to run for queries matching thousands of products. Returns ------- int The number of products matching a query. """ for kw in ["order_by", "limit", "offset"]: # Allow these function arguments to be included for compatibility with query(), # but ignore them. if kw in keywords: del keywords[kw] query = self.format_query(area, date, raw, area_relation, **keywords) _, total_count = self._load_query(query, limit=0) return total_count
def _load_query(self, query, order_by=None, limit=None, offset=0): products, count = self._load_subquery(query, order_by, limit, offset) # repeat query until all results have been loaded max_offset = count if limit is not None: max_offset = min(count, offset + limit) if max_offset > offset + self.page_size: progress = self._tqdm( desc="Querying products", initial=self.page_size, total=max_offset - offset, unit="product", ) for new_offset in range(offset + self.page_size, max_offset, self.page_size): new_limit = limit if limit is not None: new_limit = limit - new_offset + offset ret = self._load_subquery(query, order_by, new_limit, new_offset)[0] progress.update(len(ret)) products += ret progress.close() return products, count def _load_subquery(self, query, order_by=None, limit=None, offset=0): # store last query (for testing) self._last_query = query self.logger.debug("Sub-query: offset=%s, limit=%s", offset, limit) # load query results url = self._format_url(order_by, limit, offset) # Unlike POST, DHuS only accepts latin1 charset in the GET params with self.dl_limit_semaphore: response = self.session.get(url, params={"q": query.encode("latin1")}) self._check_scihub_response(response, query_string=query) # store last status code (for testing) self._last_response = response # parse response content try: json_feed = response.json()["feed"] if "error" in json_feed: message = json_feed["error"]["message"] message = message.replace(" ", "") raise QuerySyntaxError(message, response) total_results = int(json_feed["opensearch:totalResults"]) except (ValueError, KeyError): raise ServerError("API response not valid. JSON decoding failed.", response) products = json_feed.get("entry", []) # this verification is necessary because if the query returns only # one product, self.products will be a dict not a list if isinstance(products, dict): products = [products] return products, total_results def _format_url(self, order_by=None, limit=None, offset=0): if limit is None: limit = self.page_size limit = min(limit, self.page_size) url = "search?format=json&rows={}".format(limit) url += "&start={}".format(offset) if order_by: url += "&orderby={}".format(order_by) return urljoin(self.api_url, url)
[docs] @staticmethod def to_geojson(products): """Return the products from a query response as a GeoJSON with the values in their appropriate Python types. """ feature_list = [] for i, (product_id, props) in enumerate(products.items()): props = props.copy() props["id"] = product_id poly = geomet.wkt.loads(props["footprint"]) del props["footprint"] del props["gmlfootprint"] # Fix "'datetime' is not JSON serializable" for k, v in props.items(): if isinstance(v, (date, datetime)): props[k] = v.strftime("%Y-%m-%dT%H:%M:%S.%fZ") feature_list.append(geojson.Feature(geometry=poly, id=i, properties=props)) return geojson.FeatureCollection(feature_list)
[docs] @staticmethod def to_dataframe(products): """Return the products from a query response as a Pandas DataFrame with the values in their appropriate Python types. """ try: import pandas as pd except ImportError: raise ImportError("to_dataframe requires the optional dependency Pandas.") return pd.DataFrame.from_dict(products, orient="index")
[docs] @staticmethod def to_geodataframe(products): """Return the products from a query response as a GeoPandas GeoDataFrame with the values in their appropriate Python types. """ try: import geopandas as gpd import shapely.wkt except ImportError: raise ImportError( "to_geodataframe requires the optional dependencies GeoPandas and Shapely." ) crs = "EPSG:4326" # WGS84 if len(products) == 0: return gpd.GeoDataFrame(crs=crs, geometry=[]) df = SentinelAPI.to_dataframe(products) geometry = [shapely.wkt.loads(fp) for fp in df["footprint"]] # remove useless columns df.drop(["footprint", "gmlfootprint"], axis=1, inplace=True) return gpd.GeoDataFrame(df, crs=crs, geometry=geometry)
[docs] def get_product_odata(self, id, full=False): """Access OData API to get info about a product. Returns a dict containing the id, title, size, md5sum, date, footprint and download url of the product. The date field corresponds to the Start ContentDate value. If `full` is set to True, then the full, detailed metadata of the product is returned in addition to the above. Parameters ---------- id : string The UUID of the product to query full : bool Whether to get the full metadata for the Product. False by default. Returns ------- dict[str, Any] A dictionary with an item for each metadata attribute Notes ----- For a full list of mappings between the OpenSearch (Solr) and OData attribute names see the following definition files: """ url = self._get_odata_url(id, "?$format=json") if full: url += "&$expand=Attributes" with self.dl_limit_semaphore: response = self.session.get(url) self._check_scihub_response(response) values = _parse_odata_response(response.json()["d"]) if values["title"].startswith("S3"): values["manifest_name"] = "xfdumanifest.xml" values["product_root_dir"] = values["title"] + ".SEN3" else: values["manifest_name"] = "" values["product_root_dir"] = values["title"] + ".SAFE" values["quicklook_url"] = self._get_odata_url(id, "/Products('Quicklook')/$value") return values
[docs] def is_online(self, id): """Returns whether a product is online Parameters ---------- id : string UUID of the product, e.g. 'a8dd0cfd-613e-45ce-868c-d79177b916ed' Returns ------- bool True if online, False if in LTA See Also -------- :meth:`SentinelAPI.trigger_offline_retrieval()` """ # Check for more information if not self._online_attribute_used: return True url = self._get_odata_url(id, "/Online/$value") with self.dl_limit_semaphore: r = self.session.get(url) try: self._check_scihub_response(r) except ServerError as e: # Handle DHuS versions that do not set the Online attribute if "Could not find property with name: 'Online'" in e.msg: self._online_attribute_used = False return True raise return r.json()
[docs] def download(self, id, directory_path=".", checksum=True, nodefilter=None): """Download a product. Uses the filename on the server for the downloaded file, e.g. "". Incomplete downloads are continued and complete files are skipped. Parameters ---------- id : string UUID of the product, e.g. 'a8dd0cfd-613e-45ce-868c-d79177b916ed' directory_path : string, optional Where the file will be downloaded checksum : bool, default True If True, verify the downloaded file's integrity by checking its checksum. Throws InvalidChecksumError if the checksum does not match. nodefilter : callable, optional The callable is used to select which files of each product will be downloaded. If None (the default), the full products will be downloaded. See :mod:`sentinelsat.products` for sample node filters. Returns ------- product_info : dict Dictionary containing the product's info from get_product_odata() as well as the path on disk. Raises ------ InvalidChecksumError If the MD5 checksum does not match the checksum on the server. LTATriggered If the product has been archived and its retrieval was successfully triggered. LTAError If the product has been archived and its retrieval failed. """ downloader = copy(self.downloader) downloader.node_filter = nodefilter downloader.verify_checksum = checksum return, directory_path)
def _get_filename(self, product_info): if product_info["Online"]: with self.dl_limit_semaphore: req = self.session.head(product_info["url"]) self._check_scihub_response(req, test_json=False) cd = req.headers.get("Content-Disposition") if cd is not None: filename = cd.split("=", 1)[1].strip('"') return filename with self.dl_limit_semaphore: req = self.session.get( product_info["url"].replace("$value", "Attributes('Filename')/Value/$value") ) self._check_scihub_response(req, test_json=False) filename = req.text # This should cover all currently existing file types: .SAFE, .SEN3, .nc and .EOF filename = filename.replace(".SAFE", ".zip") filename = filename.replace(".SEN3", ".zip") return filename
[docs] def trigger_offline_retrieval(self, uuid): """Triggers retrieval of an offline product from the Long Term Archive. Parameters ---------- uuid : string UUID of the product Returns ------- bool True, if the product retrieval was successfully triggered. False, if the product is already online. Raises ------ LTAError If the request was not accepted due to exceeded user quota or server overload. ServerError If an unexpected response was received from server. UnauthorizedError If the provided credentials were invalid. Notes ----- """ return self.downloader.trigger_offline_retrieval(uuid)
[docs] def download_all( self, products, directory_path=".", max_attempts=10, checksum=True, n_concurrent_dl=None, lta_retry_delay=None, fail_fast=False, nodefilter=None, ): """Download a list of products. Takes a list of product IDs as input. This means that the return value of query() can be passed directly to this method. File names on the server are used for the downloaded files, e.g. "". In case of interruptions or other exceptions, downloading will restart from where it left off. Downloading is attempted at most max_attempts times to avoid getting stuck with unrecoverable errors. Parameters ---------- products : list List of product IDs directory_path : string Directory where the downloaded files will be downloaded max_attempts : int, default 10 Number of allowed retries before giving up downloading a product. checksum : bool, default True If True, verify the downloaded files' integrity by checking its MD5 checksum. Throws InvalidChecksumError if the checksum does not match. Defaults to True. n_concurrent_dl : integer, optional Number of concurrent downloads. Defaults to :attr:`SentinelAPI.concurrent_dl_limit`. lta_retry_delay : float, default 60 Number of seconds to wait between requests to the Long Term Archive. fail_fast : bool, default False if True, all other downloads are cancelled when one of the downloads fails. nodefilter : callable, optional The callable is used to select which files of each product will be downloaded. If None (the default), the full products will be downloaded. See :mod:`sentinelsat.products` for sample node filters. Notes ----- By default, raises the most recent downloading exception if all downloads failed. If ``fail_fast`` is set to True, raises the encountered exception on the first failed download instead. Returns ------- dict[string, dict] A dictionary containing the return value from download() for each successfully downloaded product. dict[string, dict] A dictionary containing the product information for products successfully triggered for retrieval from the long term archive but not downloaded. dict[string, dict] A dictionary containing the product information of products where either downloading or triggering failed. "exception" field with the exception info is included to the product info dict. """ downloader = copy(self.downloader) downloader.verify_checksum = checksum downloader.fail_fast = fail_fast downloader.max_attempts = max_attempts if n_concurrent_dl: downloader.n_concurrent_dl = n_concurrent_dl if lta_retry_delay: downloader.lta_retry_delay = lta_retry_delay downloader.node_filter = nodefilter statuses, exceptions, product_infos = downloader.download_all(products, directory_path) # Adapt results to the old download_all() API downloaded_prods = {} retrieval_triggered = {} failed_prods = {} for pid, status in statuses.items(): if pid not in product_infos: product_infos[pid] = {} if pid in exceptions: product_infos[pid]["exception"] = exceptions[pid] if status == DownloadStatus.DOWNLOADED: downloaded_prods[pid] = product_infos[pid] elif status == DownloadStatus.TRIGGERED: retrieval_triggered[pid] = product_infos[pid] else: failed_prods[pid] = product_infos[pid] ResultTuple = namedtuple("ResultTuple", ["downloaded", "retrieval_triggered", "failed"]) return ResultTuple(downloaded_prods, retrieval_triggered, failed_prods)
[docs] def download_all_quicklooks(self, products, directory_path=".", n_concurrent_dl=4): """Download quicklook for a list of products. Takes a dict of product IDs: product data as input. This means that the return value of query() can be passed directly to this method. File names on the server are used for the downloaded images, e.g. "S2A_MSIL1C_20200924T104031_N0209_R008_T35WMV_20200926T135405.jpeg". Parameters ---------- products : dict Dict of product IDs, product data directory_path : string Directory where the downloaded images will be downloaded n_concurrent_dl : integer Number of concurrent downloads Returns ------- dict[string, dict] A dictionary containing the return value from download_quicklook() for each successfully downloaded quicklook dict[string, dict] A dictionary containing the error of products where either quicklook was not available or it had an unexpected content type """ downloader = copy(self.downloader) downloader.n_concurrent_dl = n_concurrent_dl return downloader.download_all_quicklooks(products, directory_path)
[docs] def download_quicklook(self, id, directory_path="."): """Download a quicklook for a product. Uses the filename on the server for the downloaded image name, e.g. "S1A_EW_GRDH_1SDH_20141003T003840_20141003T003920_002658_002F54_4DD1.jpeg". Already downloaded images are skipped. Parameters ---------- id : string UUID of the product, e.g. 'a8dd0cfd-613e-45ce-868c-d79177b916ed' directory_path : string, optional Where the image will be downloaded Returns ------- quicklook_info : dict Dictionary containing the quicklooks's response headers as well as the path on disk. """ return self.downloader.download_quicklook(id, directory_path)
[docs] @staticmethod def get_products_size(products): """Return the total file size in GB of all products in the OpenSearch response.""" size_total = 0 for title, props in products.items(): size_product = props["size"] size_value = float(size_product.split(" ")[0]) size_unit = str(size_product.split(" ")[1]) if size_unit == "MB": size_value /= 1024.0 if size_unit == "KB": size_value /= 1024.0 * 1024.0 size_total += size_value return round(size_total, 2)
[docs] @staticmethod def check_query_length(query): """Determine whether a query to the OpenSearch API is too long. The length of a query string is limited to approximately 3898 characters but any special characters (that is, not alphanumeric or -_.*) will take up more space. Parameters ---------- query : str The query string Returns ------- float Ratio of the query length to the maximum length """ # The server uses the Java's URLEncoder implementation internally, which we are replicating here effective_length = len(quote_plus(query, safe="-_.*").replace("~", "%7E")) return effective_length / 3898
def _query_names(self, names): """Find products by their names, e.g. S1A_EW_GRDH_1SDH_20141003T003840_20141003T003920_002658_002F54_4DD1. Note that duplicates exist on server, so multiple products can be returned for each name. Parameters ---------- names : list[string] List of product names. Returns ------- dict[string, dict[str, dict]] A dictionary mapping each name to a dictionary which contains the products with that name (with ID as the key). """ products = {} for name in names: products.update(self.query(identifier=name)) # Group the products output = OrderedDict((name, dict()) for name in names) for id, metadata in products.items(): name = metadata["identifier"] output[name][id] = metadata return output
[docs] def check_files(self, paths=None, ids=None, directory=None, delete=False): """Verify the integrity of product files on disk. Integrity is checked by comparing the size and checksum of the file with the respective values on the server. The input can be a list of products to check or a list of IDs and a directory. In cases where multiple products with different IDs exist on the server for given product name, the file is considered to be correct if any of them matches the file size and checksum. A warning is logged in such situations. The corrupt products' OData info is included in the return value to make it easier to re-download the products, if necessary. Parameters ---------- paths : list[string] List of product file paths. ids : list[string] List of product IDs. directory : string Directory where the files are located, if checking based on product IDs. delete : bool Whether to delete corrupt products. Defaults to False. Returns ------- dict[str, list[dict]] A dictionary listing the invalid or missing files. The dictionary maps the corrupt file paths to a list of OData dictionaries of matching products on the server (as returned by :meth:`SentinelAPI.get_product_odata()`). """ if not ids and not paths: raise ValueError("Must provide either file paths or product IDs and a directory") if ids and not directory: raise ValueError("Directory value missing") if directory is not None: directory = Path(directory) paths = [Path(p) for p in paths] if paths else [] ids = ids or [] # Get product IDs corresponding to the files on disk names = [] if paths: names = [p.stem for p in paths] result = self._query_names(names) for product_dicts in result.values(): ids += list(product_dicts) names_from_paths = set(names) ids = set(ids) # Collect the OData information for each product # Product name -> list of matching odata dicts product_infos = defaultdict(list) for id in ids: odata = self.get_product_odata(id) name = odata["title"] product_infos[name].append(odata) # Collect if name not in names_from_paths: paths.append(directory / self._get_filename(odata)) # Now go over the list of products and check them corrupt = {} for path in paths: name = path.stem if len(product_infos[name]) > 1: self.logger.warning("%s matches multiple products on server", path) if not path.exists(): # We will consider missing files as corrupt also"%s does not exist on disk", path) corrupt[str(path)] = product_infos[name] continue is_fine = False for product_info in product_infos[name]: if path.stat().st_size == product_info["size"] and self._checksum_compare( path, product_info ): is_fine = True break if not is_fine:"%s is corrupt", path) corrupt[str(path)] = product_infos[name] if delete: path.unlink() return corrupt
def _checksum_compare(self, file_path, product_info, block_size=2**13): """Compare a given MD5 checksum with one calculated from a file.""" if "sha3-256" in product_info: checksum = product_info["sha3-256"] algo = hashlib.sha3_256() elif "md5" in product_info: checksum = product_info["md5"] algo = hashlib.md5() else: raise InvalidChecksumError("No checksum information found in product information.") file_path = Path(file_path) file_size = file_path.stat().st_size with self._tqdm( desc=f"{} checksumming", total=file_size, unit="B", unit_scale=True, leave=False, ) as progress: with open(file_path, "rb") as f: while True: block_data = if not block_data: break algo.update(block_data) progress.update(len(block_data)) return algo.hexdigest().lower() == checksum.lower() def _tqdm(self, **kwargs): """tqdm progressbar wrapper. May be overridden to customize progressbar behavior""" kwargs.update({"disable": not self.show_progressbars}) return tqdm(**kwargs)
[docs] def get_stream(self, id, **kwargs): """Exposes requests response ready to stream product to e.g. S3. Parameters ---------- id : string UUID of the product, e.g. 'a8dd0cfd-613e-45ce-868c-d79177b916ed' **kwargs Any additional parameters for :func:`requests.get()` Raises ------ LTATriggered If the product has been archived and its retrieval was successfully triggered. LTAError If the product has been archived and its retrieval failed. Returns ------- requests.Response: Opened response object """ return self.downloader.get_stream(id, **kwargs)
def _get_odata_url(self, uuid, suffix=""): return self.api_url + f"odata/v1/Products('{uuid}')" + suffix def _get_download_url(self, uuid): return self._get_odata_url(uuid, "/$value") @staticmethod def _check_scihub_response(response, test_json=True, query_string=None): """Check that the response from server has status code 2xx and that the response is valid JSON.""" # Prevent requests from needing to guess the encoding # SciHub appears to be using UTF-8 in all of their responses response.encoding = "utf-8" try: response.raise_for_status() if test_json: response.json() except (requests.HTTPError, ValueError): msg = None try: msg = response.json()["error"]["message"]["value"] except Exception: try: msg = response.headers["cause-message"] except Exception: if not response.text.lstrip().startswith("{"): try: h = html2text.HTML2Text() h.ignore_images = True h.ignore_anchors = True msg = h.handle(response.text).strip() except Exception: pass if msg is None: raise ServerError("Invalid API response", response) elif response.status_code == 401: msg = "Invalid user name or password" if "" in response.url: msg += ( ". Note that account creation and password changes may take up to a week " "to propagate to the '' API URL you are using. " "Consider switching to '' instead in the mean time." ) raise UnauthorizedError(msg, response) elif "Request Entity Too Large" in msg or "Request-URI Too Long" in msg: msg = "Server was unable to process the query due to its excessive length" if query_string is not None: length = SentinelAPI.check_query_length(query_string) msg += ( " (approximately {:.3}x times the maximum allowed). " "Consider using SentinelAPI.check_query_length() for " "client-side validation of the query string length.".format(length) ) raise QueryLengthError(msg, response) from None elif "Invalid key" in msg: msg = msg.split(" : ", 1)[-1] raise InvalidKeyError(msg, response) elif 500 <= response.status_code < 600 or msg: # 5xx: Server Error raise ServerError(msg, response) else: raise SentinelAPIError(msg, response) def _path_to_url(self, product_info, path, urltype=None): id = product_info["id"] root_dir = product_info["product_root_dir"] path = "/".join(["Nodes('{}')".format(item) for item in path.split("/")]) if urltype == "value": urltype = "/$value" elif urltype == "json": urltype = "?$format=json" elif urltype == "full": urltype = "?$format=json&$expand=Attributes" elif urltype is None: urltype = "" # else: pass urltype as is return self._get_odata_url(id, f"/Nodes('{root_dir}')/{path}{urltype}") def _get_manifest(self, product_info, path=None): path = Path(path) if path else None manifest_name = product_info["manifest_name"] url = self._path_to_url(product_info, manifest_name, "value") node_info = product_info.copy() node_info["url"] = url node_info["node_path"] = f"./{manifest_name}" del node_info["md5"] if path and path.exists(): self.logger.debug("Manifest file already available (%s), skipping download", path) data = path.read_bytes() node_info["size"] = len(data) return node_info, data url = self._path_to_url(product_info, manifest_name, "json") with self.dl_limit_semaphore: response = self.session.get(url) self._check_scihub_response(response) info = response.json()["d"] node_info["size"] = int(info["ContentLength"]) with self.dl_limit_semaphore: response = self.session.get(node_info["url"]) self._check_scihub_response(response, test_json=False) data = response.content if len(data) != node_info["size"]: raise SentinelAPIError("File corrupt: data length do not match") if path: path.parent.mkdir(parents=True, exist_ok=True) path.write_bytes(data) return node_info, data
[docs]def read_geojson(geojson_file): """Read a GeoJSON file into a GeoJSON object.""" with open(geojson_file) as f: return geojson.load(f)
[docs]def geojson_to_wkt(geojson_obj, decimals=4): """Convert a GeoJSON object to Well-Known Text. Intended for use with OpenSearch queries. 3D points are converted to 2D. Parameters ---------- geojson_obj : dict a GeoJSON object decimals : int, optional Number of decimal figures after point to round coordinate to. Defaults to 4 (about 10 meters). Returns ------- str Well-Known Text string representation of the geometry """ if "coordinates" in geojson_obj: geometry = geojson_obj elif "geometry" in geojson_obj: geometry = geojson_obj["geometry"] else: geometry = {"type": "GeometryCollection", "geometries": []} for feature in geojson_obj["features"]: geometry["geometries"].append(feature["geometry"]) def ensure_2d(geometry): if isinstance(geometry[0], (list, tuple)): return list(map(ensure_2d, geometry)) else: return geometry[:2] def check_bounds(geometry): if isinstance(geometry[0], (list, tuple)): return list(map(check_bounds, geometry)) else: if geometry[0] > 180 or geometry[0] < -180: raise ValueError("Longitude is out of bounds, check your JSON format or data") if geometry[1] > 90 or geometry[1] < -90: raise ValueError("Latitude is out of bounds, check your JSON format or data") # Discard z-coordinate, if it exists if geometry["type"] == "GeometryCollection": for idx, geo in enumerate(geometry["geometries"]): geometry["geometries"][idx]["coordinates"] = ensure_2d(geo["coordinates"]) check_bounds(geo["coordinates"]) else: geometry["coordinates"] = ensure_2d(geometry["coordinates"]) check_bounds(geometry["coordinates"]) wkt = geomet.wkt.dumps(geometry, decimals=decimals) # Strip unnecessary spaces wkt = re.sub(r"(?<!\d) ", "", wkt) return wkt
def _format_query_value(attr, value): """Format the value of a Solr query parameter.""" # Handle spaces by adding quotes around the string, if appropriate if isinstance(value, str): value = value.strip() if value == "": raise ValueError(f"Trying to filter '{attr}' with an empty string") # Handle strings surrounded by brackets specially to allow the user to make use of Solr syntax directly. # The string must not be quoted for that to work. if ( not any( value.startswith(s[0]) and value.endswith(s[1]) for s in ["[]", "{}", "//", "()", '""'] ) and "*" not in value and "?" not in value ): value = re.sub(r"\s", " ", value, re.M) value = f'"{value}"' # Handle date keywords # Keywords from is_date_attr = attr.lower() in [ "beginposition", "endposition", "date", "creationdate", "ingestiondate", ] if is_date_attr: # Automatically format date-type attributes if isinstance(value, str) and " TO " in value: # This is a string already formatted as a date interval, # e.g. '[NOW-1DAY TO NOW]' pass elif isinstance(value, (list, tuple)) and len(value) == 2: value = (format_query_date(value[0]), format_query_date(value[1])) else: raise ValueError( f"Date-type query parameter '{attr}' expects either a two-element tuple of " f"str or datetime objects or a '[<from> TO <to>]'-format string. Received {value}." ) if isinstance(value, set): raise ValueError(f"Unexpected set-type value encountered with keyword '{attr}'") elif isinstance(value, (list, tuple)): # Handle value ranges if len(value) == 2: # Allow None or "*" to be used as an unlimited bound if any(x == "" for x in value): raise ValueError(f"Trying to filter '{attr}' with an empty string") value = ["*" if x in (None, "*") else f'"{x}"' for x in value] if value == ["*", "*"] or (is_date_attr and value == ["*", "NOW"]): # Drop this keyword if both sides are unbounded return value = "[{} TO {}]".format(*value) else: raise ValueError( f"Invalid number of elements in list. Expected 2, received {len(value)}" ) return value
[docs]def format_query_date(in_date): r""" Format a date, datetime or a YYYYMMDD string input as YYYY-MM-DDThh:mm:ssZ or validate a date string as suitable for the full text search interface and return it. `None` will be converted to '\*', meaning an unlimited date bound in date ranges. Parameters ---------- in_date : str or datetime or date or None Date to be formatted Returns ------- str Formatted string Raises ------ ValueError If the input date type is incorrect or passed date string is invalid """ if in_date is None: return "*" if isinstance(in_date, (datetime, date)): return in_date.strftime("%Y-%m-%dT%H:%M:%SZ") elif not isinstance(in_date, str): raise ValueError("Expected a string or a datetime object. Received {}.".format(in_date)) in_date = in_date.strip() if in_date == "*": # '*' can be used for one-sided range queries e.g. ingestiondate:[* TO NOW-1YEAR] return in_date # Reference: # ISO-8601 date or NOW valid_date_pattern = r"^(?:\d{4}-\d\d-\d\dT\d\d:\d\d:\d\d(?:\.\d+)?Z|NOW)" # date arithmetic suffix is allowed units = r"(?:YEAR|MONTH|DAY|HOUR|MINUTE|SECOND)" valid_date_pattern += r"(?:[-+]\d+{}S?)*".format(units) # dates can be rounded to a unit of time # e.g. "NOW/DAY" for dates since 00:00 today valid_date_pattern += r"(?:/{}S?)*$".format(units) in_date = in_date.strip() if re.match(valid_date_pattern, in_date): return in_date try: return datetime.strptime(in_date, "%Y%m%d").strftime("%Y-%m-%dT%H:%M:%SZ") except ValueError: raise ValueError("Unsupported date value {}".format(in_date))
def _format_order_by(order_by): if not order_by or not order_by.strip(): return None output = [] for part in order_by.split(","): part = part.strip() dir = " asc" if part[0] == "+": part = part[1:] elif part[0] == "-": dir = " desc" part = part[1:] if not part or not part.isalnum(): raise ValueError("Invalid order by value ({})".format(order_by)) output.append(part + dir) return ",".join(output) def _parse_gml_footprint(geometry_str): # workaround for if geometry_str is None: # pragma: no cover return None geometry_xml = ET.fromstring(geometry_str) poly_coords_str = ( geometry_xml.find("{}outerBoundaryIs") .find("{}LinearRing") .findtext("{}coordinates") ) poly_coords = (coord.split(",")[::-1] for coord in poly_coords_str.split(" ")) coord_string = ",".join(" ".join(coord) for coord in poly_coords) return "POLYGON(({}))".format(coord_string) def _parse_iso_date(content): if "." in content: return datetime.strptime(content, "%Y-%m-%dT%H:%M:%S.%fZ") else: return datetime.strptime(content, "%Y-%m-%dT%H:%M:%SZ") def _parse_odata_timestamp(in_date): """Convert the timestamp received from OData JSON API to a datetime object.""" timestamp = int(in_date.replace("/Date(", "").replace(")/", "")) seconds = timestamp // 1000 ms = timestamp % 1000 return datetime.utcfromtimestamp(seconds) + timedelta(milliseconds=ms) def _parse_opensearch_response(products): """Convert a query response to a dictionary. The resulting dictionary structure is {<product id>: {<property>: <value>}}. The property values are converted to their respective Python types unless `parse_values` is set to `False`. """ converters = {"date": _parse_iso_date, "int": int, "long": int, "float": float, "double": float} # Keep the string type by default def default_converter(x): return x output = OrderedDict() for prod in products: product_dict = {} prod_id = prod["id"] output[prod_id] = product_dict for key in prod: if key == "id": continue if isinstance(prod[key], str): product_dict[key] = prod[key] else: properties = prod[key] if isinstance(properties, dict): properties = [properties] if key == "link": for p in properties: name = "link" if "rel" in p: name = "link_" + p["rel"] product_dict[name] = p["href"] else: f = converters.get(key, default_converter) for p in properties: try: product_dict[p["name"]] = f(p["content"]) except KeyError: # Sentinel-3 has one element 'arr' # which violates the name:content convention product_dict[p["name"]] = f(p["str"]) return output def _parse_odata_response(product): output = { "id": product["Id"], "title": product["Name"], "size": int(product["ContentLength"]), product["Checksum"]["Algorithm"].lower(): product["Checksum"]["Value"], "date": _parse_odata_timestamp(product["ContentDate"]["Start"]), "footprint": _parse_gml_footprint(product["ContentGeometry"]), "url": product["__metadata"]["media_src"], "Online": product.get("Online", True), "Creation Date": _parse_odata_timestamp(product["CreationDate"]), "Ingestion Date": _parse_odata_timestamp(product["IngestionDate"]), } # Parse the extended metadata, if provided converters = [int, float, _parse_iso_date] for attr in product["Attributes"].get("results", []): value = attr["Value"] for f in converters: try: value = f(attr["Value"]) break except ValueError: pass output[attr["Name"]] = value return output
[docs]def placename_to_wkt(place_name): """Geocodes the place name to rectangular bounding extents using Nominatim API and returns the corresponding 'ENVELOPE' form Well-Known-Text. Parameters ---------- place_name : str the query to geocode Raises ------ ValueError If no matches to the place name were found. Returns ------- wkt_envelope : str Bounding box of the location as an 'ENVELOPE(minX, maxX, maxY, minY)' WKT string. info : Dict[str, any] Matched location's metadata returned by Nominatim. """ rqst = requests.get( "", params={"q": place_name, "format": "geojson"}, headers={"User-Agent": "sentinelsat/" + sentinelsat_version}, ) rqst.raise_for_status() features = rqst.json()["features"] if len(features) == 0: raise ValueError('Unable to find a matching location for "{}"'.format(place_name)) # Get the First result's bounding box and description. feature = features[0] minX, minY, maxX, maxY = feature["bbox"] # ENVELOPE is a non-standard WKT format supported by Solr # wkt_envelope = "ENVELOPE({}, {}, {}, {})".format(minX, maxX, maxY, minY) info = feature["properties"] info["bbox"] = feature["bbox"] return wkt_envelope, info
[docs]def is_wkt(possible_wkt): pattern = r"^((MULTI)?(POINT|LINESTRING|POLYGON)|GEOMETRYCOLLECTION|ENVELOPE)\s*\(.+\)$" return re.match(pattern, possible_wkt.strip().upper()) is not None