mirror of
https://github.com/django-q2/django-q2.git
synced 2026-09-30 04:38:13 +08:00
45 lines
1.1 KiB
Python
45 lines
1.1 KiB
Python
"""Package signing."""
|
|
try:
|
|
import cPickle as pickle
|
|
except ImportError:
|
|
import pickle
|
|
|
|
from django.core import signing
|
|
|
|
from django_q.conf import Conf
|
|
|
|
BadSignature = signing.BadSignature
|
|
|
|
|
|
class SignedPackage(object):
|
|
|
|
"""Wraps Django's signing module with custom Pickle serializer."""
|
|
|
|
@staticmethod
|
|
def dumps(obj, compressed=Conf.COMPRESSED):
|
|
return signing.dumps(obj,
|
|
key=Conf.SECRET_KEY,
|
|
salt=Conf.PREFIX,
|
|
compress=compressed,
|
|
serializer=PickleSerializer)
|
|
|
|
@staticmethod
|
|
def loads(obj):
|
|
return signing.loads(obj,
|
|
key=Conf.SECRET_KEY,
|
|
salt=Conf.PREFIX,
|
|
serializer=PickleSerializer)
|
|
|
|
|
|
class PickleSerializer(object):
|
|
|
|
"""Simple wrapper around Pickle for signing.dumps and signing.loads."""
|
|
|
|
@staticmethod
|
|
def dumps(obj):
|
|
return pickle.dumps(obj)
|
|
|
|
@staticmethod
|
|
def loads(data):
|
|
return pickle.loads(data)
|