You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
57 lines
1.4 KiB
57 lines
1.4 KiB
import six
|
|
|
|
from six.moves import filter, map, UserDict
|
|
from collections import defaultdict
|
|
|
|
|
|
class HashMap(UserDict):
|
|
def iter(self):
|
|
return iter(self.data)
|
|
|
|
def items(self):
|
|
return six.iteritems(self.data)
|
|
|
|
def keys(self):
|
|
return six.iterkeys(self.data)
|
|
|
|
def values(self):
|
|
return six.itervalues(self.data)
|
|
|
|
def find(self, predicate):
|
|
if not callable(predicate):
|
|
raise TypeError('predicate must be callable')
|
|
|
|
for obj in self.values():
|
|
if predicate(obj):
|
|
yield obj
|
|
|
|
def find_one(self, predicate):
|
|
return next(self.find(predicate), None)
|
|
|
|
def select(self, *args, **kwargs):
|
|
if kwargs:
|
|
args += tuple([kwargs])
|
|
|
|
for obj in self.values():
|
|
for check in args:
|
|
for k, v in six.iteritems(check):
|
|
if getattr(obj, k) != v:
|
|
break
|
|
yield obj
|
|
|
|
def select_one(self, **kwargs):
|
|
return next(self.select(**kwargs), None)
|
|
|
|
def filter(self, predicate):
|
|
if not callable(predicate):
|
|
raise TypeError('predicate must be callable')
|
|
return filter(predicate, self.values())
|
|
|
|
def map(self, predicate):
|
|
if not callable(predicate):
|
|
raise TypeError('predicate must be callable')
|
|
return map(predicate, self.values())
|
|
|
|
|
|
class DefaultHashMap(defaultdict, HashMap):
|
|
pass
|
|
|