Modernize the CSRF migration helper.

Thanks Lennart Regebro for spotting that code.

1 files 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):
     """