# -*- coding: utf-8 -*-
# Copyright 2018, CS GROUP - France, https://www.csgroup.eu/
#
# This file is part of EODAG project
# https://www.github.com/CS-SI/EODAG
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""Miscellaneous utilities to be used throughout eodag.
Everything that does not fit into one of the specialised categories of utilities in
this package should go here
"""
from __future__ import annotations
import ast
import errno
import functools
import hashlib
import inspect
import logging as py_logging
import mimetypes
import os
import re
import shutil
import ssl
import string
import struct
import sys
import types
import unicodedata
import warnings
from collections import defaultdict
from copy import deepcopy as copy_deepcopy
from email.message import Message
from glob import glob
from importlib.metadata import metadata
from itertools import repeat, starmap
from pathlib import Path
from tempfile import mkdtemp
from typing import TYPE_CHECKING, Any, Callable, Iterator, Optional, Union, cast
from urllib.parse import urlparse, urlsplit
from urllib.request import url2pathname
if sys.version_info >= (3, 12):
from typing import Unpack # type: ignore # noqa
else:
from typing_extensions import Unpack # noqa
import orjson
from tqdm.auto import tqdm
from .exceptions import MisconfiguredError
from .logging import get_disable_tqdm
from .logging import logging as eodag_logging
from .streamresponse import StreamResponse, StreamResponseContent
if TYPE_CHECKING:
from jsonpath_ng import JSONPath, jsonpath
from pydantic import ValidationError as PydanticValidationError
from shapely.geometry.base import BaseGeometry
from eodag.api.product._product import EOProduct
logger = py_logging.getLogger("eodag.utils")
# Constants - Core
# ---------------
#: Used as template for unknown/custom collection usage
GENERIC_COLLECTION = "GENERIC_COLLECTION"
#: Generic provider used in :meth:`~eodag.api.core.EODataAccessGateway.import_stac_items`
#: if no existing provider can be used
GENERIC_STAC_PROVIDER = "generic_stac_provider"
#: List of known STAC search plugins. Required to complete plugin configuration with STAC plugins specific features.
STAC_SEARCH_PLUGINS = [
"GeodesSearch",
"StacSearch",
"StacListAssets",
"StaticStacSearch",
]
#: actual STAC version in EODAG
STAC_VERSION = "1.1.0"
# Constants - HTTP requests
# ------------------------
eodag_version = metadata("eodag")["Version"]
#: EODAG user agent used in HTTP requests
USER_AGENT = {"User-Agent": f"eodag/{eodag_version}"}
#: default timeout for HTTP requests (in seconds)
HTTP_REQ_TIMEOUT = 5
#: default timeout for search requests (in seconds)
DEFAULT_SEARCH_TIMEOUT = 20
#: default timeout for stream requests (in seconds)
DEFAULT_STREAM_REQUESTS_TIMEOUT = 60
#: default count for HTTP requests retry strategy
REQ_RETRY_TOTAL = 3
#: default backoff factor for HTTP requests retry strategy
REQ_RETRY_BACKOFF_FACTOR = 2
#: default status codes for which HTTP requests retry strategy is applied
REQ_RETRY_STATUS_FORCELIST = [401, 429, 500, 502, 503, 504]
#: default wait time (in minutes) between download attempts
DEFAULT_DOWNLOAD_WAIT = 0.2
#: default timeout (in minutes) for download attempts
DEFAULT_DOWNLOAD_TIMEOUT = 10
#: default token expiration margin (in seconds). Safety buffer to prevent token rejection from unexpected expiry
#: between validity check and request. Default value of :attr:`~eodag.config.PluginConfig.token_expiration_margin`
DEFAULT_TOKEN_EXPIRATION_MARGIN = 60
# Constants - Pagination
# ---------------------
#: pagination default starting page number
DEFAULT_PAGE = 1
#: default number of items per page requested by :meth:`~eodag.api.core.EODataAccessGateway.search`
DEFAULT_LIMIT = 20
#: Default maximum number of items per page requested by :meth:`~eodag.api.core.EODataAccessGateway.search_all`.
#: 50 instead of 20 (:const:`~eodag.utils.DEFAULT_LIMIT`) to increase it to the known and current minimum
#: value (mundi provider)
DEFAULT_MAX_LIMIT = 50
#: known next page token keys used to guess key in STAC providers next link responses
KNOWN_NEXT_PAGE_TOKEN_KEYS = ["token", "next", "page", "skip"]
# Constants - Metadata-mapping / default values
# --------------------------------------------
#: default projection used in metadata-mapping converters and :class:`~eodag.plugins.search.csw.CSWSearch`
DEFAULT_PROJ = "EPSG:4326"
#: default collections start date
DEFAULT_MISSION_START_DATE = "2015-01-01T00:00:00.000Z"
#: default geometry / whole world bounding box
# DEFAULT_SHAPELY_GEOMETRY is lazily computed via __getattr__ to defer shapely import
#: Online status value for ``order:status`` property
ONLINE_STATUS = "succeeded"
# Constants - Metadata-mapping / JSONPath regex
# --------------------------------------------
#: regex to detect if a string is a JSONPath expression, used in :func:`eodag.utils.string_to_jsonpath`
JSONPATH_MATCH = re.compile(r"^[\{\(]*\$(\..*)*$")
#: regex to detect if a string is a simple/workable JSONPath expression, that can be parsed using a simpler, optimized
#: approach. Used in :func:`eodag.utils.string_to_jsonpath`
WORKABLE_JSONPATH_MATCH = re.compile(r"^\$(\.[a-zA-Z0-9-_:\.\[\]\"\(\)=\?\*]+)*$")
#: regex to detect if a string is a JSONPath array field, used in :func:`eodag.utils.string_to_jsonpath`
ARRAY_FIELD_MATCH = re.compile(r"^[a-zA-Z0-9-_:]+(\[[0-9\*]+\])+$")
DEFAULT_MIME = "application/octet-stream"
# update missing mimetypes
mimetypes.add_type("text/xml", ".xsd")
mimetypes.add_type("application/x-grib", ".grib")
mimetypes.add_type("application/x-grib2", ".grib2")
# jp2 is missing on windows
mimetypes.add_type("image/jp2", ".jp2")
# ---------------------------------------------------------------------------
# PEP 562 lazy module attributes
# ---------------------------------------------------------------------------
_DEFAULT_SHAPELY_GEOMETRY = None
def _build_float_range_cls() -> type:
"""Build the :class:`FloatRange` class lazily to avoid importing ``click`` at
module load time (``click`` is only needed by the CLI)."""
import click
class FloatRange(click.types.FloatParamType):
"""A parameter that works similar to :data:`click.FLOAT` but restricts the
value to fit into a range. Fails if the value doesn't fit into the range.
"""
name = "percentage"
def __init__(
self, min: Optional[float] = None, max: Optional[float] = None
) -> None:
self.min = min
self.max = max
def convert(
self,
value: Any,
param: Optional["click.core.Parameter"],
ctx: Optional["click.core.Context"],
) -> Any:
"""Convert value"""
rv = click.types.FloatParamType.convert(self, value, param, ctx)
if (
self.min is not None
and rv < self.min
or self.max is not None
and rv > self.max
):
if self.min is None:
self.fail(
"%s is bigger than the maximum valid value %s."
% (rv, self.max),
param,
ctx,
)
elif self.max is None:
self.fail(
"%s is smaller than the minimum valid value %s."
% (rv, self.min),
param,
ctx,
)
else:
self.fail(
"%s is not in the valid range of %s to %s."
% (rv, self.min, self.max),
param,
ctx,
)
return rv
def __repr__(self) -> str:
return "FloatRange(%r, %r)" % (self.min, self.max)
return FloatRange
def __getattr__(name: str):
if name == "DEFAULT_SHAPELY_GEOMETRY":
global _DEFAULT_SHAPELY_GEOMETRY
if _DEFAULT_SHAPELY_GEOMETRY is None:
from shapely.geometry import box
_DEFAULT_SHAPELY_GEOMETRY = box(-180, -90, 180, 90)
globals()["DEFAULT_SHAPELY_GEOMETRY"] = _DEFAULT_SHAPELY_GEOMETRY
return _DEFAULT_SHAPELY_GEOMETRY
if name == "FloatRange":
cls = _build_float_range_cls()
globals()["FloatRange"] = cls
return cls
raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
def _deprecated(reason: str = "", version: Optional[str] = None) -> Callable[..., Any]:
"""Simple decorator to mark functions/methods/classes as deprecated.
Warning: Does not work with staticmethods!
@deprecate(reason="why", version="1.2")
def foo():
pass
foo()
DeprecationWarning: Call to deprecated function/method foo (why) -- Deprecated since v1.2
"""
def decorator(callable: Callable[..., Any]) -> Any:
if inspect.isclass(callable):
ctype = "class"
else:
ctype = "function/method"
cname = callable.__name__
reason_ = f"({reason})" if reason else ""
version_ = f" -- Deprecated since v{version}" if version else ""
@functools.wraps(callable)
def wrapper(*args: Any, **kwargs: Any) -> Any:
warnings.warn(
f"Call to deprecated eodag {ctype} {cname} {reason_}{version_}",
category=DeprecationWarning,
stacklevel=2,
)
return callable(*args, **kwargs)
return wrapper
return decorator
[docs]
def slugify(value: Any, allow_unicode: bool = False) -> str:
"""Copied from Django Source code, only modifying last line (no need for safe
strings).
source: https://github.com/django/django/blob/master/django/utils/text.py
Convert to ASCII if ``allow_unicode`` is ``False``. Convert spaces to hyphens.
Remove characters that aren't alphanumerics, underscores, or hyphens.
Convert to lowercase. Also strip leading and trailing whitespace.
"""
value = str(value)
if allow_unicode:
value = unicodedata.normalize("NFKC", value)
else:
value = (
unicodedata.normalize("NFKD", value)
.encode("ascii", "ignore")
.decode("ascii")
)
value = re.sub(r"[^\w\s-]", "", value).strip().lower()
return re.sub(r"[-\s]+", "-", value)
[docs]
def sanitize(value: str) -> str:
"""Sanitize string to be used as a name of a directory.
>>> sanitize('productName')
'productName'
>>> sanitize('name with multiple spaces')
'name_with_multiple_spaces'
>>> sanitize('âtre fête île alcôve bûche çà génèse où Noël ovoïde capharnaüm')
'atre_fete_ile_alcove_buche_ca_genese_ou_Noel_ovoide_capharnaum'
>>> sanitize('replace,ponctuation:;signs!?byunderscorekeeping-hyphen.dot_and_underscore') # noqa
'replace_ponctuation_signs_byunderscorekeeping-hyphen.dot_and_underscore'
"""
# remove accents
rv = strip_accents(value)
# replace punctuation signs and spaces by underscore
# keep hyphen, dot and underscore from punctuation
tobereplaced = re.sub(r"[-_.]", "", string.punctuation)
# add spaces to be removed
tobereplaced += r"\s"
rv = re.sub(r"[" + tobereplaced + r"]+", "_", rv)
return str(rv)
[docs]
def strip_accents(s: str) -> str:
"""Strip accents of a string.
>>> strip_accents('productName')
'productName'
>>> strip_accents('génèse')
'genese'
>>> strip_accents('preserve-punct-special-chars:;,?!§%$£œ')
'preserve-punct-special-chars:;,?!§%$£œ'
"""
# Mn stands for a nonspacing combining mark
return "".join(
c for c in unicodedata.normalize("NFD", s) if unicodedata.category(c) != "Mn"
)
[docs]
def uri_to_path(uri: str) -> str:
"""
Convert a file URI (e.g. ``file:///tmp``) to a local path (e.g. ``/tmp``)
"""
if not uri.startswith("file"):
raise ValueError("A file URI must be provided (e.g. 'file:///tmp'")
_, _, path, _, _ = urlsplit(uri)
# On Windows urlsplit returns the path starting with a slash ('/C:/User)
path = url2pathname(path)
# url2pathname removes it
return path
[docs]
def path_to_uri(path: str) -> str:
"""Convert a local absolute path to a file URI"""
return Path(path).as_uri()
[docs]
def mutate_dict_in_place(func: Callable[[Any], Any], mapping: dict[Any, Any]) -> None:
"""Apply func to values of mapping.
The mapping object's values are modified in-place. The function is recursive,
allowing to also modify values of nested dicts that may be level-1 values of
mapping.
:param func: A function to apply to each value of mapping which is not a dict object
:param mapping: A Python dict object
:returns: None
"""
for key, value in mapping.items():
if isinstance(value, dict):
mutate_dict_in_place(func, value)
else:
mapping[key] = func(value)
[docs]
def merge_mappings(mapping1: dict[Any, Any], mapping2: dict[Any, Any]) -> None:
"""Merge two mappings with string keys, values from ``mapping2`` overriding values
from ``mapping1``.
Do its best to detect the key in ``mapping1`` to override. For example:
>>> mapping2 = {"ext_keya": "new"}
>>> mapping1 = {"ext:keyA": "obsolete"}
>>> merge_mappings(mapping1, mapping2)
>>> mapping1
{'ext:keyA': 'new'}
If ``mapping2`` has a key that cannot be detected in ``mapping1``, this new key is
added to ``mapping1`` as is.
:param mapping1: The mapping containing values to be overridden
:param mapping2: The mapping containing values that will override the first mapping
"""
# A mapping between mapping1 keys as lowercase strings and original mapping1 keys
m1_keys_lowercase = {key.lower().replace(":", "_"): key for key in mapping1}
for key, value in mapping2.items():
if isinstance(value, dict):
try:
merge_mappings(mapping1[key], value)
except KeyError:
# If the key from mapping2 is not in mapping1, it is either key is
# the lowercased form of the corresponding key in mapping1 or because
# key is a new key to be added in mapping1
current_value = mapping1.setdefault(m1_keys_lowercase.get(key, key), {})
if not current_value:
current_value.update(value)
else:
merge_mappings(current_value, value)
else:
# Even for "scalar" values (a.k.a not nested structures), first check if
# the key from mapping1 is not the lowercase version of a key in mapping2.
# Otherwise, create the key in mapping1. This is the meaning of
# `m1_keys_lowercase.get(key, key)`
current_value = mapping1.get(m1_keys_lowercase.get(key, key))
if current_value is not None:
try:
# If current or new value is a list (search queryable parameter), simply replace current with new
if (
isinstance(value, list)
and not isinstance(current_value, list)
or not isinstance(value, list)
and isinstance(current_value, list)
):
mapping1[m1_keys_lowercase.get(key, key)] = value
else:
mapping1[m1_keys_lowercase.get(key, key)] = cast_scalar_value(
value, type(current_value)
)
except (TypeError, ValueError):
# Ignore any override value that does not have the same type
# as the default value
logger.debug(
"Ignored '%s' setting override from '%s' to '%s', (could not cast %s to %s)",
key,
current_value,
value,
type(value),
type(current_value),
)
pass
else:
mapping1[key] = value
[docs]
def maybe_generator(obj: Any) -> Iterator[Any]:
"""Generator function that get an arbitrary object and generate values from it if
the object is a generator."""
if isinstance(obj, types.GeneratorType):
for elt in obj:
yield elt
else:
yield obj
[docs]
class DownloadedCallback:
"""Example class for callback after each download in :meth:`~eodag.api.core.EODataAccessGateway.download_all`"""
[docs]
def __call__(self, product: EOProduct) -> None:
"""Callback
:param product: The downloaded EO product
"""
logger.debug("Download finished for the product %s", product)
[docs]
class ProgressCallback(tqdm):
"""A callable used to render progress to users for long running processes.
It inherits from :class:`tqdm.auto.tqdm`, and accepts the same arguments on
instantiation: ``iterable``, ``desc``, ``total``, ``leave``, ``file``, ``ncols``,
``mininterval``, ``maxinterval``, ``miniters``, ``ascii``, ``disable``, ``unit``,
``unit_scale``, ``dynamic_ncols``, ``smoothing``, ``bar_format``, ``initial``,
``position``, ``postfix``, ``unit_divisor``.
It can be globally disabled using ``eodag.utils.logging.setup_logging(0)`` or
``eodag.utils.logging.setup_logging(level, no_progress_bar=True)``, and
individually disabled using ``disable=True``.
"""
def __init__(self, *args: Any, **kwargs: Any) -> None:
self.kwargs = kwargs.copy()
if "unit" not in kwargs:
kwargs["unit"] = "B"
if "unit_scale" not in kwargs:
kwargs["unit_scale"] = True
if "desc" not in kwargs:
kwargs["desc"] = ""
if "position" not in kwargs:
kwargs["position"] = 0
if "disable" not in kwargs:
kwargs["disable"] = get_disable_tqdm()
if "dynamic_ncols" not in kwargs:
kwargs["dynamic_ncols"] = True
super(ProgressCallback, self).__init__(*args, **kwargs)
def __call__(self, increment: int, total: Optional[int] = None) -> None:
"""Update the progress bar.
:param increment: Amount of data already processed
:param total: (optional) Maximum amount of data to be processed
"""
if total is not None and total != self.total:
self.reset(total=total)
self.update(increment)
def copy(self, *args: Any, **kwargs: Any) -> ProgressCallback:
"""Returns another progress callback using the same initial
keyword-arguments.
Optional ``args`` and ``kwargs`` parameters will be used to create a
new :class:`~eodag.utils.ProgressCallback` instance, overriding initial
`kwargs`.
"""
return ProgressCallback(*args, **dict(self.kwargs, **kwargs))
[docs]
def repeatfunc(func: Callable[..., Any], n: int, *args: Any) -> starmap:
"""Call ``func`` ``n`` times with ``args``"""
return starmap(func, repeat(args, n))
[docs]
def makedirs(dirpath: str) -> None:
"""Create a directory in filesystem with parents if necessary"""
try:
os.makedirs(dirpath)
except OSError as err:
# Reraise the error unless it's about an already existing directory
if err.errno != errno.EEXIST or not os.path.isdir(dirpath):
raise
[docs]
def rename_subfolder(dirpath: str, name: str) -> None:
"""Rename first subfolder found in ``dirpath`` with given ``name``,
raise :class:`RuntimeError` if no subfolder can be found
:param dirpath: path to the directory containing the subfolder
:param name: new name of the subfolder
:raises: :class:`RuntimeError`
Example:
>>> import os
>>> import tempfile
>>> with tempfile.TemporaryDirectory() as tmpdir:
... somefolder = os.path.join(tmpdir, "somefolder")
... otherfolder = os.path.join(tmpdir, "otherfolder")
... os.makedirs(somefolder)
... assert os.path.isdir(somefolder) and not os.path.isdir(otherfolder)
... rename_subfolder(tmpdir, "otherfolder")
... assert not os.path.isdir(somefolder) and os.path.isdir(otherfolder)
Before::
$ tree <tmp-folder>
<tmp-folder>
└── somefolder
└── somefile
After::
$ tree <tmp-folder>
<tmp-folder>
└── otherfolder
└── somefile
"""
try:
subdir, *_ = (p for p in glob(os.path.join(dirpath, "*")) if os.path.isdir(p))
except ValueError:
raise RuntimeError(f"No subfolder was found in {dirpath}")
os.rename(
subdir,
os.path.join(dirpath, name),
)
[docs]
def rename_with_version(file_path: str, suffix: str = "old") -> str:
"""
Renames a file by appending and incrementing a version number if a conflict exists.
:param file_path: full path of the file to rename
:param suffix: suffix preceding version number in case of name conflict
:returns: new file path with the version appended or incremented
Example:
>>> import tempfile
>>> from pathlib import Path
>>> with tempfile.TemporaryDirectory() as tmpdir:
... file_path = (Path(tmpdir) / "foo.txt")
... file_path.touch()
... (Path(tmpdir) / "foo_old1.txt").touch()
... expected = str(Path(tmpdir) / "foo_old2.txt")
... assert expected == rename_with_version(str(file_path))
"""
if not os.path.isfile(file_path):
raise FileNotFoundError(f"The file '{file_path}' does not exist.")
dir_path, file_name = os.path.split(file_path)
file_base, file_ext = os.path.splitext(file_name)
new_file_path = file_path
# loop and iterate on conflicting existing files
version = 0
while os.path.exists(new_file_path):
version += 1
new_file_name = f"{file_base}_{suffix}{version}{file_ext}"
new_file_path = os.path.join(dir_path, new_file_name)
# Rename the file
os.rename(file_path, new_file_path)
return new_file_path
[docs]
def jsonpath_parse_dict_items(
jsonpath_dict: dict[str, Any], values_dict: dict[str, Any]
) -> dict[Any, Any]:
"""Recursively parse :class:`jsonpath_ng.JSONPath` elements in dict
>>> import jsonpath_ng.ext as jsonpath
>>> from jsonpath_ng.ext import parse
>>> jsonpath_parse_dict_items(
... {"foo": {"bar": parse("$.a.b")}, "qux": [parse("$.c"), parse("$.c")]},
... {"a":{"b":"baz"}, "c":"quux"}
... ) == {'foo': {'bar': 'baz'}, 'qux': ['quux', 'quux']}
True
:param jsonpath_dict: Dictionary having :class:`jsonpath_ng.JSONPath` values that need to be parsed
:param values_dict: Values dict used as args for parsing
:returns: Updated dict
"""
return dict_items_recursive_apply(jsonpath_dict, parse_jsonpath, **values_dict)
[docs]
def update_nested_dict(
old_dict: dict[Any, Any],
new_dict: dict[Any, Any],
extend_list_values: bool = False,
allow_empty_values: bool = False,
allow_extend_duplicates: bool = True,
) -> dict[Any, Any]:
"""Update recursively ``old_dict`` items with ``new_dict`` ones
>>> update_nested_dict(
... {"a": {"a.a": 1, "a.b": 2}, "b": 3},
... {"a": {"a.a": 10}}
... ) == {'a': {'a.a': 10, 'a.b': 2}, 'b': 3}
True
>>> update_nested_dict(
... {"a": {"a.a": [1, 2]}},
... {"a": {"a.a": [10, 2]}},
... extend_list_values=True,
... allow_extend_duplicates=True
... ) == {'a': {'a.a': [1, 2, 10, 2]}}
True
>>> update_nested_dict(
... {"a": {"a.a": [1, 2]}},
... {"a": {"a.a": [10, 2]}},
... extend_list_values=True,
... allow_extend_duplicates=False
... ) == {'a': {'a.a': [1, 2, 10]}}
True
>>> update_nested_dict(
... {"a": {"a.a": 1, "a.b": 2}, "b": 3},
... {"a": {"a.a": None}},
... ) == {'a': {'a.a': 1, 'a.b': 2}, 'b': 3}
True
>>> update_nested_dict(
... {"a": {"a.a": 1, "a.b": 2}, "b": 3},
... {"a": {"a.a": None}},
... allow_empty_values=True
... ) == {'a': {'a.a': None, 'a.b': 2}, 'b': 3}
True
:param old_dict: Dict to be updated
:param new_dict: Incomming dict
:param extend_list_values: (optional) Extend old_dict value if both old/new values are lists
:param allow_empty_values: (optional) Allow update with empty values
:returns: Updated dict
"""
for k, v in new_dict.items():
if k in old_dict.keys():
if isinstance(v, dict) and isinstance(old_dict[k], dict):
old_dict[k] = update_nested_dict(
old_dict[k],
v,
extend_list_values=extend_list_values,
allow_empty_values=allow_empty_values,
allow_extend_duplicates=allow_extend_duplicates,
)
elif (
extend_list_values
and isinstance(old_dict[k], list)
and isinstance(v, list)
and (
# no common elements
not any([x for x in v if x in old_dict[k]])
# common elements
or any([x for x in v if x in old_dict[k]])
and allow_extend_duplicates
)
):
old_dict[k].extend(v)
elif (
extend_list_values
and isinstance(old_dict[k], list)
and isinstance(v, list)
# common elements
and any([x for x in v if x in old_dict[k]])
and not allow_extend_duplicates
):
old_dict[k].extend([x for x in v if x not in old_dict[k]])
elif (v and not allow_empty_values) or allow_empty_values:
old_dict[k] = v
else:
old_dict[k] = v
return old_dict
[docs]
def items_recursive_apply(
input_obj: Union[dict[Any, Any], list[Any]],
apply_method: Callable[..., Any],
**apply_method_parameters: Any,
) -> Union[dict[Any, Any], list[Any]]:
"""Recursive apply method to items contained in input object (dict or list)
>>> items_recursive_apply(
... {"foo": {"bar":"baz"}, "qux": ["a","b"]},
... lambda k,v,x: v.upper()+x, **{"x":"!"}
... ) == {'foo': {'bar': 'BAZ!'}, 'qux': ['A!', 'B!']}
True
>>> items_recursive_apply(
... [{"foo": {"bar":"baz"}}, "qux"],
... lambda k,v,x: v.upper()+x,
... **{"x":"!"})
[{'foo': {'bar': 'BAZ!'}}, 'QUX!']
>>> items_recursive_apply(
... "foo",
... lambda k,v,x: v.upper()+x,
... **{"x":"!"})
'foo'
:param input_obj: Input object (dict or list)
:param apply_method: Method to be applied to dict elements
:param apply_method_parameters: Optional parameters passed to the method
:returns: Updated object
"""
if isinstance(input_obj, dict):
return dict_items_recursive_apply(
input_obj, apply_method, **apply_method_parameters
)
elif isinstance(input_obj, list):
return list_items_recursive_apply(
input_obj, apply_method, **apply_method_parameters
)
else:
logger.warning("Could not use items_recursive_apply on %s" % type(input_obj))
return input_obj
[docs]
def dict_items_recursive_apply(
config_dict: dict[Any, Any],
apply_method: Callable[..., Any],
**apply_method_parameters: Any,
) -> dict[Any, Any]:
"""Recursive apply method to dict elements
>>> dict_items_recursive_apply(
... {"foo": {"bar": "baz"}, "qux": ["a", "b"]},
... lambda k, v, x: v.upper() + x, **{"x": "!"}
... ) == {'foo': {'bar': 'BAZ!'}, 'qux': ['A!', 'B!']}
True
:param config_dict: Input nested dictionary
:param apply_method: Method to be applied to dict elements
:param apply_method_parameters: Optional parameters passed to the method
:returns: Updated dict
"""
result_dict: dict[Any, Any] = deepcopy(config_dict)
for dict_k, dict_v in result_dict.items():
if isinstance(dict_v, dict):
result_dict[dict_k] = dict_items_recursive_apply(
dict_v, apply_method, **apply_method_parameters
)
elif any(isinstance(dict_v, t) for t in (list, tuple)):
result_dict[dict_k] = list_items_recursive_apply(
list(dict_v), apply_method, **apply_method_parameters
)
else:
result_dict[dict_k] = apply_method(
dict_k, dict_v, **apply_method_parameters
)
return result_dict
[docs]
def list_items_recursive_apply(
config_list: list[Any],
apply_method: Callable[..., Any],
**apply_method_parameters: Any,
) -> list[Any]:
"""Recursive apply method to list elements
>>> list_items_recursive_apply(
... [{"foo": {"bar": "baz"}}, "qux"],
... lambda k, v, x: v.upper() + x,
... **{"x": "!"})
[{'foo': {'bar': 'BAZ!'}}, 'QUX!']
:param config_list: Input list containing nested lists/dicts
:param apply_method: Method to be applied to list elements
:param apply_method_parameters: Optional parameters passed to the method
:returns: Updated list
"""
result_list = deepcopy(config_list)
for list_idx, list_v in enumerate(result_list):
if isinstance(list_v, dict):
result_list[list_idx] = dict_items_recursive_apply(
list_v, apply_method, **apply_method_parameters
)
elif any(isinstance(list_v, t) for t in (list, tuple)):
result_list[list_idx] = list_items_recursive_apply(
list_v, apply_method, **apply_method_parameters
)
else:
result_list[list_idx] = apply_method(
list_idx, list_v, **apply_method_parameters
)
return result_list
[docs]
def items_recursive_sort(
input_obj: Union[list[Any], dict[Any, Any]],
) -> Union[list[Any], dict[Any, Any]]:
"""Recursive sort dict items contained in input object (dict or list)
>>> items_recursive_sort(
... {"b": {"b": "c", "a": 0}, "a": ["b", {2: 0, 0: 1, 1: 2}]},
... ) == {"a": ["b", {0: 1, 1: 2, 2: 0}], "b": {"a": 0, "b": "c"}}
True
>>> items_recursive_sort(["b", {2: 0, 0: 1, 1:2}])
['b', {0: 1, 1: 2, 2: 0}]
>>> items_recursive_sort("foo")
'foo'
:param input_obj: Input object (dict or list)
:returns: Updated object
"""
if isinstance(input_obj, dict):
return dict_items_recursive_sort(input_obj)
elif isinstance(input_obj, list):
return list_items_recursive_sort(input_obj)
else:
logger.warning("Could not use items_recursive_sort on %s" % type(input_obj))
return input_obj
[docs]
def dict_items_recursive_sort(config_dict: dict[Any, Any]) -> dict[Any, Any]:
"""Recursive sort dict elements
>>> dict_items_recursive_sort(
... {"b": {"b": "c", "a": 0}, "a": ["b", {2: 0, 0: 1, 1: 2}]},
... ) == {"a": ["b", {0: 1, 1: 2, 2: 0}], "b": {"a": 0, "b": "c"}}
True
:param config_dict: Input nested dictionary
:returns: Updated dict
"""
result_dict: dict[Any, Any] = deepcopy(config_dict)
for dict_k, dict_v in result_dict.items():
if isinstance(dict_v, dict):
result_dict[dict_k] = dict_items_recursive_sort(dict_v)
elif any(isinstance(dict_v, t) for t in (list, tuple)):
result_dict[dict_k] = list_items_recursive_sort(dict_v)
else:
result_dict[dict_k] = dict_v
return dict(sorted(result_dict.items()))
[docs]
def list_items_recursive_sort(config_list: list[Any]) -> list[Any]:
"""Recursive sort dicts in list elements
>>> list_items_recursive_sort(["b", {2: 0, 0: 1, 1: 2}])
['b', {0: 1, 1: 2, 2: 0}]
:param config_list: Input list containing nested lists/dicts
:returns: Updated list
"""
result_list: list[Any] = deepcopy(config_list)
for list_idx, list_v in enumerate(result_list):
if isinstance(list_v, dict):
result_list[list_idx] = dict_items_recursive_sort(list_v)
elif any(isinstance(list_v, t) for t in (list, tuple)):
result_list[list_idx] = list_items_recursive_sort(list_v)
else:
result_list[list_idx] = list_v
return result_list
[docs]
def string_to_jsonpath(*args: Any, force: bool = False) -> Union[str, JSONPath]:
"""Get :class:`jsonpath_ng.JSONPath` for ``$.foo.bar`` like string
>>> from jsonpath_ng.jsonpath import Child, Fields, Index, Root, Slice
>>> string_to_jsonpath(None, "$.foo.bar")
Child(Child(Root(), Fields('foo')), Fields('bar'))
>>> string_to_jsonpath("$.foo.bar")
Child(Child(Root(), Fields('foo')), Fields('bar'))
>>> string_to_jsonpath("foo")
'foo'
>>> string_to_jsonpath("foo", force=True)
Fields('foo')
>>> string_to_jsonpath('$.foo[0][*]') == Child(
... Child(Child(Root(), Fields('foo')), Index(0)),
... Slice(start=None, end=None, step=None),
... )
True
:param args: Last arg as input string value, to be converted
:param force: force conversion even if input string is not detected as a :class:`jsonpath_ng.JSONPath`
:returns: Parsed value
"""
from jsonpath_ng.jsonpath import Child, Fields, Index, Root, Slice
path_str: str = args[-1]
if JSONPATH_MATCH.match(str(path_str)) or force:
try:
common_jsonpath = "$"
common_jsonpath_parsed = Root()
# combine with common jsonpath if possible
if WORKABLE_JSONPATH_MATCH.match(path_str):
path_suffix = path_str[len(common_jsonpath) + 1 :]
path_splits = path_suffix.split(".") if path_suffix else []
parsed_path = common_jsonpath_parsed
for path_split in path_splits:
path_split = path_split.strip("'").strip('"')
if "[" in path_split and ARRAY_FIELD_MATCH.match(path_split):
# handle nested array
indexed_path_and_indexes = path_split[:-1].split("[")
indexed_path = indexed_path_and_indexes[0]
parsed_path = Child(parsed_path, Fields(indexed_path))
for idx in range(len(indexed_path_and_indexes) - 1):
index: Union[int, str] = (
indexed_path_and_indexes[idx + 1][:-1]
if idx < len(indexed_path_and_indexes) - 2
else indexed_path_and_indexes[idx + 1]
)
# wildcard index
if index == "*":
parsed_path = Child(
parsed_path,
Slice(start=None, end=None, step=None),
)
continue
try:
index = int(index)
except ValueError:
# unsupported index
parsed_path = cached_parse(path_str)
break
# integer index
parsed_path = Child(
parsed_path,
Index(index),
)
elif "[" in path_split:
# unsupported array field
parsed_path = cached_parse(path_str)
break
else:
parsed_path = Child(parsed_path, Fields(path_split))
return parsed_path
else:
return cached_parse(path_str)
except Exception: # jsonpath_ng does not provide a proper exception
# If str_value does not contain a jsonpath, return it as is
return path_str
else:
return path_str
[docs]
def parse_jsonpath(
key: str, jsonpath_obj: Union[str, jsonpath.Child], **values_dict: dict[str, Any]
) -> Optional[str]:
"""Parse jsonpah in ``jsonpath_obj`` using ``values_dict``
>>> import jsonpath_ng.ext as jsonpath
>>> from jsonpath_ng.ext import parse
>>> parse_jsonpath(None, parse("$.foo.bar"), **{"foo": {"bar": "baz"}})
'baz'
:param key: Input item key
:param jsonpath_obj: Input item value, to be parsed
:param values_dict: Values used as args for parsing
:returns: Parsed value
"""
from jsonpath_ng import jsonpath
if isinstance(jsonpath_obj, jsonpath.Child):
match = jsonpath_obj.find(values_dict)
return match[0].value if len(match) == 1 else None
else:
return jsonpath_obj
[docs]
def nested_pairs2dict(pairs: Union[list[Any], Any]) -> Union[Any, dict[Any, Any]]:
"""Create a dict using nested pairs
>>> nested_pairs2dict([["foo", [["bar", "baz"]]]])
{'foo': {'bar': 'baz'}}
:param pairs: Pairs of key / value
:returns: Created dict
"""
d = {}
try:
for k, v in pairs:
if isinstance(v, list):
v = nested_pairs2dict(v)
d[k] = v
except ValueError:
return pairs
return d
[docs]
def get_geometry_from_various(
locations_config: list[dict[str, Any]] = [], **query_args: Any
) -> Optional[BaseGeometry]:
"""Creates a ``shapely.geometry`` using given query kwargs arguments
:param locations_config: (optional) EODAG locations configuration
:param query_args: Query kwargs arguments from :meth:`~eodag.api.core.EODataAccessGateway.search`
:returns: shapely Geometry found
:raises shapely.errors.ShapelyError: Error while creating shapely geometry
:raises TypeError: Unexpected geometry type
:raises ValueError: Location name is wrong or its value does not match
"""
import shapefile
import shapely.wkt
from shapely.geometry import Polygon, shape
from shapely.geometry.base import GEOMETRY_TYPES, BaseGeometry
geom = None
if "geometry" in query_args:
geom_arg = query_args["geometry"]
bbox_keys = ["lonmin", "latmin", "lonmax", "latmax"]
if isinstance(geom_arg, dict) and geom_arg.get("type") in GEOMETRY_TYPES:
# geojson geometry
geom = cast(BaseGeometry, shape(geom_arg))
elif isinstance(geom_arg, dict) and all(k in geom_arg for k in bbox_keys):
# bbox dict
geom = Polygon(
(
(geom_arg["lonmin"], geom_arg["latmin"]),
(geom_arg["lonmin"], geom_arg["latmax"]),
(geom_arg["lonmax"], geom_arg["latmax"]),
(geom_arg["lonmax"], geom_arg["latmin"]),
)
)
elif isinstance(geom_arg, (list, tuple)) and len(geom_arg) >= 4:
# bbox list
geom = Polygon(
(
(geom_arg[0], geom_arg[1]),
(geom_arg[0], geom_arg[3]),
(geom_arg[2], geom_arg[3]),
(geom_arg[2], geom_arg[1]),
)
)
elif isinstance(geom_arg, str):
# WKT geometry
geom = shapely.wkt.loads(geom_arg)
elif isinstance(geom_arg, BaseGeometry):
geom = geom_arg
elif geom_arg is None:
pass
else:
raise TypeError("Unexpected geometry type: {}".format(type(geom_arg)))
# look for location name in locations configuration
locations_dict = {loc["name"]: loc for loc in locations_config}
# The location query kwargs can either be in query_args or in query_args["locations"],
# support for which were added in 2.0.0 and 2.1.0 respectively.
# The location query kwargs in query_args is supported for backward compatibility,
# the recommended usage is that they are in query_args["locations"]
locations = query_args.get("locations")
locations = locations if locations is not None else {}
# In query_args["locations"] we can check that the location_names are correct
locations = locations if locations is not None else {}
for location_name in locations:
if location_name not in locations_dict:
raise ValueError(
f"The location name {location_name} is wrong. "
f"It must be one of: {locations_dict.keys()}"
)
query_locations = {**query_args, **locations}
for arg in query_locations.keys():
if arg in locations_dict.keys():
found = False
pattern = rf"{query_locations[arg]}"
attr = locations_dict[arg]["attr"]
with shapefile.Reader(locations_dict[arg]["path"]) as shp:
for shaperec in shp.shapeRecords():
if re.search(pattern, str(shaperec.record[attr])):
found = True
new_geom = shape(shaperec.shape)
# get geoms union
geom = new_geom.union(geom) if geom else new_geom
if not found:
raise ValueError(
f"No match found for the search location '{arg}' "
f"with the pattern '{pattern}'."
)
return geom
def _ecmwf_point_to_lonlat(
point: list[Any], axes: Optional[Union[list[str], str]] = None
) -> tuple[float, float]:
if not isinstance(point, (list, tuple)) or len(point) < 2:
raise TypeError("ECMWF feature point must be a list of at least 2 values")
if axes is None:
lat, lon = point[0], point[1]
else:
axis_list = [axes] if isinstance(axes, str) else list(axes)
axis_list_lower = [str(axis).lower() for axis in axis_list]
lat_index = None
lon_index = None
for name in ("latitude", "lat"):
if name in axis_list_lower:
lat_index = axis_list_lower.index(name)
break
for name in ("longitude", "lon"):
if name in axis_list_lower:
lon_index = axis_list_lower.index(name)
break
if lat_index is None or lon_index is None:
lat, lon = point[0], point[1]
else:
lat = point[lat_index]
lon = point[lon_index]
return float(lon), float(lat)
[docs]
def get_geometry_from_ecmwf_feature(geom: dict[str, Any]) -> Optional[BaseGeometry]:
"""
Creates a ``shapely.geometry`` from an ECMWF Polytope feature.
Supported ECMWF feature types:
- ``polygon``: returns a :class:`~shapely.geometry.Polygon`
- ``boundingbox``: returns a :class:`~shapely.geometry.Polygon` via :func:`~shapely.geometry.box`
- ``position``, ``timeseries``, ``verticalprofile``: return a :class:`~shapely.geometry.Point`
- ``trajectory``: returns a :class:`~shapely.geometry.LineString`
- ``circle``: no direct Shapely equivalent, returns ``None`` and lets default geometry take over
:param geom: ECMWF Polytope feature.
:returns: A Shapely geometry or ``None`` for circle features.
"""
if not isinstance(geom, dict):
raise TypeError("Geometry must be a dictionary")
if "type" not in geom:
raise TypeError("Geometry type must be specified")
from shapely.geometry import LineString, Point, Polygon, box
geom_type = geom["type"].lower()
axes = geom.get("axes")
if geom_type == "polygon":
if "shape" not in geom:
raise TypeError("Missing shape in the geometry")
if not isinstance(geom["shape"], list):
raise TypeError("Geometry shape must be a list")
polygon_shape: list[Any] = geom["shape"]
polygon_args = [_ecmwf_point_to_lonlat(point, axes) for point in polygon_shape]
return Polygon(polygon_args)
if geom_type == "boundingbox":
if "points" not in geom:
raise TypeError("Missing points in the geometry")
if not isinstance(geom["points"], list):
raise TypeError("Geometry points must be a list")
bbox_points = [_ecmwf_point_to_lonlat(point, axes) for point in geom["points"]]
lon_values = [lon for lon, _ in bbox_points]
lat_values = [lat for _, lat in bbox_points]
return box(min(lon_values), min(lat_values), max(lon_values), max(lat_values))
if geom_type in ("position", "timeseries", "verticalprofile"):
if "points" not in geom:
raise TypeError("Missing points in the geometry")
if geom_type == "timeseries" and geom.get("time_axis") in (None, ""):
raise TypeError("Missing time_axis in the geometry")
if not isinstance(geom["points"], list) or not geom["points"]:
raise TypeError("Geometry points must be a non-empty list")
lon, lat = _ecmwf_point_to_lonlat(geom["points"][0], axes)
return Point(lon, lat)
if geom_type == "trajectory":
if "points" not in geom:
raise TypeError("Missing points in the geometry")
if geom.get("inflation") in (None, ""):
raise TypeError("Missing inflation in the geometry")
if not isinstance(geom["points"], list) or len(geom["points"]) < 2:
raise TypeError("Trajectory points must be a list of at least 2 positions")
line_coords = [_ecmwf_point_to_lonlat(point, axes) for point in geom["points"]]
return LineString(line_coords)
if geom_type == "circle":
if "center" not in geom:
raise TypeError("Missing center in the geometry")
if geom.get("radius") in (None, ""):
raise TypeError("Missing radius in the geometry")
return None
raise TypeError(
"Unsupported ECMWF geometry type. "
"Supported types are: polygon, boundingbox, position, timeseries, verticalprofile, trajectory, circle."
)
[docs]
def get_geometry_from_ecmwf_area(
area: Union[str, list[float]]
) -> Optional[BaseGeometry]:
"""
Creates a ``shapely.geometry`` from bounding box in area format.
area format: [max_lat,min_lon,min_lat,max_lon] or "max_lat/min_lon/min_lat/max_lon"
:param area: bounding box in area format.
:returns: A Shapely polygon.
"""
if isinstance(area, str):
area = [float(x) for x in area.split("/")]
if len(area) != 4:
raise ValueError("The area must be a list of 4 values")
max_lat, min_lon, min_lat, max_lon = area
bbox = [min_lon, min_lat, max_lon, max_lat]
return get_geometry_from_various(geometry=bbox)
[docs]
def get_geometry_from_ecmwf_location(
location: dict[str, float],
) -> Optional[BaseGeometry]:
"""
Creates a ``shapely.geometry`` from a single location.
location format: {"latitude": float, "longitude": float}
:param location: dictionary with latitude and longitude
:returns: A Shapely polygon.
"""
if not isinstance(location, dict):
raise TypeError("Location type must be a dictionary")
if not all(f in location for f in ("latitude", "longitude")):
raise ValueError("The location must contains the latitude and the longitude")
lat = location["latitude"]
lon = location["longitude"]
bbox = [lon, lat, lon, lat]
return get_geometry_from_various(geometry=bbox)
[docs]
class MockResponse:
"""Fake requests response"""
def __init__(
self,
json_data: Any = None,
status_code: int = 200,
headers: Optional[dict[str, Any]] = None,
) -> None:
self.json_data = json_data
self.status_code = status_code
self.content = json_data
self.headers = headers
[docs]
def json(self) -> Any:
"""Return json data"""
return self.json_data
def __iter__(self):
yield self
def __enter__(self):
return self
def __exit__(self, exc_type, exc_val, exc_tb):
pass
[docs]
def raise_for_status(self) -> None:
"""raises an exception when the status is not ok"""
if self.status_code != 200:
from requests import HTTPError, Response
response = Response()
response.status_code = self.status_code
raise HTTPError(response=response)
[docs]
def md5sum(file_path: str) -> str:
"""Get file MD5 checksum
>>> import os
>>> md5sum(os.devnull)
'd41d8cd98f00b204e9800998ecf8427e'
:param file_path: input file path
:returns: MD5 checksum
"""
hash_md5 = hashlib.md5()
with open(file_path, "rb") as f:
for chunk in iter(lambda: f.read(4096), b""):
hash_md5.update(chunk)
return hash_md5.hexdigest()
[docs]
def obj_md5sum(data: Any) -> str:
"""Get MD5 checksum from JSON serializable object
>>> obj_md5sum(None)
'37a6259cc0c1dae299a7866489dff0bd'
:param data: JSON serializable input object
:returns: MD5 checksum
"""
return hashlib.md5(orjson.dumps(data, option=orjson.OPT_SORT_KEYS)).hexdigest()
[docs]
@functools.lru_cache()
def cached_parse(str_to_parse: str) -> JSONPath:
"""Cached :func:`jsonpath_ng.ext.parse`
>>> cached_parse.cache_clear()
>>> cached_parse("$.foo")
Child(Root(), Fields('foo'))
>>> cached_parse.cache_info()
CacheInfo(hits=0, misses=1, maxsize=128, currsize=1)
>>> cached_parse("$.foo")
Child(Root(), Fields('foo'))
>>> cached_parse.cache_info()
CacheInfo(hits=1, misses=1, maxsize=128, currsize=1)
>>> cached_parse("$.bar")
Child(Root(), Fields('bar'))
>>> cached_parse.cache_info()
CacheInfo(hits=1, misses=2, maxsize=128, currsize=2)
:param str_to_parse: string to parse as :class:`jsonpath_ng.JSONPath`
:returns: parsed :class:`jsonpath_ng.JSONPath`
"""
from jsonpath_ng.ext import parse
return parse(str_to_parse)
@functools.lru_cache()
def _mutable_cached_yaml_load(config_path: str) -> Any:
import yaml
with open(
os.path.abspath(os.path.realpath(config_path)), mode="r", encoding="utf-8"
) as fh:
return yaml.load(fh, Loader=yaml.SafeLoader)
[docs]
def cached_yaml_load(config_path: str) -> dict[str, Any]:
"""Cached :func:`yaml.load`
:param config_path: path to the yaml configuration file
:returns: loaded yaml configuration
"""
return copy_deepcopy(_mutable_cached_yaml_load(config_path))
@functools.lru_cache()
def _mutable_cached_yaml_load_all(config_path: str) -> list[Any]:
import yaml
with open(config_path, "r") as fh:
return list(yaml.load_all(fh, Loader=yaml.Loader))
[docs]
def cached_yaml_load_all(config_path: str) -> list[Any]:
"""Cached :func:`yaml.load_all`
Load all configurations stored in the configuration file as separated yaml documents
:param config_path: path to the yaml configuration file
:returns: list of configurations
"""
return copy_deepcopy(_mutable_cached_yaml_load_all(config_path))
[docs]
def get_bucket_name_and_prefix(
url: str, bucket_path_level: Optional[int] = None
) -> tuple[Optional[str], Optional[str]]:
"""Extract bucket name and prefix from URL
:param url: (optional) URL to use as product.location
:param bucket_path_level: (optional) bucket location index in path.split('/')
:returns: bucket_name and prefix as str
"""
bucket, prefix = None, None
scheme, netloc, path, _, _, _ = urlparse(url)
subdomain = netloc.split(".")[0]
path = path.strip("/")
if "/" in path and scheme and subdomain == "s3" and bucket_path_level is None:
bucket, prefix = path.split("/", 1)
elif scheme and bucket_path_level is None:
bucket = subdomain
prefix = path
elif not scheme and bucket_path_level is None:
prefix = path
elif bucket_path_level is not None:
parts = path.split("/")
bucket, prefix = (
parts[bucket_path_level],
"/".join(parts[(bucket_path_level + 1) :]),
)
return bucket, prefix
[docs]
def flatten_top_directories(
nested_dir_root: str, common_subdirs_path: Optional[str] = None
) -> None:
"""Flatten directory structure, removing common empty sub-directories
:param nested_dir_root: Absolute path of the directory structure to flatten
:param common_subdirs_path: (optional) Absolute path of the desired subdirectory to remove
"""
if not common_subdirs_path:
subpaths_list = [p for p in Path(nested_dir_root).glob("**/*") if p.is_file()]
common_subdirs_path = os.path.commonpath(subpaths_list)
if Path(common_subdirs_path).is_file():
common_subdirs_path = os.path.dirname(common_subdirs_path)
if nested_dir_root != common_subdirs_path:
logger.debug(f"Flatten {common_subdirs_path} to {nested_dir_root}")
tmp_path = mkdtemp()
shutil.copytree(common_subdirs_path, tmp_path, dirs_exist_ok=True)
shutil.rmtree(nested_dir_root)
shutil.move(tmp_path, nested_dir_root)
[docs]
def deepcopy(sth: Any) -> Any:
"""Customized and faster deepcopy inspired by https://stackoverflow.com/a/45858907
``_copy_list`` and ``_copy_dict`` dispatchers available for the moment
:param sth: Object to copy
:returns: Copied object
"""
_dispatcher: dict[type[Any], Callable[..., Any]] = {}
def _copy_list(
input_list: list[Any], dispatch: dict[type[Any], Callable[..., Any]]
):
ret = input_list.copy()
for idx, item in enumerate(ret):
cp = dispatch.get(type(item))
if cp is not None:
ret[idx] = cp(item, dispatch)
return ret
def _copy_dict(
input_dict: dict[Any, Any], dispatch: dict[type[Any], Callable[..., Any]]
):
ret = input_dict.copy()
for key, value in ret.items():
cp = dispatch.get(type(value))
if cp is not None:
ret[key] = cp(value, dispatch)
return ret
_dispatcher[list] = _copy_list
_dispatcher[dict] = _copy_dict
cp = _dispatcher.get(type(sth))
if cp is None:
return sth
else:
return cp(sth, _dispatcher)
[docs]
def cast_scalar_value(value: Any, new_type: Any) -> Any:
"""Convert a scalar (not nested) value type to the given one
>>> cast_scalar_value('1', int)
1
>>> cast_scalar_value(1, str)
'1'
>>> cast_scalar_value('false', bool)
False
:param value: the scalar value to convert
:param new_type: the wanted type
:returns: scalar ``value`` converted to ``new_type``
"""
if isinstance(value, str) and new_type is bool:
# Bool is a type with special meaning in Python, thus the special
# case
if value.capitalize() not in ("True", "False"):
raise ValueError(
"Only true or false strings (case insensitive) are allowed for booleans"
)
# Get the real Python value of the boolean. e.g: value='tRuE'
# => eval(value.capitalize())=True.
# str.capitalize() transforms the first character of the string
# to a capital letter
return eval(value.capitalize())
return new_type(value)
[docs]
def guess_file_type(file: str) -> str:
"""Guess the mime type of a file or URL based on its extension,
using eodag extended mimetypes definition
>>> guess_file_type('foo.tiff')
'image/tiff'
>>> guess_file_type('foo.grib')
'application/x-grib'
:param file: file url or path
:returns: guessed mime type
"""
mime_type, _ = mimetypes.guess_type(file, False)
if mime_type is None:
mime_type = DEFAULT_MIME
elif mime_type == "text/xml":
return "application/xml"
return mime_type
[docs]
def guess_extension(type: str) -> Optional[str]:
"""Guess extension from mime type, using eodag extended mimetypes definition
>>> guess_extension('image/tiff')
'.tiff'
>>> guess_extension('application/x-grib')
'.grib'
:param type: mime type
:returns: guessed file extension
"""
return mimetypes.guess_extension(type, strict=False)
[docs]
@functools.lru_cache(maxsize=2)
def get_ssl_context(ssl_verify: bool) -> ssl.SSLContext:
"""
Returns an SSL context based on ``ssl_verify`` argument.
:param ssl_verify: :attr:`~eodag.config.PluginConfig.ssl_verify` parameter
:returns: An SSL context object.
"""
ctx = ssl.create_default_context()
if not ssl_verify:
ctx.check_hostname = False
ctx.verify_mode = ssl.CERT_NONE
else:
ctx.check_hostname = True
ctx.verify_mode = ssl.CERT_REQUIRED
return ctx
[docs]
def sort_dict(input_dict: dict[str, Any]) -> dict[str, Any]:
"""
Recursively sorts a dict by keys.
:param input_dict: input dict
:returns: sorted dict
>>> sort_dict({"b": {"c": 1, "a": 2, "b": 3}, "a": 4})
{'a': 4, 'b': {'a': 2, 'b': 3, 'c': 1}}
"""
return {
k: sort_dict(v) if isinstance(v, dict) else v
for k, v in sorted(input_dict.items())
}
[docs]
def dict_md5sum(input_dict: dict[str, Any]) -> str:
"""
Hash nested dictionary
:param input_dict: input dict
:returns: hash
>>> hd = dict_md5sum({"b": {"c": 1, "a": 2, "b": 3}, "a": 4})
>>> hd
'a195bcef1bb3b419e9e74b7cc5db8098'
>>> assert(dict_md5sum({"a": 4, "b": {"b": 3, "c": 1, "a": 2}}) == hd)
"""
return obj_md5sum(sort_dict(input_dict))
[docs]
def remove_str_array_quotes(input_str: str) -> str:
"""Remove quotes around arrays to avoid json parsing errors
:param input_str: string to format
:returns: string without quotes surrounding array brackets
>>> remove_str_array_quotes('"a":"["a", "b"]"')
'"a":["a", "b"]'
>>> remove_str_array_quotes('{"a":"["a", "b"]", "b": ["c", "d"]}')
'{"a":["a", "b"], "b": ["c", "d"]}'
"""
output_str = ""
for i in range(0, len(input_str)):
if i < len(input_str) - 1 and input_str[i] == '"' and input_str[i + 1] == "[":
continue
if input_str[i] == '"' and input_str[i - 1] == "]":
continue
output_str += input_str[i]
return output_str
[docs]
def parse_le_uint32(data: bytes) -> int:
"""
Parse little-endian unsigned 4-byte integer.
>>> parse_le_uint32(b'\\x01\\x00\\x00\\x00')
1
>>> parse_le_uint32(b'\\xff\\xff\\xff\\xff')
4294967295
"""
return struct.unpack("<I", data)[0]
[docs]
def parse_le_uint16(data: bytes) -> int:
"""
Parse little-endian unsigned 2-byte integer.
>>> parse_le_uint16(b'\\x01\\x00')
1
>>> parse_le_uint16(b'\\xff\\xff')
65535
"""
return struct.unpack("<H", data)[0]
[docs]
def get_collection_dates(
collection_dict: dict[str, Any],
) -> tuple[Optional[str], Optional[str]]:
"""Extract mission start and end dates from collection configuration.
Extracts dates from the extent.temporal.interval structure.
:param collection_dict: Collection configuration dictionary
:returns: Tuple of (mission_start_date, mission_end_date) as ISO strings or None
Example:
>>> get_collection_dates({
... "extent": {"temporal": {"interval": [["2017-10-13T00:00:00Z", "2023-12-31T23:59:59Z"]]}}
... })
('2017-10-13T00:00:00Z', '2023-12-31T23:59:59Z')
>>> get_collection_dates({
... "extent": {"temporal": {"interval": [["2017-10-13T00:00:00Z", None]]}}
... })
('2017-10-13T00:00:00Z', None)
>>> get_collection_dates({})
(None, None)
"""
extent_interval = (
collection_dict.get("extent", {})
.get("temporal", {})
.get("interval", [[None, None]])
)
if not extent_interval or len(extent_interval) == 0:
return None, None
mission_start = (
extent_interval[0][0]
if len(extent_interval) > 0 and len(extent_interval[0]) > 0
else None
)
mission_end = (
extent_interval[0][1]
if len(extent_interval) > 0 and len(extent_interval[0]) > 1
else None
)
return mission_start, mission_end
__all__ = [
"GENERIC_COLLECTION",
"GENERIC_STAC_PROVIDER",
"STAC_SEARCH_PLUGINS",
"STAC_VERSION",
"USER_AGENT",
"HTTP_REQ_TIMEOUT",
"DEFAULT_SEARCH_TIMEOUT",
"DEFAULT_STREAM_REQUESTS_TIMEOUT",
"REQ_RETRY_TOTAL",
"REQ_RETRY_BACKOFF_FACTOR",
"REQ_RETRY_STATUS_FORCELIST",
"DEFAULT_DOWNLOAD_WAIT",
"DEFAULT_DOWNLOAD_TIMEOUT",
"DEFAULT_TOKEN_EXPIRATION_MARGIN",
"DEFAULT_PAGE",
"DEFAULT_LIMIT",
"DEFAULT_MAX_LIMIT",
"KNOWN_NEXT_PAGE_TOKEN_KEYS",
"DEFAULT_PROJ",
"DEFAULT_MISSION_START_DATE",
"DEFAULT_SHAPELY_GEOMETRY",
"ONLINE_STATUS",
"JSONPATH_MATCH",
"WORKABLE_JSONPATH_MATCH",
"ARRAY_FIELD_MATCH",
"FloatRange",
"StreamResponse",
"StreamResponseContent",
"DownloadedCallback",
"ProgressCallback",
"MockResponse",
"Unpack",
"_deprecated",
"slugify",
"sanitize",
"strip_accents",
"uri_to_path",
"path_to_uri",
"mutate_dict_in_place",
"merge_mappings",
"maybe_generator",
"repeatfunc",
"makedirs",
"rename_subfolder",
"rename_with_version",
"format_dict_items",
"jsonpath_parse_dict_items",
"update_nested_dict",
"items_recursive_apply",
"dict_items_recursive_apply",
"list_items_recursive_apply",
"items_recursive_sort",
"dict_items_recursive_sort",
"list_items_recursive_sort",
"string_to_jsonpath",
"format_string",
"parse_jsonpath",
"nested_pairs2dict",
"get_geometry_from_various",
"get_geometry_from_ecmwf_feature",
"get_geometry_from_ecmwf_area",
"get_geometry_from_ecmwf_location",
"md5sum",
"obj_md5sum",
"cached_parse",
"cached_yaml_load",
"cached_yaml_load_all",
"get_bucket_name_and_prefix",
"flatten_top_directories",
"deepcopy",
"parse_header",
"cast_scalar_value",
"guess_file_type",
"guess_extension",
"get_ssl_context",
"sort_dict",
"dict_md5sum",
"remove_str_array_quotes",
"parse_le_uint32",
"parse_le_uint16",
"format_pydantic_error",
"get_collection_dates",
"eodag_logging",
]
