1.sql语句参数化
创建testInsertParam.py文件,向学生表中插入一条数据
#encoding=utf-8
import pymysql
try:
conn=pymysql.connect(host='localhost',port=3306,db='test1',user='root',passwd='mysql',charset='utf8')
cs1=conn.cursor()
students_n=input("请输入学生姓名:")
params=[students_n] #将输入的要操作的值放入列表中
count=cs1.execute('insert into students(sname) values(%s)',params)
#不管values对应的值是什么类型,都用%s占位符表示
print(count)
conn.commit()
cs1.close()
conn.close()
except Exception as e:
print(e.message)
2.封装
观察前面的文件发现,除了sql语句及参数不同,其它语句都是一样的。为了方便,我们可以创建MysqlHelper.py文件,定义类,将这些重复性语句进行封装。
#encoding=utf8
import pymysql
class MysqlHelper():
#定义初始化函数
def __init__(self,host,port,db,user,passwd,charset='utf8'):
self.host=host
self.port=port
self.db=db
self.user=user
self.passwd=passwd
self.charset=charset
#定义连接函数,用于创建连接对象和游标对象
def connect(self):
self.conn=pymysql.connect(host=self.host,port=self.port,db=self.db,user=self.user,passwd=self.passwd,charset=self.charset)
self.cursor=self.conn.cursor()
#定影关闭函数,用于关闭连接对象和游标对象
def close(self):
self.cursor.close()
self.conn.close()
#获取一条查询数据
def get_one(self,sql,params=()):
result=None
try:
self.connect()
self.cursor.execute(sql, params)
result = self.cursor.fetchone()
self.close()
#此处的close()是调用了本类中的close()函数
except Exception, e:
print e.message
return result
#获取多条查询数据
def get_all(self,sql,params=()):
list=()
#定义list为空元组
try:
self.connect()
self.cursor.execute(sql,params)
list=self.cursor.fetchall()
self.close()
except Exception,e:
print e.message
return list
#定义编辑函数,用于对数据库的增删改操作
def __edit(self,sql,params):
count=0
try:
self.connect()
count=self.cursor.execute(sql,params)
self.conn.commit()
self.close()
except Exception,e:
print e.message
return count
def insert(self,sql,params=()):
return self.__edit(sql,params)
def delete(self, sql, params=()):
return self.__edit(sql, params)
def update(self, sql, params=()):
return self.__edit(sql, params)
3.调用封装类完成数据库操作
(1)创建testInsertWrap.py文件,使用封装好的帮助类完成插入操作
#encoding=utf8
from MysqlHelper import *
sql='insert into students(sname,gender) values(%s,%s)'
sname=input("请输入用户名:")
gender=input("请输入性别,1为男,0为女")
params=[sname,bool(gender)]
mysqlHelper=MysqlHelper('localhost',3306,'test1','root','mysql')
count=mysqlHelper.insert(sql,params)
if count==1:
print('ok')
else:
print('error')
(2)创建testGetOneWrap.py文件,使用封装好的帮助类完成查询最新一行数据操作#encoding=utf8
from MysqlHelper import *
sql='select sname,gender from students order by id desc'
helper=MysqlHelper('localhost',3306,'test1','root','mysql')
one=helper.get_one(sql)
print(one)