dig implementation in Python

dig.py

def dig(dic, *keys):
    keys = list(keys)
    if isinstance(keys[0], list):
        return dig(dic, *keys[0])
    if isinstance(dic, dict) and keys[0] in dic or \
       isinstance(dic, list) and keys[0] < len(dic):
        if len(keys) == 1:
            return dic[keys[0]]
        return dig(dic[keys[0]], *keys[1:])
    return None

if __name__ == '__main__':
    h = { 'foo': { 'bar': { 'baz': 1 } } }
    print(dig(h, 'foo', 'bar', 'baz'))

    g = { 'foo': [10, 11, 12] }
    print(dig(g, 'foo', 1))