mirror of
https://github.com/django-q2/django-q2.git
synced 2026-10-09 21:30:55 +08:00
Linting and hints
This commit is contained in:
+13
-18
@@ -1,44 +1,39 @@
|
||||
"""Package signing."""
|
||||
try:
|
||||
import cPickle as pickle
|
||||
except ImportError:
|
||||
import pickle
|
||||
import pickle
|
||||
|
||||
from django_q import core_signing as signing
|
||||
|
||||
from django_q.conf import Conf
|
||||
|
||||
BadSignature = signing.BadSignature
|
||||
|
||||
|
||||
class SignedPackage:
|
||||
|
||||
"""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)
|
||||
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)
|
||||
return signing.loads(
|
||||
obj, key=Conf.SECRET_KEY, salt=Conf.PREFIX, serializer=PickleSerializer
|
||||
)
|
||||
|
||||
|
||||
class PickleSerializer:
|
||||
|
||||
"""Simple wrapper around Pickle for signing.dumps and signing.loads."""
|
||||
|
||||
@staticmethod
|
||||
def dumps(obj):
|
||||
def dumps(obj) -> bytes:
|
||||
return pickle.dumps(obj, protocol=pickle.HIGHEST_PROTOCOL)
|
||||
|
||||
@staticmethod
|
||||
def loads(data):
|
||||
def loads(data) -> any:
|
||||
return pickle.loads(data)
|
||||
|
||||
Reference in New Issue
Block a user