Source code for obspy.core.util.attribdict

# -*- coding: utf-8 -*-
AttribDict class for ObsPy.

    The ObsPy Development Team (
    GNU Lesser General Public License, Version 3
import copy
import warnings

import numpy as np

from .. import compatibility

[docs]def _attribdict_equal(v1, v2, depth=5): """ Robust comparison of possibly nested AttribDict entries or AttribDicts """ if depth == 0: return False elif isinstance(v1, AttribDict) and isinstance(v2, AttribDict): keys = set(v1.keys()) | set(v2.keys()) return all(_attribdict_equal(v1.get(k), v2.get(k), depth=depth-1) for k in keys) elif isinstance(v1, AttribDict) or isinstance(v2, AttribDict): return False elif isinstance(v1, np.ndarray) and isinstance(v2, np.ndarray): return np.shape(v1) == np.shape(v2) and np.all(v1 == v2) elif isinstance(v1, np.ndarray) or isinstance(v2, np.ndarray): return False else: try: return v1 == v2 except Exception: return False
[docs]class AttribDict(compatibility.collections_abc.MutableMapping): """ A class which behaves like a dictionary. :type data: dict, optional :param data: Dictionary with initial keywords. .. rubric:: Basic Usage You may use the following syntax to change or access data in this class. >>> stats = AttribDict() >>> = 'BW' >>> stats['station'] = 'ROTZ' >>> print(stats.get('network')) BW >>> print(stats['network']) BW >>> print(stats.station) ROTZ >>> x = stats.keys() >>> x = sorted(x) >>> print(x[0], x[1]) network station """ defaults = {} readonly = [] warn_on_non_default_key = False do_not_warn_on = [] _types = {}
[docs] def __init__(self, *args, **kwargs): """ An AttribDict can be initialized in two ways. It can be given an existing dictionary as a simple argument or alternatively all keyword arguments will become (key, value) pairs. >>> attrib_dict_1 = AttribDict({"a":1, "b":2}) >>> attrib_dict_2 = AttribDict(a=1, b=2) >>> attrib_dict_1 #doctest: +SKIP AttribDict({'a': 1, 'b': 2}) >>> assert(attrib_dict_1 == attrib_dict_2) """ # set default values directly #: Calling the Subclassed dict update method self.__dict__.update(self.defaults) # use overwritable update method to set arguments #: Subclassed update method self.update(dict(*args, **kwargs))
[docs] def __repr__(self): return "%s(%s)" % (self.__class__.__name__, self.__dict__)
[docs] def __getitem__(self, name, default=None): try: return self.__dict__[name] except KeyError: # check if we got any default value given at class level if name in self.defaults: return self.defaults[name] # if both are missing check for a given default value if default is None: raise return default
[docs] def __setitem__(self, key, value): if key in self.readonly: msg = 'Attribute "%s" in %s object is read only!' raise AttributeError(msg % (key, self.__class__.__name__)) if self.warn_on_non_default_key and key not in self.defaults: # issue warning if not a default key # (and not in the list of exceptions) if key in self.do_not_warn_on: pass else: msg = ('Setting attribute "{}" which is not a default ' 'attribute ("{}").').format( key, '", "'.join(self.defaults.keys())) warnings.warn(msg) # Type checking/warnings if key in self._types and not isinstance(value, self._types[key]): value = self._cast_type(key, value) mapping_instance = isinstance(value, compatibility.collections_abc.Mapping) attr_dict_instance = isinstance(value, AttribDict) if mapping_instance and not attr_dict_instance: self.__dict__[key] = AttribDict(value) else: self.__dict__[key] = value
[docs] def __delitem__(self, name): del self.__dict__[name]
[docs] def __getattr__(self, name, default=None): """ Py3k hasattr() expects an AttributeError no KeyError to be raised if the attribute is not found. """ try: return self.__getitem__(name, default) except KeyError as e: raise AttributeError(e.args[0])
__setattr__ = __setitem__ __delattr__ = __delitem__
[docs] def copy(self): return copy.deepcopy(self)
[docs] def update(self, adict={}): for (key, value) in adict.items(): if key in self.readonly: continue self.__setitem__(key, value)
[docs] def _pretty_str(self, priorized_keys=[], min_label_length=16): """ Return better readable string representation of AttribDict object. :type priorized_keys: list[str], optional :param priorized_keys: Keywords of current AttribDict which will be shown before all other keywords. Those keywords must exists otherwise an exception will be raised. Defaults to empty list. :type min_label_length: int, optional :param min_label_length: Minimum label length for keywords. Defaults to ``16``. :return: String representation of current AttribDict object. """ keys = list(self.keys()) # determine longest key name for alignment of all items try: i = max(max([len(k) for k in keys]), min_label_length) except ValueError: # no keys return "" pattern = "%%%ds: %%s" % (i) # check if keys exist other_keys = [k for k in keys if k not in priorized_keys] # priorized keys first + all other keys keys = priorized_keys + sorted(other_keys) head = [pattern % (k, self.__dict__[k]) for k in keys] return "\n".join(head)
[docs] def _cast_type(self, key, value): """ Cast type of value to type required in _types dict. :type key: str :param key: The key from __setattr__. :param value: The value being set to key. :return: value cast to correct type. """ typ = self._types[key] new_type = ( typ[0] if isinstance(typ, compatibility.collections_abc.Sequence) else typ) msg = ('Attribute "%s" must be of type %s, not %s. Attempting to ' 'cast %s to %s') % (key, typ, type(value), value, new_type) warnings.warn(msg) return new_type(value)
[docs] def __iter__(self): return iter(self.__dict__)
[docs] def __len__(self): return len(self.__dict__)
if __name__ == '__main__': import doctest doctest.testmod(exclude_empty=True)