# HG changeset patch
# User Marcin Kuzminski <marcin@python-works.com>
# Date 1274403691 -7200
# Node ID 83c7ee1b5f5c1918d2f82daa2e74024b4b9f1e86
# Parent  52bbeb1e813fddb443506bd035752817e1661651
improved timerproxy with sqllogging, and new way of sqlformat queries

diff -r 52bbeb1e813f -r 83c7ee1b5f5c pylons_app/lib/timerproxy.py
--- a/pylons_app/lib/timerproxy.py	Fri May 21 02:44:40 2010 +0200
+++ b/pylons_app/lib/timerproxy.py	Fri May 21 03:01:31 2010 +0200
@@ -10,18 +10,28 @@
     normal = '\x1b[0m'
     return COLOR_SEQ % COLOR_SQL + sql + normal 
 
+def one_space_trim(s):
+    if s.find("  ") == -1:
+        return s
+    else:
+        s = s.replace('  ', ' ')
+        return one_space_trim(s)
+    
 def format_sql(sql):
     sql = color_sql(sql)
-    sql = sql.replace('SELECT', '\n    SELECT \n\t')\
-        .replace('FROM', '\n    FROM')\
-        .replace('ORDER BY', '\n    ORDER BY')\
-        .replace('LIMIT', '\n    LIMIT')\
-        .replace('WHERE', '\n    WHERE')\
-        .replace('AND', '\n    AND')\
-        .replace('LEFT', '\n    LEFT')\
-        .replace('INNER', '\n    INNER')\
-        .replace('INSERT', '\n    INSERT')\
-        .replace('DELETE', '\n    DELETE')
+    sql = sql.replace('\n', '')
+    sql = one_space_trim(sql)
+    sql = sql\
+        .replace('SELECT', '\n\tSELECT \n\t')\
+        .replace('FROM', '\n\tFROM')\
+        .replace('ORDER BY', '\n\tORDER BY')\
+        .replace('LIMIT', '\n\tLIMIT')\
+        .replace('WHERE', '\n\tWHERE')\
+        .replace('AND', '\n\tAND')\
+        .replace('LEFT', '\n\tLEFT')\
+        .replace('INNER', '\n\tINNER')\
+        .replace('INSERT', '\n\tINSERT')\
+        .replace('DELETE', '\n\tDELETE')
     return sql