Show More
@@ -1,46 +1,55 | |||
|
1 | 1 | # dicthelpers.py - helper routines for Python dicts |
|
2 | 2 | # |
|
3 | 3 | # Copyright 2013 Facebook |
|
4 | 4 | # |
|
5 | 5 | # This software may be used and distributed according to the terms of the |
|
6 | 6 | # GNU General Public License version 2 or any later version. |
|
7 | 7 | |
|
8 |
def |
|
|
8 | def diff(d1, d2, default=None): | |
|
9 | '''Return all key-value pairs that are different between d1 and d2. | |
|
10 | ||
|
11 | This includes keys that are present in one dict but not the other, and | |
|
12 | keys whose values are different. The return value is a dict with values | |
|
13 | being pairs of values from d1 and d2 respectively, and missing values | |
|
14 | represented as default.''' | |
|
9 | 15 | res = {} |
|
10 |
if d1 is d2 |
|
|
16 | if d1 is d2: | |
|
11 | 17 | # same dict, so diff is empty |
|
12 | 18 | return res |
|
13 | 19 | |
|
14 | 20 | for k1, v1 in d1.iteritems(): |
|
15 | 21 | if k1 in d2: |
|
16 | 22 | v2 = d2[k1] |
|
17 |
if |
|
|
23 | if v1 != v2: | |
|
18 | 24 | res[k1] = (v1, v2) |
|
19 | 25 | else: |
|
20 | 26 | res[k1] = (v1, default) |
|
21 | 27 | |
|
28 | for k2 in d2: | |
|
29 | if k2 not in d1: | |
|
30 | res[k2] = (default, d2[k2]) | |
|
31 | ||
|
32 | return res | |
|
33 | ||
|
34 | def join(d1, d2, default=None): | |
|
35 | '''Return all key-value pairs from both d1 and d2. | |
|
36 | ||
|
37 | This is akin to an outer join in relational algebra. The return value is a | |
|
38 | dict with values being pairs of values from d1 and d2 respectively, and | |
|
39 | missing values represented as default.''' | |
|
40 | res = {} | |
|
41 | ||
|
42 | for k1, v1 in d1.iteritems(): | |
|
43 | if k1 in d2: | |
|
44 | res[k1] = (v1, d2[k1]) | |
|
45 | else: | |
|
46 | res[k1] = (v1, default) | |
|
47 | ||
|
22 | 48 | if d1 is d2: |
|
23 | 49 | return res |
|
24 | 50 | |
|
25 | 51 | for k2 in d2: |
|
26 | 52 | if k2 not in d1: |
|
27 | 53 | res[k2] = (default, d2[k2]) |
|
28 | 54 | |
|
29 | 55 | return res |
|
30 | ||
|
31 | def diff(d1, d2, default=None): | |
|
32 | '''Return all key-value pairs that are different between d1 and d2. | |
|
33 | ||
|
34 | This includes keys that are present in one dict but not the other, and | |
|
35 | keys whose values are different. The return value is a dict with values | |
|
36 | being pairs of values from d1 and d2 respectively, and missing values | |
|
37 | represented as default.''' | |
|
38 | return _diffjoin(d1, d2, default, True) | |
|
39 | ||
|
40 | def join(d1, d2, default=None): | |
|
41 | '''Return all key-value pairs from both d1 and d2. | |
|
42 | ||
|
43 | This is akin to an outer join in relational algebra. The return value is a | |
|
44 | dict with values being pairs of values from d1 and d2 respectively, and | |
|
45 | missing values represented as default.''' | |
|
46 | return _diffjoin(d1, d2, default, False) |
General Comments 0
You need to be logged in to leave comments.
Login now