Merge pull request #42 from gh640/fix_no_attribute_MIDDLEWARE_CLASSES…

…_error Issue #41: Fix MIDDLEWARE_CLASSES error with `runserver`
HassenPy · Mar 1, 2018 · 1652b7a · 1652b7a
2 parents a192e51 + 360bffb
commit 1652b7a
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/django_pdb/management/commands/runserver.py b/django_pdb/management/commands/runserver.py
@@ -54,10 +54,13 @@ def handle(self, *args, **options):
         pdb_option = options.pop('pdb')
         ipdb_option = options.pop('ipdb')
 
-        middleware = 'django_pdb.middleware.PdbMiddleware'
+        pdb_middleware = 'django_pdb.middleware.PdbMiddleware'
+        middleware = (settings.MIDDLEWARE
+                      if hasattr(settings, 'MIDDLEWARE')
+                      else settings.MIDDLEWARE_CLASSES)
         if ((pdb_option or settings.DEBUG)
-            and middleware not in settings.MIDDLEWARE_CLASSES):
-            settings.MIDDLEWARE_CLASSES += (middleware,)
+            and pdb_middleware not in middleware):
+            middleware += (pdb_middleware,)
 
         self.pm = options.pop('pm')
         if self.pm: