Python之pymysql的学习总结

最近学习了pymysql的基本操作：增删改查，这里做一个封装。

import pymysql as ps
class MysqlHelper:
    def __init__(self, host, user, password, database, charset):
        self.host = host
        self.user = user
        self.password = password
        self.database = database
        self.charset = charset
        self.db = None
        self.curs = None
        '''
        这里写的是MysqlHepler的构造函数，方便链接数据库，只要创建对象的时候把参数传入即可
        host：连接的主机地址
        user:你的MySql的用户一般是root
        password:安装数据库时写的地址
        database：你要操作的数据库的名字
        charset:就是数据库的编码格式
       '''

    def connect(self):
        self.db = ps.connect(host=self.host, user=self.user, password=self.password, database=self.database,charset=self.charset)
        #连接数据库
        self.curs = self.db.cursor(ps.cursors.DictCursor)#查询时的返回值设置为字典的类型

    def close(self):
        self.curs.close()
        self.db.close()
        #关闭游标和数据库

    def find(self, sql):
        self.connect()  #获取连接
        try:
            result = self.curs.execute(sql) #通过游标执行sql语句
            fc = self.curs.fetchall()      #拿到查询的所有结果  如果只要一条就写 ：self.curs.fetchone()
            print("Success")
            return fc
        except:
            print('find error')
        finally:
            self.close()

    def update(self, sql):     #因为数据库 的增，删，改 不需要返回值，只要执行sql语句所以就写一个方法
        self.connect()         #获取连接
        try:
            count = self.curs.execute(sql)  #通过游标执行sql语句
            self.db.commit()           #提交处理
            print("Success")
            return count
        except:
            print('update error')
            self.db.rollback()     #失败的话数据回滚

    def update_by_params(self, sql, params):
        self.connect()
        try:
            count = self.curs.execute(sql%params)   #通过游标执行sql语句 ，处理SQL的拼接
            self.db.commit()
            print("ok")
            return count
        except:
            print('update_by_params error')
            self.db.rollback()
        '''
        这里写的另一种格式的sql语句，为的是方便传变量进SQL语句中
        update(self, sql)的sql写法是：
            sql=insert into Student(Sname,Sid,Ssex,Sage) values('xiaoming','001',1,30)
        update_by_params(self, sql, params)的sql写法是：
            sql=insert into Student(Sname,Sid,Ssex,Sage) values('%s','%s',%d,%d)
            update_by_params（sql,('xiaoming','001',1,30)）
       '''

下面来运行一下看看效果：

首先我在数据库建立了一张表如图，其中id字段是自动增长的。

 

1.用update方法插入数据

from PyMySql.mySQLHelper import MysqlHelper
#上面的是导入类
host='localhost'
user='root'
password='xxx'#这是你的密码
database='xxx'#这是你的数据库
charset='utf8'

mh = MysqlHelper(host,user,password,database,charset)
sql="insert into Student(Sname,Sid,Ssex,Sage) values('xiaoming','001',0,18)"
mh.update(sql)

2.用update_by_params方法插入数据

mh = MysqlHelper(host,user,password,database,charset)
sql="insert into Student(Sname,Sid,Ssex,Sage) values('%s','%s',%d,%d)"
name='xiaowang'
id='002'
sex=1
age=18
mh.update_by_params(sql,(name,id,sex,age))

插入的效果：

类似的删和改都是一样的

 3.查找数据

mh = MysqlHelper(host,user,password,database,charset)
sql="select * from Student"
data = mh.find(sql)
print(data)

查找的效果：

返回的是字典，因为之前在游标处设置了

参考的资料：

http://www.runoob.com/python3/python3-mysql.html