# =================================================================
#
# Authors: Tom Kralidis <tomkralidis@gmail.com>
#
# Copyright (c) 2022 Tom Kralidis
#
# Permission is hereby granted, free of charge, to any person
# obtaining a copy of this software and associated documentation
# files (the "Software"), to deal in the Software without
# restriction, including without limitation the rights to use,
# copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the
# Software is furnished to do so, subject to the following
# conditions:
#
# The above copyright notice and this permission notice shall be
# included in all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
# EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES
# OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
# NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
# HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
# WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
# FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
# OTHER DEALINGS IN THE SOFTWARE.
#
# =================================================================
"""Generic util functions used in the code"""
import base64
from typing import List
from datetime import date, datetime, time
from decimal import Decimal
from enum import Enum
import json
import logging
import mimetypes
import os
from pathlib import Path
import re
from typing import Any, IO, Union
from urllib.request import urlopen
from urllib.parse import urlparse
from shapely.geometry import Polygon
import dateutil.parser
from jinja2 import Environment, FileSystemLoader, select_autoescape
from babel.support import Translations
from jinja2.exceptions import TemplateNotFound
import yaml
from pygeoapi import __version__
from pygeoapi import l10n
from pygeoapi.provider.base import ProviderTypeError
LOGGER = logging.getLogger(__name__)
DATETIME_FORMAT = '%Y-%m-%dT%H:%M:%S.%fZ'
TEMPLATES = Path(__file__).parent.resolve() / 'templates'
mimetypes.add_type('text/plain', '.yaml')
mimetypes.add_type('text/plain', '.yml')
[docs]def dategetter(date_property: str, collection: dict) -> str:
"""
Attempts to obtain a date value from a collection.
:param date_property: property representing the date
:param collection: dictionary to check within
:returns: `str` (ISO8601) representing the date (allowing
for an open interval using null)
"""
value = collection.get(date_property)
if value is None:
return None
return value.isoformat()
[docs]def get_typed_value(value: str) -> Union[float, int, str]:
"""
Derive true type from data value
:param value: value
:returns: value as a native Python data type
"""
try:
if '.' in value: # float?
value2 = float(value)
elif len(value) > 1 and value.startswith('0'):
value2 = value
else: # int?
value2 = int(value)
except ValueError: # string (default)?
value2 = value
return value2
[docs]def yaml_load(fh: IO) -> dict:
"""
serializes a YAML files into a pyyaml object
:param fh: file handle
:returns: `dict` representation of YAML
"""
# support environment variables in config
# https://stackoverflow.com/a/55301129
path_matcher = re.compile(r'.*\$\{([^}^{]+)\}.*')
def path_constructor(loader, node):
env_var = path_matcher.match(node.value).group(1)
if env_var not in os.environ:
msg = f'Undefined environment variable {env_var} in config'
raise EnvironmentError(msg)
return get_typed_value(os.path.expandvars(node.value))
class EnvVarLoader(yaml.SafeLoader):
pass
EnvVarLoader.add_implicit_resolver('!path', path_matcher, None)
EnvVarLoader.add_constructor('!path', path_constructor)
return yaml.load(fh, Loader=EnvVarLoader)
[docs]def str2bool(value: Union[bool, str]) -> bool:
"""
helper function to return Python boolean
type (source: https://stackoverflow.com/a/715468)
:param value: value to be evaluated
:returns: `bool` of whether the value is boolean-ish
"""
value2 = False
if isinstance(value, bool):
value2 = value
else:
value2 = value.lower() in ('yes', 'true', 't', '1', 'on')
return value2
[docs]def to_json(dict_: dict, pretty: bool = False) -> str:
"""
Serialize dict to json
:param dict_: `dict` of JSON representation
:param pretty: `bool` of whether to prettify JSON (default is `False`)
:returns: JSON string representation
"""
if pretty:
indent = 4
else:
indent = None
return json.dumps(dict_, default=json_serial,
indent=indent)
[docs]def file_modified_iso8601(filepath: Path) -> str:
"""
Provide a file's ctime in ISO8601
:param filepath: path to file
:returns: string of ISO8601
"""
return datetime.fromtimestamp(
os.path.getctime(filepath)).strftime('%Y-%m-%dT%H:%M:%SZ')
[docs]def human_size(nbytes: int) -> str:
"""
Provides human readable file size
source: https://stackoverflow.com/a/14996816
:param nbytes: int of file size (bytes)
:param units: list of unit abbreviations
:returns: string of human readable filesize
"""
suffixes = ['B', 'K', 'M', 'G', 'T', 'P']
i = 0
while nbytes >= 1024 and i < len(suffixes)-1:
nbytes /= 1024.
i += 1
if suffixes[i] == 'K':
f = str(int(nbytes)).rstrip('0').rstrip('.')
elif suffixes[i] == 'B':
return nbytes
else:
f = f'{nbytes:.1f}'.rstrip('0').rstrip('.')
return f'{f}{suffixes[i]}'
[docs]def get_path_basename(urlpath: str) -> str:
"""
Helper function to derive file basename
:param urlpath: URL path
:returns: string of basename of URL path
"""
return Path(urlpath).name
[docs]def json_serial(obj: Any) -> str:
"""
helper function to convert to JSON non-default
types (source: https://stackoverflow.com/a/22238613)
:param obj: `object` to be evaluated
:returns: JSON non-default type to `str`
"""
if isinstance(obj, (datetime, date, time)):
return obj.isoformat()
elif isinstance(obj, bytes):
try:
LOGGER.debug('Returning as UTF-8 decoded bytes')
return obj.decode('utf-8')
except UnicodeDecodeError:
LOGGER.debug('Returning as base64 encoded JSON object')
return base64.b64encode(obj)
elif isinstance(obj, Decimal):
return float(obj)
elif type(obj).__name__ == 'int64':
return int(obj)
elif type(obj).__name__ == 'float64':
return float(obj)
elif isinstance(obj, l10n.Locale):
return l10n.locale2str(obj)
msg = f'{obj} type {type(obj)} not serializable'
LOGGER.error(msg)
raise TypeError(msg)
[docs]def is_url(urlstring: str) -> bool:
"""
Validation function that determines whether a candidate URL should be
considered a URI. No remote resource is obtained; this does not check
the existence of any remote resource.
:param urlstring: `str` to be evaluated as candidate URL.
:returns: `bool` of whether the URL looks like a URL.
"""
try:
result = urlparse(urlstring)
return bool(result.scheme and result.netloc)
except ValueError:
return False
[docs]def render_j2_template(config: dict, template: Path,
data: dict, locale_: str = None) -> str:
"""
render Jinja2 template
:param config: dict of configuration
:param template: template (relative path)
:param data: dict of data
:param locale_: the requested output Locale
:returns: string of rendered template
"""
custom_templates = False
try:
templates_path = config['server']['templates']['path']
env = Environment(loader=FileSystemLoader(templates_path),
extensions=['jinja2.ext.i18n',
'jinja2.ext.autoescape'],
autoescape=select_autoescape(['html', 'xml']))
custom_templates = True
LOGGER.debug(f'using custom templates: {templates_path}')
except (KeyError, TypeError):
env = Environment(loader=FileSystemLoader(TEMPLATES),
extensions=['jinja2.ext.i18n',
'jinja2.ext.autoescape'],
autoescape=select_autoescape(['html', 'xml']))
LOGGER.debug(f'using default templates: {TEMPLATES}')
env.filters['to_json'] = to_json
env.filters['format_datetime'] = format_datetime
env.filters['format_duration'] = format_duration
env.filters['human_size'] = human_size
env.globals.update(to_json=to_json)
env.filters['get_path_basename'] = get_path_basename
env.globals.update(get_path_basename=get_path_basename)
env.filters['get_breadcrumbs'] = get_breadcrumbs
env.globals.update(get_breadcrumbs=get_breadcrumbs)
env.filters['filter_dict_by_key_value'] = filter_dict_by_key_value
env.globals.update(filter_dict_by_key_value=filter_dict_by_key_value)
translations = Translations.load('locale', [locale_])
env.install_gettext_translations(translations)
try:
template = env.get_template(template)
except TemplateNotFound as err:
if custom_templates:
LOGGER.debug(err)
LOGGER.debug('Custom template not found; using default')
env = Environment(loader=FileSystemLoader(TEMPLATES),
extensions=['jinja2.ext.i18n'])
env.install_gettext_translations(translations)
template = env.get_template(template)
else:
raise
return template.render(config=l10n.translate_struct(config, locale_, True),
data=data, locale=locale_, version=__version__)
[docs]def get_mimetype(filename: str) -> str:
"""
helper function to return MIME type of a given file
:param filename: filename (with extension)
:returns: MIME type of given filename
"""
return mimetypes.guess_type(filename)[0]
[docs]def get_breadcrumbs(urlpath: str) -> list:
"""
helper function to make breadcrumbs from a URL path
:param urlpath: URL path
:returns: `list` of `dict` objects of labels and links
"""
links = []
tokens = urlpath.split('/')
s = ''
for t in tokens:
if s:
s += '/' + t
else:
s = t
links.append({
'href': s,
'title': t,
})
return links
[docs]def filter_dict_by_key_value(dict_: dict, key: str, value: str) -> dict:
"""
helper function to filter a dict by a dict key
:param dict_: ``dict``
:param key: dict key
:param value: dict key value
:returns: filtered ``dict``
"""
return {k: v for (k, v) in dict_.items() if v[key] == value}
[docs]def filter_providers_by_type(providers: list, type: str) -> dict:
"""
helper function to filter a list of providers by type
:param providers: ``list``
:param type: str
:returns: filtered ``dict`` provider
"""
providers_ = {provider['type']: provider for provider in providers}
return providers_.get(type)
[docs]def get_provider_by_type(providers: list, provider_type: str) -> dict:
"""
helper function to load a provider by a provider type
:param providers: ``list`` of providers
:param provider_type: type of provider (e.g. feature)
:returns: provider based on type
"""
LOGGER.debug(f'Searching for provider type {provider_type}')
try:
p = (next(d for i, d in enumerate(providers)
if d['type'] == provider_type))
except (RuntimeError, StopIteration):
raise ProviderTypeError('Invalid provider type requested')
return p
[docs]def get_provider_default(providers: list) -> dict:
"""
helper function to get a resource's default provider
:param providers: ``list`` of providers
:returns: filtered ``dict``
"""
try:
default = (next(d for i, d in enumerate(providers) if 'default' in d
and d['default']))
LOGGER.debug('found default provider type')
except StopIteration:
LOGGER.debug('no default provider type. Returning first provider')
default = providers[0]
LOGGER.debug(f"Default provider: {default['type']}")
return default
[docs]class JobStatus(Enum):
"""
Enum for the job status options specified in the WPS 2.0 specification
"""
# From the specification
accepted = 'accepted'
running = 'running'
successful = 'successful'
failed = 'failed'
dismissed = 'dismissed'
[docs]def read_data(path: Union[Path, str]) -> Union[bytes, str]:
"""
helper function to read data (file or networrk)
"""
LOGGER.debug(f'Attempting to read {path}')
if isinstance(path, Path) or not path.startswith(('http', 's3')):
LOGGER.debug('local file on disk')
with Path(path).open('rb') as fh:
return fh.read()
else:
LOGGER.debug('network file')
with urlopen(path) as r:
return r.read()
[docs]def url_join(*parts: list) -> str:
"""
helper function to join a URL from a number of parts/fragments.
Implemented because urllib.parse.urljoin strips subpaths from
host urls if they are specified
Per https://github.com/geopython/pygeoapi/issues/695
:param parts: list of parts to join
:returns: str of resulting URL
"""
return '/'.join([p.strip().strip('/') for p in parts])
[docs]def get_envelope(coords_list: List[List[float]]) -> list:
"""
helper function to get the envelope for a given coordinates
list through the Shapely API.
:param coords_list: list of coordinates
:returns: list of the envelope's coordinates
"""
coords = [tuple(item) for item in coords_list]
polygon = Polygon(coords)
bounds = polygon.bounds
return [[bounds[0], bounds[3]],
[bounds[2], bounds[1]]]