优化工单模块,增加企微模块
This commit is contained in:
34
sg_wechat_enterprise/we_api/session/shovestorage.py
Normal file
34
sg_wechat_enterprise/we_api/session/shovestorage.py
Normal file
@@ -0,0 +1,34 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
from __future__ import absolute_import, unicode_literals
|
||||
from wechatpy.session import SessionStorage
|
||||
|
||||
|
||||
class ShoveStorage(SessionStorage):
|
||||
|
||||
def __init__(self, shove, prefix='wechatpy'):
|
||||
self.shove = shove
|
||||
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)
|
||||
try:
|
||||
return self.shove[key]
|
||||
except KeyError:
|
||||
return default
|
||||
|
||||
def set(self, key, value, ttl=None):
|
||||
if value is None:
|
||||
return
|
||||
|
||||
key = self.key_name(key)
|
||||
self.shove[key] = value
|
||||
|
||||
def delete(self, key):
|
||||
key = self.key_name(key)
|
||||
try:
|
||||
del self.shove[key]
|
||||
except KeyError:
|
||||
pass
|
||||
Reference in New Issue
Block a user