Minor refactoring

stamparm · stamparm · commit 605d73cc3d4a · 2012-11-29T12:21:12.000+01:00
diff --git a/lib/request/inject.py b/lib/request/inject.py
@@ -397,7 +397,7 @@ def getValue(expression, blind=True, union=True, error=True, time=True, fromUser
             found = False
             count = 0
 
-            if query and not 'COUNT(*)' in query:
+            if query and not re.search(r"COUNT.*FROM.*\(.*DISTINCT", query, re.I):
                 query = query.replace("DISTINCT ", "")
 
             if not conf.forceDns:
