From: Jason Kirtland <jek@discorporate.us>
Date: Thu, 9 Aug 2007 22:10:16 +0000 (+0000)
Subject: Allow '$' in bind param detection [ticket:719], added test suite & fixed an edge... 
X-Git-Tag: rel_0_4beta1~21
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=ce0d72e68aac6ad6c7569ffe76ca5c6c02d2718c;p=thirdparty%2Fsqlalchemy%2Fsqlalchemy.git

Allow '$' in bind param detection [ticket:719], added test suite & fixed an edge case
---

diff --git a/lib/sqlalchemy/ansisql.py b/lib/sqlalchemy/ansisql.py
index 94d69a8cef..3d5e917cf4 100644
--- a/lib/sqlalchemy/ansisql.py
+++ b/lib/sqlalchemy/ansisql.py
@@ -45,8 +45,8 @@ LEGAL_CHARACTERS = util.Set(string.ascii_lowercase +
                             string.digits + '_$')
 ILLEGAL_INITIAL_CHARACTERS = util.Set(string.digits + '$')
 
-BIND_PARAMS = re.compile(r'(?<![:\w\x5c]):(\w+)(?!:)', re.UNICODE)
-BIND_PARAMS_ESC = re.compile(r'\x5c(:\w+)(?!:)', re.UNICODE)
+BIND_PARAMS = re.compile(r'(?<![:\w\$\x5c]):([\w\$]+)(?![:\w\$])', re.UNICODE)
+BIND_PARAMS_ESC = re.compile(r'\x5c(:[\w\$]+)(?![:\w\$])', re.UNICODE)
 
 OPERATORS =  {
     operator.and_ : 'AND',
diff --git a/test/sql/query.py b/test/sql/query.py
index 48a28a9a56..4f569f1c02 100644
--- a/test/sql/query.py
+++ b/test/sql/query.py
@@ -1,7 +1,7 @@
 import testbase
 import datetime
 from sqlalchemy import *
-from sqlalchemy import exceptions
+from sqlalchemy import exceptions, ansisql
 from testlib import *
 
 
@@ -164,6 +164,37 @@ class QueryTest(PersistTest):
         s = users.select(users.c.user_name==u)
         r = s.execute(someshortname='fred').fetchall()
         assert len(r) == 1
+
+    def test_bindparam_detection(self):
+        dialect = ansisql.ANSIDialect(default_paramstyle='qmark')
+        prep = lambda q: dialect.compile(sql.text(q)).string
+
+        def a_eq(got, wanted):
+            if got != wanted:
+                print "Wanted %s" % wanted
+                print "Received %s" % got
+            self.assert_(got == wanted)
+
+        a_eq(prep('select foo'), 'select foo')
+        a_eq(prep("time='12:30:00'"), "time='12:30:00'")
+        a_eq(prep(u"time='12:30:00'"), u"time='12:30:00'")
+        a_eq(prep(":this:that"), ":this:that")
+        a_eq(prep(":this :that"), "? ?")
+        a_eq(prep("(:this),(:that :other)"), "(?),(? ?)")
+        a_eq(prep("(:this),(:that:other)"), "(?),(:that:other)")
+        a_eq(prep("(:this),(:that,:other)"), "(?),(?,?)")
+        a_eq(prep("(:that_:other)"), "(:that_:other)")
+        a_eq(prep("(:that_ :other)"), "(? ?)")
+        a_eq(prep("(:that_other)"), "(?)")
+        a_eq(prep("(:that$other)"), "(?)")
+        a_eq(prep("(:that$:other)"), "(:that$:other)")
+        a_eq(prep(".:that$ :other."), ".? ?.")
+
+        a_eq(prep(r'select \foo'), r'select \foo')
+        a_eq(prep(r"time='12\:30:00'"), r"time='12\:30:00'")
+        a_eq(prep(":this \:that"), "? :that")
+        a_eq(prep(r"(\:that$other)"), "(:that$other)")
+        a_eq(prep(r".\:that$ :other."), ".:that$ ?.")
         
     def testdelete(self):
         users.insert().execute(user_id = 7, user_name = 'jack')