pk/atst
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use json instead of pickle

Browse Source
This commit is contained in:
richard-dds 2018-11-16 14:12:42 -05:00
parent af105e08bc
commit 29f9520dda
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
atst/utils/form_cache.py
View File

@ -1,5 +1,5 @@
from hashlib import sha256
import pickle
import json
DEFAULT_CACHE_NAME = "formcache"
@ -15,7 +15,7 @@ class FormCache(object):
return self.read(cache_key)
def write(self, formdata, expiry_seconds=3600, key_prefix=DEFAULT_CACHE_NAME):
value = pickle.dumps(formdata)
value = json.dumps(formdata)
hash_ = sha256().hexdigest()
self.redis.setex(
name=self._key(key_prefix, hash_), value=value, time=expiry_seconds
@ -24,7 +24,7 @@ class FormCache(object):
def read(self, formdata_key, key_prefix=DEFAULT_CACHE_NAME):
data = self.redis.get(self._key(key_prefix, formdata_key))
return pickle.loads(data) if data is not None else {}
return json.loads(data) if data is not None else {}
@staticmethod
def _key(prefix, hash_):