自己写的Python数据库连接类和sql语句拼接方法

这个工具类十分简单和简洁。

sql拼接方法

# encoding=utf-8
from django.http import HttpResponse
from anyjson import serialize
from django.http import HttpResponse
from anyjson import serialize
import MySQLdb def safe(s):
return MySQLdb.escape_string(s) def get_i_sql(table, dict):
'''
生成insert的sql语句
@table,插入记录的表名
@dict,插入的数据,字典
'''
sql = 'insert into %s set ' % table
sql += dict_2_str(dict)
return sql def get_s_sql(table, keys, conditions, isdistinct=0):
'''
生成select的sql语句
@table,查询记录的表名
@key,需要查询的字段
@conditions,插入的数据,字典
@isdistinct,查询的数据是否不重复
'''
if isdistinct:
sql = 'select distinct %s ' % ",".join(keys)
else:
sql = 'select %s ' % ",".join(keys)
sql += ' from %s ' % table
if conditions:
sql += ' where %s ' % dict_2_str_and(conditions)
return sql def get_u_sql(table, value, conditions):
'''
生成update的sql语句
@table,查询记录的表名
@value,dict,需要更新的字段
@conditions,插入的数据,字典
'''
sql = 'update %s set ' % table
sql += dict_2_str(value)
if conditions:
sql += ' where %s ' % dict_2_str_and(conditions)
return sql def get_d_sql(table, conditions):
'''
生成detele的sql语句
@table,查询记录的表名 @conditions,插入的数据,字典
'''
sql = 'delete from %s ' % table
if conditions:
sql += ' where %s ' % dict_2_str_and(conditions)
return sql def dict_2_str(dictin):
'''
将字典变成,key='value',key='value' 的形式
'''
tmplist = []
for k, v in dictin.items():
tmp = "%s='%s'" % (str(k), safe(str(v)))
tmplist.append(' ' + tmp + ' ')
return ','.join(tmplist) def dict_2_str_and(dictin):
'''
将字典变成,key='value' and key='value'的形式
'''
tmplist = []
for k, v in dictin.items():
tmp = "%s='%s'" % (str(k), safe(str(v)))
tmplist.append(' ' + tmp + ' ')
return ' and '.join(tmplist)

数据库连接类

class SqlConn():
def __init__(self):
self.conn= DBpool.pool.connection()
self.cur=self.conn.cursor()
def cur(self):
return self.cur()
def commit(self):
self.conn.commit()
def execute(self,sql,fetchone=0):
self.cur.execute(sql)
return self.cur.fetchone() if fetchone else self.cur.fetchall()
def last_id(self,table):
sql='SELECT LAST_INSERT_ID() from %s'%table
return self.execute(sql,1)[0]
def close(self):
self.cur.close()
self.conn.close()

self.conn= DBpool.pool.connection()
中,我用的是连接池连接,这里可以改成普通的数据库连接

查询结果解析方法

def fSqlResult(r,key_list):
#r @tuple 数据库fetchall的结果
#key_list @list 查询字段的keys
# format SQL Result 格式化数据库查询的结果,转化成包含多个字典的列表格式,即((1,2),(3,4))->[{"key1":1,"key2":2},{"key1":3,"key2":4}]
#返回 @dict 查询结果
mlist=[]
l=len(key_list)
if r:
for item in r:
tmp={}
for i in range(l):
tmp[key_list[i]]=str(item[i])
mlist.append(tmp)
return mlist

使用方法:

例如要执行sql语句:select username,pwd,type,age,birth from user where username='abc';

conn=SqlConn()
key_list=['username','pwd','type','age','birth']
sql=s_sql('user',key_list,{'username':'abc'}
r=conn.execute(sql)
data=fSqlResult(r,key_list)
conn.close()

data的格式为[{'username':'i am username','pwd':'i am pwd','type':'student','age':12,'birth':'2014-7-2'}]
方便快捷,适用于经常需要用到数据库的CRUD连接的应用。

上一篇:TCP & UDP & IP


下一篇:cocos2dx调用浏览器打开网址