Merge pull request #62 from mattrobenolt/filtered-queries

Added support to filter queries based on regex patterns

Author: dcramer

README.rst

@@ -117,6 +117,11 @@ Outputs queries as they happen to the terminal, including time taken.
 Disable SQL query truncation (used in SQLRealTimeModule) with the ``DEVSERVER_TRUNCATE_SQL`` setting::
   
 	DEVSERVER_TRUNCATE_SQL = False
+Filter SQL queries with the ``DEVSERVER_FILTER_SQL`` setting::
+  
+	DEVSERVER_FILTER_SQL = (
+		re.compile('djkombu_\w+'),  # Filter all queries related to Celery
+	)
 
 devserver.modules.sql.SQLSummaryModule
 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

devserver/modules/sql.py

@@ -61,10 +61,14 @@ def execute(self, sql, params=()):
         formatted_sql = sql % (params if isinstance(params, dict) else tuple(params))
         if self.logger:
             message = formatted_sql
-            if settings.DEVSERVER_TRUNCATE_SQL:
-                message = truncate_sql(message, aggregates=settings.DEVSERVER_TRUNCATE_AGGREGATES)
-            message = sqlparse.format(message, reindent=True, keyword_case='upper')
-            self.logger.debug(message)
+            if settings.DEVSERVER_FILTER_SQL:
+                if any(filter_.search(message) for filter_ in settings.DEVSERVER_FILTER_SQL):
+                    message = None
+            if message is not None:
+                if settings.DEVSERVER_TRUNCATE_SQL:
+                    message = truncate_sql(message, aggregates=settings.DEVSERVER_TRUNCATE_AGGREGATES)
+                message = sqlparse.format(message, reindent=True, keyword_case='upper')
+                self.logger.debug(message)
 
         start = datetime.now()
 

devserver/settings.py

@@ -10,6 +10,7 @@
     # 'devserver.modules.cache.CacheSummaryModule',
 ))
 
+DEVSERVER_FILTER_SQL = getattr(settings, 'DEVSERVER_FILTER_SQL', False)
 DEVSERVER_TRUNCATE_SQL = getattr(settings, 'DEVSERVER_TRUNCATE_SQL', True)
 
 DEVSERVER_TRUNCATE_AGGREGATES = getattr(settings, 'DEVSERVER_TRUNCATE_AGGREGATES', getattr(settings, 'DEVSERVER_TRUNCATE_AGGREGATES', False))