Skip to content

utils

Common utility functions.

These functions may be used in general throughout the OPTIMADE Gateway Python code.

clean_python_types(data) async

Turn any types into MongoDB-friendly Python types.

Use dict() method for Pydantic models. Use value property for Enums. Turn tuples and sets into lists.

Source code in optimade_gateway/common/utils.py 
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
async def clean_python_types(data: "Any") -> "Any":
    """Turn any types into MongoDB-friendly Python types.

    Use `dict()` method for Pydantic models.
    Use `value` property for Enums.
    Turn tuples and sets into lists.
    """
    res: "Any" = None
    if isinstance(data, (list, tuple, set)):
        res = []
        for datum in data:
            res.append(await clean_python_types(datum))
    elif isinstance(data, dict):
        res = {}
        for key in list(data.keys()):
            res[key] = await clean_python_types(data[key])
    elif isinstance(data, BaseModel):
        # Pydantic model
        res = await clean_python_types(data.dict())
    elif isinstance(data, Enum):
        res = await clean_python_types(data.value)
    elif isinstance(data, type):
        res = await clean_python_types(f"{data.__module__}.{data.__name__}")
    else:
        # Unknown or other basic type, e.g., str, int, etc.
        res = data
    return res

get_resource_attribute(resource, field, default=None, disambiguate=True)

Return a resource's field's value

Get the field value no matter if the resource is a pydantic model or a Python dictionary.

Determine ambiguous field values and return them if desired (disambiguate). For example, if "attributes.base_url" is requested for a LinksResource it can be either a string, a Link model or a dictionary resembling the Link model.

Parameters:

Name Type Description Default
resource Union[BaseModel, Dict[str, Any], None]

The resource, from which to get the field value.

 required
field str

The resource field. This can be a dot-separated nested field, e.g., "attributes.base_url".

 required
default Any

The default value to return if field does not exist.

 None
disambiguate bool

Whether or not to "shortcut" a field value. For example, for attributes.base_url, if True, this would return the string value or the value of it's "href" key.

 True

Returns:

Type Description
Any

The resource's field's value.
Source code in optimade_gateway/common/utils.py 
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
def get_resource_attribute(
    resource: "Union[BaseModel, Dict[str, Any], None]",
    field: str,
    default: "Any" = None,
    disambiguate: bool = True,
) -> "Any":
    """Return a resource's field's value

    Get the field value no matter if the resource is a pydantic model or a Python dictionary.

    Determine ambiguous field values and return them if desired (`disambiguate`).
    For example, if
    [`"attributes.base_url"`](https://www.optimade.org/optimade-python-tools/api_reference/models/links/#optimade.models.links.LinksResourceAttributes.base_url)
    is requested for a
    [`LinksResource`](https://www.optimade.org/optimade-python-tools/api_reference/models/links/#optimade.models.links.LinksResource)
    it can be either a string, a
    [`Link`](https://www.optimade.org/optimade-python-tools/api_reference/models/jsonapi/#optimade.models.jsonapi.Link)
    model or a dictionary resembling the `Link` model.

    Parameters:
        resource: The resource, from which to get the field value.
        field: The resource field. This can be a dot-separated nested field, e.g.,
            `"attributes.base_url"`.
        default: The default value to return if `field` does not exist.
        disambiguate: Whether or not to "shortcut" a field value.
            For example, for `attributes.base_url`, if `True`, this would return the
            string value or the value of it's `"href"` key.

    Returns:
        The resource's field's value.

    """
    if isinstance(resource, BaseModel):
        _get_attr = getattr
    elif isinstance(resource, dict):

        def _get_attr(mapping: dict, key: str, default: "Any") -> "Any":  # type: ignore[misc]
            return mapping.get(key, default)

    elif resource is None:
        # Allow passing `None`, but simply return `default`
        return default
    else:
        raise TypeError(
            "resource must be either a pydantic model or a Python dictionary, it was of "
            f"type {type(resource)!r}"
        )

    fields = field.split(".")
    for _ in fields[:-1]:
        resource = _get_attr(resource, _, {})
    field = fields[-1]
    value = _get_attr(resource, field, default)

    if disambiguate:
        if field in ("base_url", "next", "prev", "last", "first"):
            if not isinstance(value, str):
                value = _get_attr(value, "href", default)

    return value