1949
|
1 |
#!/bin/env python
|
3617
|
2 |
#Copyright ReportLab Europe Ltd. 2000-2012
|
1949
|
3 |
#see license.txt for license details
|
2332
|
4 |
#history http://www.reportlab.co.uk/cgi-bin/viewcvs.cgi/public/reportlab/trunk/reportlab/lib/set_ops.py
|
3029
|
5 |
__version__=''' $Id$ '''
|
|
6 |
__doc__="""From before Python had a Set class..."""
|
|
7 |
|
1949
|
8 |
import types
|
|
9 |
|
|
10 |
def __set_coerce(t, S):
|
3721
|
11 |
if t is list:
|
1949
|
12 |
return list(S)
|
3721
|
13 |
elif t is tuple:
|
1949
|
14 |
return tuple(S)
|
3721
|
15 |
elif t is bytes:
|
3794
|
16 |
return ''.join(S)
|
1949
|
17 |
return S
|
|
18 |
|
|
19 |
def unique(seq):
|
|
20 |
result = []
|
|
21 |
for i in seq:
|
|
22 |
if i not in result:
|
|
23 |
result.append(i)
|
|
24 |
return __set_coerce(type(seq), result)
|
|
25 |
|
|
26 |
def intersect(seq1, seq2):
|
|
27 |
result = []
|
3721
|
28 |
if type(seq1) != type(seq2) and type(seq2) == bytes: seq2 = list(seq2)
|
1949
|
29 |
for i in seq1:
|
|
30 |
if i in seq2 and i not in result: result.append(i)
|
|
31 |
return __set_coerce(type(seq1), result)
|
|
32 |
|
|
33 |
def union(seq1, seq2):
|
|
34 |
if type(seq1) == type(seq2):
|
|
35 |
return unique(seq1 + seq2)
|
3721
|
36 |
if type(seq1) == list or type(seq2) == list:
|
1949
|
37 |
return unique(list(seq1) + list(seq2))
|
3721
|
38 |
if type(seq1) == tuple or type(seq2) == tuple:
|
1949
|
39 |
return unique(tuple(seq1) + tuple(seq2))
|
|
40 |
return unique(list(seq1) + list(seq2))
|