forked from STI/Astroport.ONE
67 lines
1.4 KiB
Python
67 lines
1.4 KiB
Python
# SPDX-License-Identifier: MIT
|
|
|
|
"""
|
|
Commonly useful filters for `attr.asdict`.
|
|
"""
|
|
|
|
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 isinstance(cls, type)),
|
|
frozenset(cls for cls in what if isinstance(cls, str)),
|
|
frozenset(cls for cls in what if isinstance(cls, Attribute)),
|
|
)
|
|
|
|
|
|
def include(*what):
|
|
"""
|
|
Include *what*.
|
|
|
|
:param what: What to include.
|
|
:type what: `list` of classes `type`, field names `str` or
|
|
`attrs.Attribute`\\ s
|
|
|
|
:rtype: `callable`
|
|
|
|
.. versionchanged:: 23.1.0 Accept strings with field names.
|
|
"""
|
|
cls, names, attrs = _split_what(what)
|
|
|
|
def include_(attribute, value):
|
|
return (
|
|
value.__class__ in cls
|
|
or attribute.name in names
|
|
or attribute in attrs
|
|
)
|
|
|
|
return include_
|
|
|
|
|
|
def exclude(*what):
|
|
"""
|
|
Exclude *what*.
|
|
|
|
:param what: What to exclude.
|
|
:type what: `list` of classes `type`, field names `str` or
|
|
`attrs.Attribute`\\ s.
|
|
|
|
:rtype: `callable`
|
|
|
|
.. versionchanged:: 23.3.0 Accept field name string as input argument
|
|
"""
|
|
cls, names, attrs = _split_what(what)
|
|
|
|
def exclude_(attribute, value):
|
|
return not (
|
|
value.__class__ in cls
|
|
or attribute.name in names
|
|
or attribute in attrs
|
|
)
|
|
|
|
return exclude_
|