Skip to content

Commit

Permalink
[py3] Fixed encoding issues in cache key generation
Browse files Browse the repository at this point in the history
  • Loading branch information
claudep committed Aug 13, 2012
1 parent d774ad7 commit 45baaab
Show file tree
Hide file tree
Showing 3 changed files with 7 additions and 6 deletions.
6 changes: 3 additions & 3 deletions django/core/cache/backends/base.py
@@ -1,9 +1,9 @@
"Base Cache class."
from __future__ import unicode_literals

import warnings

from django.core.exceptions import ImproperlyConfigured, DjangoRuntimeWarning
from django.utils.encoding import smart_bytes
from django.utils.importlib import import_module

class InvalidCacheBackendError(ImproperlyConfigured):
Expand All @@ -23,7 +23,7 @@ def default_key_func(key, key_prefix, version):
the `key_prefix'. KEY_FUNCTION can be used to specify an alternate
function with custom key making behavior.
"""
return ':'.join([key_prefix, str(version), smart_bytes(key)])
return ':'.join([key_prefix, str(version), key])

def get_key_func(key_func):
"""
Expand Down Expand Up @@ -62,7 +62,7 @@ def __init__(self, params):
except (ValueError, TypeError):
self._cull_frequency = 3

self.key_prefix = smart_bytes(params.get('KEY_PREFIX', ''))
self.key_prefix = params.get('KEY_PREFIX', '')
self.version = params.get('VERSION', 1)
self.key_func = get_key_func(params.get('KEY_FUNCTION', None))

Expand Down
3 changes: 2 additions & 1 deletion django/utils/cache.py
Expand Up @@ -16,6 +16,7 @@
An example: i18n middleware would need to distinguish caches by the
"Accept-language" header.
"""
from __future__ import unicode_literals

import hashlib
import re
Expand Down Expand Up @@ -170,7 +171,7 @@ def _i18n_cache_key_suffix(request, cache_key):
# User-defined tzinfo classes may return absolutely anything.
# Hence this paranoid conversion to create a valid cache key.
tz_name = force_text(get_current_timezone_name(), errors='ignore')
cache_key += '.%s' % tz_name.encode('ascii', 'ignore').replace(' ', '_')
cache_key += '.%s' % tz_name.encode('ascii', 'ignore').decode('ascii').replace(' ', '_')
return cache_key

def _generate_cache_key(request, method, headerlist, key_prefix):
Expand Down
4 changes: 2 additions & 2 deletions tests/regressiontests/cache/tests.py
Expand Up @@ -1308,7 +1308,7 @@ def test_cache_key_i18n_timezone(self):
# This is tightly coupled to the implementation,
# but it's the most straightforward way to test the key.
tz = force_text(timezone.get_current_timezone_name(), errors='ignore')
tz = tz.encode('ascii', 'ignore').replace(' ', '_')
tz = tz.encode('ascii', 'ignore').decode('ascii').replace(' ', '_')
response = HttpResponse()
key = learn_cache_key(request, response)
self.assertIn(tz, key, "Cache keys should include the time zone name when time zones are active")
Expand All @@ -1320,7 +1320,7 @@ def test_cache_key_no_i18n (self):
request = self._get_request()
lang = translation.get_language()
tz = force_text(timezone.get_current_timezone_name(), errors='ignore')
tz = tz.encode('ascii', 'ignore').replace(' ', '_')
tz = tz.encode('ascii', 'ignore').decode('ascii').replace(' ', '_')
response = HttpResponse()
key = learn_cache_key(request, response)
self.assertNotIn(lang, key, "Cache keys shouldn't include the language name when i18n isn't active")
Expand Down

0 comments on commit 45baaab

Please sign in to comment.