You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
68 lines
2.2 KiB
68 lines
2.2 KiB
# -*- coding: utf-8 -*-
|
|
# Copyright 2009-2021 Joshua Bronson. All Rights Reserved.
|
|
#
|
|
# This Source Code Form is subject to the terms of the Mozilla Public
|
|
# License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
# file, You can obtain one at http://mozilla.org/MPL/2.0/.
|
|
|
|
|
|
"""Functions for iterating over items in a mapping."""
|
|
|
|
import typing as _t
|
|
from collections.abc import Mapping
|
|
from itertools import chain
|
|
|
|
from ._typing import KT, VT, IterItems, MapOrIterItems
|
|
|
|
|
|
_NULL_IT: IterItems = iter(())
|
|
|
|
|
|
def _iteritems_mapping_or_iterable(arg: MapOrIterItems[KT, VT]) -> IterItems[KT, VT]:
|
|
"""Yield the items in *arg*.
|
|
|
|
If *arg* is a :class:`~collections.abc.Mapping`, return an iterator over its items.
|
|
Otherwise return an iterator over *arg* itself.
|
|
"""
|
|
return iter(arg.items() if isinstance(arg, Mapping) else arg)
|
|
|
|
|
|
def _iteritems_args_kw(*args: MapOrIterItems[KT, VT], **kw: VT) -> IterItems[KT, VT]:
|
|
"""Yield the items from the positional argument (if given) and then any from *kw*.
|
|
|
|
:raises TypeError: if more than one positional argument is given.
|
|
"""
|
|
args_len = len(args)
|
|
if args_len > 1:
|
|
raise TypeError(f'Expected at most 1 positional argument, got {args_len}')
|
|
it: IterItems = ()
|
|
if args:
|
|
arg = args[0]
|
|
if arg:
|
|
it = _iteritems_mapping_or_iterable(arg)
|
|
if kw:
|
|
iterkw = iter(kw.items())
|
|
it = chain(it, iterkw) if it else iterkw
|
|
return it or _NULL_IT
|
|
|
|
|
|
@_t.overload
|
|
def inverted(arg: _t.Mapping[KT, VT]) -> IterItems[VT, KT]: ...
|
|
@_t.overload
|
|
def inverted(arg: IterItems[KT, VT]) -> IterItems[VT, KT]: ...
|
|
def inverted(arg: MapOrIterItems[KT, VT]) -> IterItems[VT, KT]:
|
|
"""Yield the inverse items of the provided object.
|
|
|
|
If *arg* has a :func:`callable` ``__inverted__`` attribute,
|
|
return the result of calling it.
|
|
|
|
Otherwise, return an iterator over the items in `arg`,
|
|
inverting each item on the fly.
|
|
|
|
*See also* :attr:`bidict.BidirectionalMapping.__inverted__`
|
|
"""
|
|
inv = getattr(arg, '__inverted__', None)
|
|
if callable(inv):
|
|
return inv() # type: ignore [no-any-return]
|
|
return ((val, key) for (key, val) in _iteritems_mapping_or_iterable(arg))
|