A hacky pluck function for python, used to flatten data

python-pluck.py

def pluck(obj, *keys):
    def i(o, k):
        if "." in k:
            (sub, rest) = k.split('.', 1)
            if not sub in o:
                return None
            val = o[sub]
            if isinstance(val, list):
                return (k, list(i(x, rest)[1] for x in val))
            (_, val) = i(val, rest)
            return (k, val)
        if k not in o:
            return (k, None)
        return (k, o[k])
    if isinstance(obj, list):
        return list(dict(i(o, k) for k in keys) for o in obj)
    return dict(i(obj, k) for k in keys)