Как правильно создать подкласс dict и переопределить __getitem__ & __setitem__

Question

Я отлаживаю некоторый код и хочу выяснить, когда происходит доступ к определенному словарю. Ну, на самом деле это класс, который подкласс dict и реализует пару дополнительных функций. В любом случае, я хотел бы сделать это подклассом dict и добавить переопределение __getitem__ и __setitem__ для получения некоторого результата отладки. Прямо сейчас у меня есть

class DictWatch(dict):
    def __init__(self, *args):
        dict.__init__(self, args)

    def __getitem__(self, key):
        val = dict.__getitem__(self, key)
        log.info("GET %s['%s'] = %s" % str(dict.get(self, 'name_label')), str(key), str(val)))
        return val

    def __setitem__(self, key, val):
        log.info("SET %s['%s'] = %s" % str(dict.get(self, 'name_label')), str(key), str(val)))
        dict.__setitem__(self, key, val)

'name_label' - это ключ, который в конечном итоге будет установлен, и я хочу использовать его для идентификации вывода. Затем я изменил класс, который я использую, на подкласс DictWatch вместо dict и изменил вызов суперконструктора. Тем не менее, кажется, ничего не происходит. Я думал, что я умен, но мне интересно, должен ли я идти в другом направлении.

Спасибо за помощь!

Answer 1

Другая проблема при создании подкласса dict заключается в том, что встроенный __init__ не вызывает update, а встроенный update не вызывает __setitem__. Итак, если вы хотите, чтобы все операции setitem проходили через вашу функцию __setitem__, вы должны убедиться, что она вызывается вами:

class DictWatch(dict):
    def __init__(self, *args, **kwargs):
        self.update(*args, **kwargs)

    def __getitem__(self, key):
        val = dict.__getitem__(self, key)
        print 'GET', key
        return val

    def __setitem__(self, key, val):
        print 'SET', key, val
        dict.__setitem__(self, key, val)

    def __repr__(self):
        dictrepr = dict.__repr__(self)
        return '%s(%s)' % (type(self).__name__, dictrepr)

    def update(self, *args, **kwargs):
        print 'update', args, kwargs
        for k, v in dict(*args, **kwargs).iteritems():
            self[k] = v

Answer 2

То, что вы делаете, должно работать. Я проверил ваш класс, и, кроме пропущенных открывающих скобок в ваших логах, он работает просто отлично. Есть только две вещи, о которых я могу думать. Во-первых, правильно ли настроен вывод вашего оператора журнала? Возможно, вам придется поставить logging.basicConfig(level=logging.DEBUG) в верхней части вашего сценария.

Во-вторых, __getitem__ и __setitem__ вызываются только при [] доступах. Поэтому убедитесь, что вы получаете доступ только к DictWatch через d[key], а не d.get() и d.set()

Answer 3

Это не должно действительно изменить результат (который должен работать, для хороших пороговых значений регистрации): ваш init должен быть:

def __init__(self,*args,**kwargs) : dict.__init__(self,*args,**kwargs) 

вместо этого, потому что если вы вызываете свой методс DictWatch ([(1,2), (2,3)]) или DictWatch (a = 1, b = 2) это не удастся.

(или, лучше, не определять конструктор дляэто)

Answer 4

Рассмотрим подклассы UserDict или UserList. Эти классы предназначены для использования в подклассах, тогда как обычные dict и list - нет, и содержат оптимизации.

Answer 5

Все, что вам нужно сделать, это

class BatchCollection(dict):
    def __init__(self, inpt={}):
        super(BatchCollection, self).__init__(inpt)

Пример использования для моего личного использования

### EXAMPLE
class BatchCollection(dict):
    def __init__(self, inpt={}):
        super(BatchCollection, self).__init__(inpt)

    def __setitem__(self, key, item):
        if (isinstance(key, tuple) and len(key) == 2
                and isinstance(item, collections.Iterable)):
            # self.__dict__[key] = item
            super(BatchCollection, self).__setitem__(key, item)
        else:
            raise Exception(
                "Valid key should be a tuple (database_name, table_name) "
                "and value should be iterable")

Примечание : протестировано только в python3