使MySQLdb的SQL查詢中支持?及:xxx的查詢方式
來源：程序員人生 發布時間：2014-02-05 16:47:36 閱讀次數：4024次
MySQLdb中的sql的語法目前支持如下語法：
insert into users(id, name, domain, created_at)values(%s, %s, %s, %s)
但很多的一些通用的sql的語法卻是下面這樣
insert into users(id, name, domain, created_at)values(?, ?, ?, ?)
insert into users(id, name, domain, created_at)values(:id, :name, :domain, :created_at)
所以我對MySQLdb的Cursor進行了擴展，使其能夠支持帶?以及:xxx的查詢方式
Python代碼：
class BaseCursor(MySQLdb.cursors.BaseCursor):
def _get_query_parameters(self, query, params):
if isinstance(params, dict) and params:
p = re.compile(':w+')
return (p.sub('%s', query),
[params[param_token[1:]] for param_token in p.findall(query)])
else:
return (query.replace('?', '%s'), params)
def execute(self, query, args=None):
(query, params) = self._get_query_parameters(query, args)
return MySQLdb.cursors.BaseCursor.execute(self, query, params)
class Cursor(CursorStoreResultMixIn, CursorTupleRowsMixIn,
BaseCursor):
#調用的代碼如下：
cursor=Cursor(self._dbconn)
try:
cursor.execute('insert into users(name, domain)values(:name, :domain)', dict(name='xxx', domain='hahaha'))
return cursor.lastrowid
finally:
cursor.close()
