diff options
author | Jason R. Coombs <jaraco@jaraco.com> | 2022-02-06 12:43:29 -0500 |
---|---|---|
committer | Jason R. Coombs <jaraco@jaraco.com> | 2022-02-06 13:46:02 -0500 |
commit | b257d137ae2bcf1ef2e188f20e60f3ca5770e090 (patch) | |
tree | 4f5bf0e3c019814e1ad92d1060843cf0ef088417 /setuptools/_itertools.py | |
parent | 161ff0ff6f679967d323e9fd461eff312d0f12e6 (diff) | |
download | python-setuptools-git-b257d137ae2bcf1ef2e188f20e60f3ca5770e090.tar.gz |
In egg_info, port use of pkg_resources.EntryPoint to importlib.metadata
Diffstat (limited to 'setuptools/_itertools.py')
-rw-r--r-- | setuptools/_itertools.py | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/setuptools/_itertools.py b/setuptools/_itertools.py new file mode 100644 index 00000000..b8bf6d21 --- /dev/null +++ b/setuptools/_itertools.py @@ -0,0 +1,23 @@ +from setuptools.extern.more_itertools import consume # noqa: F401 + + +# copied from jaraco.itertools 6.1 +def ensure_unique(iterable, key=lambda x: x): + """ + Wrap an iterable to raise a ValueError if non-unique values are encountered. + + >>> list(ensure_unique('abc')) + ['a', 'b', 'c'] + >>> consume(ensure_unique('abca')) + Traceback (most recent call last): + ... + ValueError: Duplicate element 'a' encountered. + """ + seen = set() + seen_add = seen.add + for element in iterable: + k = key(element) + if k in seen: + raise ValueError(f"Duplicate element {element!r} encountered.") + seen_add(k) + yield element |