view roundup/cgi/PageTemplates/MultiMapping.py @ 6822:5053ee6c846b

memorydb fixes for otks tests. in test setup otks was not set. Also fixed import from dbm used by memorydb OneTimeKeys to inherit from OneTimeKeys rather than Sessions.
author John Rouillard <rouilj@ieee.org>
date Sun, 07 Aug 2022 01:26:30 -0400
parents 23b8e6067f7c
children
line wrap: on
line source

import operator

class MultiMapping:
    def __init__(self, *stores):
        self.stores = list(stores)
    def __getitem__(self, key):
        for store in self.stores:
            if key in store:
                return store[key]
        raise KeyError(key)
    _marker = []
    def get(self, key, default=_marker):
        for store in self.stores:
            if key in store:
                return store[key]
        if default is self._marker:
            raise KeyError(key)
        return default
    def __len__(self):
        return sum([len(x) for x in self.stores])
    def push(self, store):
        self.stores.append(store)
    def pop(self):
        return self.stores.pop()
    def items(self):
        l = []
        for store in self.stores:
            l = l + list(store.items())
        return l

Roundup Issue Tracker: http://roundup-tracker.org/