Source code for mapsbr.helpers.utils

import numpy as np


@np.vectorize
def vectorized_get(dictionary, key):
    """
    Helper vectorized function to get keys
    from a dictionary.
    """
    return dictionary.get(key, -1)


def is_iter(something):
    """
    Helper vectorized function to test
    if something is an iterable other
    than a str.
    """
    try:
        iter(something)
        return True and not isinstance(something, str)
    except TypeError:
        return False


@np.vectorize
def is_number(something):
    """
    Helper vectorized function to test
    if a value is or represents a number.
    """
    try:
        int(something)
        return True
    except ValueError:
        return False


def get_features(geojson):
    """
    Helper function to extract features
    from dictionary. If it doesn't find
    it, raise a value error with a more
    informative error message.
    """
    try:
        features = geojson["features"]
    except KeyError:
        raise KeyError(f"{geojson} is an invalid GeoJSON. Not a feature collection")
    return features


@np.vectorize
def from_iso88591_to_utf8(string):
    """
    Convert weird characters to UTF-8.
    For example, "médio" should be "médio".
    """
    try:
        return bytes(string, "iso-8859-1").decode("utf-8")
    except UnicodeDecodeError:
        return string