gozerbot.persist.pdol

pickled dict of lists

class gozerbot.persist.pdol.Pdol(fname)

Bases: gozerbot.persist.persist.Persist

pickled dict of lists

add(item, what)

add what to items list

adduniq(item, what)

add what to items list if item not yet added.

delete(item, what)

remove what from item’s list.

extend(item, what)

extend item with what.

get(item)

get items list.

new(what)

reset list of what.

remove(item, what)

remove what from item.

CODE

# gozerbot/pdol.py
#
#

""" pickled dict of lists """

__status__ = "seen"

gozerbot imports

from persist import Persist

Pdol class

class Pdol(Persist):

    """ pickled dict of lists """

    def __init__(self, fname):
        Persist.__init__(self, fname)
        if not self.data: self.data = {}

    def __iter__(self, name):
        """ loop over the pickled dict of lists. """
        return self.data[name].__iter__()

    def __getitem__(self, item):
        """ get item from the pdol. """
        if self.data.has_key(item): return self.data[item]

    def __delitem__(self, item):
        """ remove entry from pdol. """
        if self.data.has_key(item): self.data.__delitem__(item) ; return 1

    def __setitem__(self, item, what):
        """ set item on pdol. """
        if self.data.has_key(item): self.data[item].append(what)
        else: self.data[item] = [what]
        return 1

    def add(self, item, what):
        """ add what to items list """
        return self.__setitem__(item, what)

    def adduniq(self, item, what):
        """ add what to items list if item not yet added. """
        if not self.data.has_key(item): self.new(item)
        if what not in self.data[item]: return self.__setitem__(item, what)

    def get(self, item):
        """ get items list. """
        return self.__getitem__(item)

    def new(self, what):
        """ reset list of what. """
        self.data[what] = []

    def delete(self, item, what):
        """ remove what from item's list. """
        del self.data[item][what]

    def extend(self, item, what):
        """ extend item with what. """
        if not self.data.has_key(item):
            self.new(item)
        self.data[item].extend(what)

    def remove(self, item, what):
        """ remove what from item. """
        try: self.data[item].remove(what) ; return 1
        except (ValueError, KeyError): return 0