1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
|
#!/usr/bin/python3 -i
#
# Copyright (c) 2019 Collabora, Ltd.
#
# SPDX-License-Identifier: Apache-2.0
#
# Author(s): Ryan Pavlik <ryan.pavlik@collabora.com>
"""Provides general-purpose data structures."""
class DictOfStringSets:
"""A dictionary where the values are sets of strings.
Has some convenience functions to allow easier maintenance via
the .add method."""
def __init__(self, d=None):
self.d = {}
if d:
for k, v in d.items():
self.add(k, v)
def __getitem__(self, k):
return self.d[k]
def __contains__(self, k):
return k in self.d
def get(self, k, default=None):
return self.d.get(k, default)
def get_dict(self):
return self.d
def items(self):
"""Return an iterator like dict().items()."""
return self.d.items()
def keys(self):
"""Return an iterator over keys."""
return self.d.keys()
def values(self):
"""Return an iterator over values."""
return self.d.values()
def add_key(self, k):
"""Ensure the set for the given key exists."""
if k not in self.d:
self.d[k] = set()
def add(self, k, v):
self.add_key(k)
if isinstance(v, str):
v = (v, )
if not isinstance(v, set):
v = set(v)
self.d[k].update(v)
|
