1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
|
"""
This module contains the container classes to create objects
that persist directly in a Redis server.
"""
import collections
from functools import partial
class Container(object):
"""Create a container object saved in Redis.
Arguments:
key -- the Redis key this container is stored at
db -- the Redis client object. Default: None
When ``db`` is not set, the gets the default connection from
``redisco.connection`` module.
"""
def __init__(self, key, db=None, pipeline=None):
self._db = db
self.key = key
self.pipeline = pipeline
def clear(self):
"""Remove container from Redis database."""
del self.db[self.key]
def __getattribute__(self, att):
if att in object.__getattribute__(self, 'DELEGATEABLE_METHODS'):
return partial(getattr(object.__getattribute__(self, 'db'), att), self.key)
else:
return object.__getattribute__(self, att)
@property
def db(self):
if self.pipeline:
return self.pipeline
if self._db:
return self._db
if hasattr(self, 'db_cache') and self.db_cache:
return self.db_cache
else:
from redisco import connection
self.db_cache = connection
return self.db_cache
DELEGATEABLE_METHODS = ()
class Hash(Container, collections.MutableMapping):
def __getitem__(self, att):
return self.hget(att)
def __setitem__(self, att, val):
self.hset(att, val)
def __delitem__(self, att):
self.hdel(att)
def __len__(self):
return self.hlen()
def __iter__(self):
return self.hgetall().__iter__()
def __contains__(self, att):
return self.hexists(att)
def __repr__(self):
return "<%s '%s' %s>" % (self.__class__.__name__, self.key, self.hgetall())
def keys(self):
return self.hkeys()
def values(self):
return self.hvals()
def _get_dict(self):
return self.hgetall()
def _set_dict(self, new_dict):
self.clear()
self.update(new_dict)
dict = property(_get_dict, _set_dict)
DELEGATEABLE_METHODS = ('hlen', 'hset', 'hdel', 'hkeys', 'hgetall', 'hvals',
'hget', 'hexists', 'hincrby', 'hmget', 'hmset')
|
