statdict

dictionairy to keep stats.

class fbf.utils.statdict.StatDict

Bases: fbf.utils.lazydict.LazyDict

dictionary to hold stats

countdictkeys(d)

count all keys of a dict.

countdictvalues(d)

count all keys of a dict.

down(end=100, limit=None)

return lowest items

downitem(item, value=1)

decrease item

set(item, value)

set item to value

top(start=1, limit=None)

return highest items

upitem(item, value=1)

increase item

CODE

# fbf/utils/statdict.py
#
#

""" dictionairy to keep stats. """

fbf imports

from fbf.utils.lazydict import LazyDict
from fbf.lib.errors import WrongType

basic imports

import functools

classes

class StatDict(LazyDict):

    """ dictionary to hold stats """

    def __iadd__(self, b):
        if not type(b) == StatDict: raise WrongType(type(b))
        for key, value in b.items():
            self.upitem(key, value)
        return self

    def set(self, item, value):
        """ set item to value """
        self[item] = value

    def upitem(self, item, value=1):
        """ increase item """
        if item not in self:
            self[item] = value
            return
        self[item] += value

    def downitem(self, item, value=1):
        """ decrease item """
        if item not in self:
            self[item] = value
            return
        self[item] -= value

    def top(self, start=1, limit=None):
        """ return highest items """
        result = []
        for item, value in self.items():
            if value >= start: result.append((item, value))
        result.sort()
        if limit: result =  result[:limit]
        return result

    def down(self, end=100, limit=None):
        """ return lowest items """
        result = []
        for item, value in self.items():
            if value <= end: result.append((item, value))
        result.sort(reverse=True)
        if limit: return result[:limit]
        else: return result

    def countdictkeys(self, d):
        """ count all keys of a dict. """
        for key in d:
            self.upitem(key)

    def countdictvalues(self, d):
        """ count all keys of a dict. """
        for key, value in d.items():
            try: self.upitem(value)
            except TypeError: pass