Purge keys less dumber

Author: twneale

sunlight/debugcache.py

@@ -73,8 +73,10 @@ def set(self, key, val):
 
     def purge(self, *keys):
         if not keys:
-            return self.mongo.drop_database()
-        self.mongo.reponses.remove({'_id': {'$in': keys}})
+            spec = {}
+        else:
+            spec = {'_id': {'$in': keys}}
+        self.mongo.reponses.remove(keys)
 
 
 class BaseCache(object):
@@ -85,6 +87,15 @@ def __init__(self):
             format='%(asctime)s %(levelname)s %(message)s')
         self.logger = logging.getLogger('cache')
 
+    def set_backend(self, backend_name):
+        try:
+            self.backend = backends[backend_name]()
+            self.logger.info('Changed cache backend to %r.' % self.backend)
+        except KeyError:
+            raise ValueError('No backend named %r is defined.' % backend_name)
+
+    enable = set_backend
+
     def disable(self):
         '''Disable the cache. Will wipe out an in-memory cache.
         '''
@@ -98,15 +109,6 @@ def purge(self):
         self.backend.purge()
         self.logger.info('...done.')
 
-    def set_backend(self, backend_name):
-        try:
-            self.backend = backends[backend_name]()
-            self.logger.info('Changed cache backend to %r.' % self.backend)
-        except KeyError:
-            raise ValueError('No backend named %r is defined.' % backend_name)
-
-    enable = set_backend
-
     def get_key(self, *args, **kwargs):
         '''Create a cache key based on the input to the wrapped callable.
         '''