|
@@ -8,6 +8,7 @@
|
|
|
"""
|
|
|
from __future__ import absolute_import
|
|
|
|
|
|
+import sys
|
|
|
try: # pragma: no cover
|
|
|
import cassandra
|
|
|
except ImportError: # pragma: no cover
|
|
@@ -16,7 +17,6 @@ except ImportError: # pragma: no cover
|
|
|
from celery import states
|
|
|
from celery.exceptions import ImproperlyConfigured
|
|
|
from celery.utils.log import get_logger
|
|
|
-
|
|
|
from .base import BaseBackend
|
|
|
|
|
|
__all__ = ['NewCassandraBackend']
|
|
@@ -156,6 +156,10 @@ class NewCassandraBackend(BaseBackend):
|
|
|
"""Store return value and status of an executed task."""
|
|
|
self._get_connection(write=True)
|
|
|
|
|
|
+ import sys
|
|
|
+ if sys.version_info > (3,):
|
|
|
+ buffer = memoryview
|
|
|
+
|
|
|
self._session.execute(self._write_stmt, (
|
|
|
task_id,
|
|
|
status,
|