reportlab/lib/set_ops.py
author rgbecker
Tue, 10 Jun 2003 09:59:06 +0000
changeset 1949 6032704b94db
child 2332 2a7ab4405e18
permissions -rw-r--r--
Initial check in
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
1949
6032704b94db Initial check in
rgbecker
parents:
diff changeset
     1
#!/bin/env python
6032704b94db Initial check in
rgbecker
parents:
diff changeset
     2
#copyright ReportLab Inc. 2000
6032704b94db Initial check in
rgbecker
parents:
diff changeset
     3
#see license.txt for license details
6032704b94db Initial check in
rgbecker
parents:
diff changeset
     4
#history http://cvs.sourceforge.net/cgi-bin/cvsweb.cgi/reportlab/lib/set_ops.py?cvsroot=reportlab
6032704b94db Initial check in
rgbecker
parents:
diff changeset
     5
#$Header: /tmp/reportlab/reportlab/lib/set_ops.py,v 1.1 2003/06/10 09:59:06 rgbecker Exp $
6032704b94db Initial check in
rgbecker
parents:
diff changeset
     6
import types
6032704b94db Initial check in
rgbecker
parents:
diff changeset
     7
import string
6032704b94db Initial check in
rgbecker
parents:
diff changeset
     8
6032704b94db Initial check in
rgbecker
parents:
diff changeset
     9
def __set_coerce(t, S):
6032704b94db Initial check in
rgbecker
parents:
diff changeset
    10
    if t is types.ListType:
6032704b94db Initial check in
rgbecker
parents:
diff changeset
    11
        return list(S)
6032704b94db Initial check in
rgbecker
parents:
diff changeset
    12
    elif t is types.TupleType:
6032704b94db Initial check in
rgbecker
parents:
diff changeset
    13
        return tuple(S)
6032704b94db Initial check in
rgbecker
parents:
diff changeset
    14
    elif t is types.StringType:
6032704b94db Initial check in
rgbecker
parents:
diff changeset
    15
        return string.join(S, '')
6032704b94db Initial check in
rgbecker
parents:
diff changeset
    16
    return S
6032704b94db Initial check in
rgbecker
parents:
diff changeset
    17
6032704b94db Initial check in
rgbecker
parents:
diff changeset
    18
def unique(seq):
6032704b94db Initial check in
rgbecker
parents:
diff changeset
    19
    result = []
6032704b94db Initial check in
rgbecker
parents:
diff changeset
    20
    for i in seq:
6032704b94db Initial check in
rgbecker
parents:
diff changeset
    21
        if i not in result:
6032704b94db Initial check in
rgbecker
parents:
diff changeset
    22
            result.append(i)
6032704b94db Initial check in
rgbecker
parents:
diff changeset
    23
    return __set_coerce(type(seq), result)
6032704b94db Initial check in
rgbecker
parents:
diff changeset
    24
6032704b94db Initial check in
rgbecker
parents:
diff changeset
    25
def intersect(seq1, seq2):
6032704b94db Initial check in
rgbecker
parents:
diff changeset
    26
    result = []
6032704b94db Initial check in
rgbecker
parents:
diff changeset
    27
    if type(seq1) != type(seq2) and type(seq2) == types.StringType: seq2 = list(seq2)
6032704b94db Initial check in
rgbecker
parents:
diff changeset
    28
    for i in seq1:
6032704b94db Initial check in
rgbecker
parents:
diff changeset
    29
        if i in seq2 and i not in result: result.append(i)
6032704b94db Initial check in
rgbecker
parents:
diff changeset
    30
    return __set_coerce(type(seq1), result)
6032704b94db Initial check in
rgbecker
parents:
diff changeset
    31
6032704b94db Initial check in
rgbecker
parents:
diff changeset
    32
def union(seq1, seq2):
6032704b94db Initial check in
rgbecker
parents:
diff changeset
    33
    if type(seq1) == type(seq2):
6032704b94db Initial check in
rgbecker
parents:
diff changeset
    34
        return unique(seq1 + seq2)
6032704b94db Initial check in
rgbecker
parents:
diff changeset
    35
    if type(seq1) == types.ListType or type(seq2) == types.ListType:
6032704b94db Initial check in
rgbecker
parents:
diff changeset
    36
        return unique(list(seq1) + list(seq2))
6032704b94db Initial check in
rgbecker
parents:
diff changeset
    37
    if type(seq1) == types.TupleType or type(seq2) == types.TupleType:
6032704b94db Initial check in
rgbecker
parents:
diff changeset
    38
        return unique(tuple(seq1) + tuple(seq2))
6032704b94db Initial check in
rgbecker
parents:
diff changeset
    39
    return unique(list(seq1) + list(seq2))