tyler.durden
/
compose
mirror of https://github.com/docker/compose.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

utils: Add unique_everseen (from itertools recipies). 

Signed-off-by: Antony MECHIN <antony.mechin@docker.com>
This commit is contained in:
Antony MECHIN 2018-09-24 18:08:17 +02:00
parent bbcfce4029
commit de1958c5ff
2 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
compose/utils.py
View File

@ -170,3 +170,12 @@ def truncate_id(value):
if len(value) > 12: if len(value) > 12:
return value[:12] return value[:12]
return value return value
def unique_everseen(iterable, key=lambda x: x):
"List unique elements, preserving order. Remember all elements ever seen."
seen = set()
for element in iterable:
if key(element) not in seen:
seen.add(element)
yield element

6
tests/unit/utils_test.py
View File

@ -68,3 +68,9 @@ class TestParseBytes(object):
assert utils.parse_bytes(123) == 123 assert utils.parse_bytes(123) == 123
assert utils.parse_bytes('foobar') is None assert utils.parse_bytes('foobar') is None
assert utils.parse_bytes('123') == 123 assert utils.parse_bytes('123') == 123
class TestMoreItertools(object):
def test_unique_everseen(self):
assert list(utils.unique_everseen([2, 1, 2, 1])) == [2, 1]
assert list(utils.unique_everseen([2, 1, 2, 1], hash)) == [2, 1]