程序员人生 网站导航

使MySQLdb的SQL查询中支持?及:xxx的查询方式

栏目:综合技术时间:2014-02-05 16:47:36

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()
------分隔线----------------------------
------分隔线----------------------------

最新技术推荐