problem with interface of operator.itemgetter

dou dou nirvana117 at gmail.com
Fri Aug 14 05:16:18 CEST 2009


I have a function to do some thing like LEFT JOIN in SQL, the function use
the itemgetter to get the "ON" and "SELECT" parameters of the two table(list
of list), the problem is that itemgetter may return a value or a tuple of
values, because of the inconsistent return type of itemgetter, I have to do
2^3 IF check,
the function like this:

def left_join(table1, on_index1, table2, on_index2, getter1, getter2):
    """do thing like:
    SELECT on1(table1), g1(table1), g2(table2) FROM table1
    LEFT JOIN table2
    ON on1(table1) == on2(table2)
    """
    on1 = itemgetter(*on_index1)
    on2 = itemgetter(*on_index2)
    g1 = itemgetter(*getter1)
    g2 = itemgetter(*getter2)

    d2 = {}
    d2 = dict((on2(r2), r2) for r2 in table2)

    #if itemgetter always return tuple, below could simple as one line
    #return [ list(on1(r1) + g1(r1) + g2(d2.get(on1(r1)))) for r1 in table1
]

    len_on = len(on_index1)
    len_g1 = len(getter1)
    len_g2 = len(getter2)

    if len_on == 1:
        if len_g1 == 1 and len_g2 == 1:
            return [ [on1(r1), g1(r1), g2(d2.get(on1(r1)))] for r1 in table1
]
        elif len_g1 == 1 and len_g2 > 1:
            return [ list((on1(r1),g1(r1))+g2(d2.get(on1(r1)))) for r1 in
table1 ]
        elif len_g1 > 1 and len_g2 == 1:
            return [ list((on1(r1),)+g1(r1)+(g2(d2.get(on1(r1))),))  for r1
in table1 ]
        else: #len_g1 > 1 and len_g2 > 1:
            return [ list((on1(r1),)+g1(r1)+g2(d2.get(on1(r1)))) for r1 in
table1 ]
    else: # len_on > 1
        if len_g1 == 1 and len_g2 == 1:
            return [ list(on1(r1))+[g1(r1),g2(d2.get(on1(r1)))] for r1 in
table1 ]
        elif len_g1 == 1 and len_g2 > 1:
            return [ list(on1(r1)+(g1(r1),)+g2(d2.get(on1(r1)))) for r1 in
table1 ]
        elif len_g1 > 1 and len_g2 == 1:
            return [ list(on1(r1)+g1(r1)+(g2(d2.get(on1(r1))),))  for r1 in
table1 ]
        else: #len_g1 > 1 and len_g2 > 1:
            return [ list(on1(r1)+g1(r1)+g2(d2.get(on1(r1)))) for r1 in
table1 ]

so is there a way to force itemgetter to return tuple even when
itemgetter(only_have_one_argument)? or some other function to do this?

Thanks.
-------------- next part --------------
An HTML attachment was scrubbed...
URL: <http://mail.python.org/pipermail/python-list/attachments/20090814/104029d4/attachment.html>


More information about the Python-list mailing list