| 123456789101112131415161718192021222324252627282930313233343536 |
- # -*- 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)
|
