diff --git a/django_elasticache/cluster_utils.py b/django_elasticache/cluster_utils.py index 60594ca..ff1fe09 100644 --- a/django_elasticache/cluster_utils.py +++ b/django_elasticache/cluster_utils.py @@ -2,7 +2,7 @@ utils for discovery cluster """ from distutils.version import StrictVersion -from django.utils.encoding import smart_text +from django.utils.encoding import smart_str import re from telnetlib import Telnet @@ -35,7 +35,7 @@ def get_cluster_info(host, port, ignore_cluster_errors=False): if len(version_list) not in [2, 3] or version_list[0] != b'VERSION': raise WrongProtocolData('version', res) version = version_list[1] - if StrictVersion(smart_text(version)) >= StrictVersion('1.4.14'): + if StrictVersion(smart_str(version)) >= StrictVersion('1.4.14'): cmd = b'config get cluster\n' else: cmd = b'get AmazonElastiCache:cluster\n' @@ -50,8 +50,8 @@ def get_cluster_info(host, port, ignore_cluster_errors=False): return { 'version': version, 'nodes': [ - '{0}:{1}'.format(smart_text(host), - smart_text(port)) + '{0}:{1}'.format(smart_str(host), + smart_str(port)) ] } @@ -67,8 +67,8 @@ def get_cluster_info(host, port, ignore_cluster_errors=False): try: for node in ls[2].split(b' '): host, ip, port = node.split(b'|') - nodes.append('{0}:{1}'.format(smart_text(ip or host), - smart_text(port))) + nodes.append('{0}:{1}'.format(smart_str(ip or host), + smart_str(port))) except ValueError: raise WrongProtocolData(cmd, res) return { diff --git a/django_elasticache/memcached.py b/django_elasticache/memcached.py index a78fbd7..31b2030 100644 --- a/django_elasticache/memcached.py +++ b/django_elasticache/memcached.py @@ -3,7 +3,7 @@ """ import socket from functools import wraps -from django.core.cache import InvalidCacheBackendError +from django.core.cache.backends.base import InvalidCacheBackendError from django.core.cache.backends.memcached import PyLibMCCache from .cluster_utils import get_cluster_info