From c6964feb54a15e87de5e0390d07fe5fff69729c7 Mon Sep 17 00:00:00 2001 From: Aymeric Augustin Date: Fri, 17 May 2013 18:15:08 +0200 Subject: [PATCH] Modernize the CSRF migration helper. Thanks Lennart Regebro for spotting that code. --- extras/csrf_migration_helper.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/extras/csrf_migration_helper.py b/extras/csrf_migration_helper.py index 62a8fb92af..2a8853494c 100755 --- a/extras/csrf_migration_helper.py +++ b/extras/csrf_migration_helper.py @@ -143,7 +143,7 @@ def get_template_dirs(): """ from django.conf import settings dirs = set() - if ('django.template.loaders.filesystem.load_template_source' in settings.TEMPLATE_LOADERS + if ('django.template.loaders.filesystem.load_template_source' in settings.TEMPLATE_LOADERS or 'django.template.loaders.filesystem.Loader' in settings.TEMPLATE_LOADERS): dirs.update(map(unicode, settings.TEMPLATE_DIRS)) @@ -281,12 +281,10 @@ def search_python_list(python_code, template_names): Returns a list of tuples, each one being: (filename, line number) """ - retval = [] + retval = set() for tn in template_names: - retval.extend(search_python(python_code, tn)) - retval = list(set(retval)) - retval.sort() - return retval + retval.update(search_python(python_code, tn)) + return sorted(retval) def search_python(python_code, template_name): """