Minor bug fix

bdamele · bdamele · commit 14219a3dac92 · 2011-04-10T22:44:08.000Z
diff --git a/lib/core/agent.py b/lib/core/agent.py
@@ -329,7 +329,7 @@ def nullCastConcatFields(self, fields):
         if not Backend.getDbms():
             return fields
 
-        if fields.startswith("(CASE") or fields.startswith("SUBSTR"):
+        if fields.startswith("(CASE") or fields.startswith("SUBSTR") or fields.startswith("MID("):
             nulledCastedConcatFields = fields
         else:
             fields = fields.replace(", ", ",")
@@ -369,7 +369,7 @@ def getFields(self, query):
         fieldsSelectFrom = getCompiledRegex("\ASELECT%s\s+(.+?)\s+FROM\s+" % prefixRegex, re.I).search(query)
         fieldsExists = getCompiledRegex("EXISTS(.*)", re.I).search(query)
         fieldsSelect = getCompiledRegex("\ASELECT%s\s+(.*)" % prefixRegex, re.I).search(query)
-        fieldsSubstr = getCompiledRegex("\ASUBSTR", re.I).search(query)
+        fieldsSubstr = getCompiledRegex("\A(SUBSTR|MID\()", re.I).search(query)
         fieldsNoSelect = query
 
         if fieldsSubstr:
