# Dictionaries

import itertools

import util

class IdDict(dict):

    '''A dict that stores objects by id()'''

    def add(self, *values):

        for value in values: self[id(value)] = value

        return self

    def add_vars(self, vars_): return self.add(*vars_.values())

class MergeDict:

    '''A dict that checks each of several dicts'''

    def __init__(self, *dicts): self.dicts = dicts

    def __getitem__(self, key):

        for dict_ in self.dicts:

            try: return dict_[key]

            except KeyError: pass

        raise # reraise last KeyError

class AttrsDictView:

    '''A dict view of an object's attributes

    @pre If you want __iter__() to work, value must have a __dict__

    '''

    def __init__(self, value): self.value = value

    def __iter__(self): return iter(self.value.__dict__)

    def __getitem__(self, key): return getattr(self.value, key)

def make_hashable(value):

    if isinstance(value, list): value = tuple(value)

    elif isinstance(value, dict): value = util.NamedTuple(**value)

    return value

def join(dict0, dict1):

    '''Joins dict0 (A->B mapping) to dict1 (B->C mapping) SQL-style.

    If a value in dict0 has no key in dict1, uses the value itself.

    '''

    new_dict = {}

    for out, in_ in dict0.iteritems():

        new_dict[out] = dict1.get(in_, in_) # use in_ if no match found

    return new_dict