-
-
Notifications
You must be signed in to change notification settings - Fork 798
/
memcachedstorage.py
36 lines (28 loc) · 987 Bytes
/
memcachedstorage.py
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
# -*- coding: utf-8 -*-
from __future__ import absolute_import, unicode_literals
from wechatpy.session import SessionStorage
from wechatpy.utils import to_text
from wechatpy.utils import json
class MemcachedStorage(SessionStorage):
def __init__(self, mc, prefix='wechatpy'):
for method_name in ('get', 'set', 'delete'):
assert hasattr(mc, method_name)
self.mc = mc
self.prefix = prefix
def key_name(self, key):
return '{0}:{1}'.format(self.prefix, key)
def get(self, key, default=None):
key = self.key_name(key)
value = self.mc.get(key)
if value is None:
return default
return json.loads(to_text(value))
def set(self, key, value, ttl=0):
if value is None:
return
key = self.key_name(key)
value = json.dumps(value)
self.mc.set(key, value, ttl)
def delete(self, key):
key = self.key_name(key)
self.mc.delete(key)