Mercurial > hg
view mercurial/thirdparty/attr/filters.py @ 46413:ed43b6fa847e stable
copyright: update to 2021
Differential Revision: https://phab.mercurial-scm.org/D9885
author | Matt Harbison <matt_harbison@yahoo.com> |
---|---|
date | Wed, 27 Jan 2021 00:54:57 -0500 |
parents | a5493a251ad3 |
children | e1c586b9a43c |
line wrap: on
line source
""" Commonly useful filters for :func:`attr.asdict`. """ from __future__ import absolute_import, division, print_function from ._compat import isclass from ._make import Attribute def _split_what(what): """ Returns a tuple of `frozenset`s of classes and attributes. """ return ( frozenset(cls for cls in what if isclass(cls)), frozenset(cls for cls in what if isinstance(cls, Attribute)), ) def include(*what): r""" Whitelist *what*. :param what: What to whitelist. :type what: :class:`list` of :class:`type` or :class:`attr.Attribute`\ s :rtype: :class:`callable` """ cls, attrs = _split_what(what) def include_(attribute, value): return value.__class__ in cls or attribute in attrs return include_ def exclude(*what): r""" Blacklist *what*. :param what: What to blacklist. :type what: :class:`list` of classes or :class:`attr.Attribute`\ s. :rtype: :class:`callable` """ cls, attrs = _split_what(what) def exclude_(attribute, value): return value.__class__ not in cls and attribute not in attrs return exclude_