feat: collapse whitespace in multiline sql

project_runpy/heidi.py

@@ -114,7 +114,10 @@ class ReadableSqlFilter(logging.Filter):
 
     def filter(self, record):
         # https://github.com/django/django/blob/febe136d4c3310ec8901abecca3ea5ba2be3952c/django/db/backends/utils.py#L106-L131
-        duration, sql, *__ = record.args
+        duration, sql, *record_args = record.args
+        if sql and "\n" in sql[:28]:
+            sql = " ".join(sql.strip().split())
+            record.args = (duration, sql, *record_args)
         if not sql or "SELECT" not in sql[:28]:
             # WISHLIST what's the most performant way to see if 'SELECT' was
             # used?

test_project_runpy.py

@@ -176,6 +176,25 @@ def test_filter_formats_ignores_select_without_from(self):
         self.assertTrue(logging_filter.filter(record))
         self.assertEqual(original_sql, record.args[1])
 
+    def test_filter_collapses_multiline_sql(self):
+        long_sql = """
+            (0.002)
+                SELECT VERSION(),
+                       @@sql_mode,
+                       @@default_storage_engine,
+                       @@sql_auto_is_null,
+                       @@lower_case_table_names,
+                       CONVERT_TZ('2001-01-01 01:00:00', 'UTC', 'UTC') IS NOT NULL
+            ; args=None
+            """
+        logging_filter = ReadableSqlFilter()
+        record = mock.MagicMock(args=(1.0, long_sql))
+        self.assertIn("\n", record.args[1])
+
+        self.assertTrue(logging_filter.filter(record))
+
+        self.assertNotIn("\n", record.args[1])
+
 
 if __name__ == "__main__":
     unittest.main()