48 lines
1.1 KiB
Python
48 lines
1.1 KiB
Python
|
import json
|
||
|
|
|
||
|
|
import redis
|
||
|
|
from flask_session import RedisSessionInterface
|
||
|
|
|
||
|
|
import settings
|
||
|
|
from funding.bin.utils import json_encoder
|
||
|
|
|
||
|
|
|
||
|
|
def redis_args():
|
||
|
|
args = {
|
||
|
|
"host": settings.REDIS_HOST,
|
||
|
|
"port": settings.REDIS_PORT,
|
||
|
|
'socket_connect_timeout': 2,
|
||
|
|
'socket_timeout': 2,
|
||
|
|
'retry_on_timeout': True,
|
||
|
|
'decode_responses': True
|
||
|
|
}
|
||
|
|
if settings.REDIS_PASSWD:
|
||
|
|
args["password"] = settings.REDIS_PASSWD
|
||
|
|
return args
|
||
|
|
|
||
|
|
|
||
|
|
class JsonRedisSerializer:
|
||
|
|
@staticmethod
|
||
|
|
def loads(val):
|
||
|
|
try:
|
||
|
|
return json.loads(val).get("wow", {})
|
||
|
|
except ValueError:
|
||
|
|
return
|
||
|
|
|
||
|
|
@staticmethod
|
||
|
|
def dumps(val):
|
||
|
|
try:
|
||
|
|
return json.dumps({"wow": val})
|
||
|
|
except ValueError:
|
||
|
|
return
|
||
|
|
|
||
|
|
|
||
|
|
class JsonRedis(RedisSessionInterface):
|
||
|
|
serializer = JsonRedisSerializer
|
||
|
|
|
||
|
|
def __init__(self, key_prefix, use_signer=False, decode_responses=True):
|
||
|
|
super(JsonRedis, self).__init__(
|
||
|
|
redis=redis.Redis(**redis_args()),
|
||
|
|
key_prefix=key_prefix,
|
||
|
|
use_signer=use_signer)
|