python 操作 mysql (简单好用的封装)

@[TOC](python 操作 mysql (简单好用的封装))

本文需要使用 pymysql 包

导包(使用代理网址，速度快些)

pip install  -i https://pypi.douban.com/simple PyMySQL

工具类的封装

	工具简单的封装查询和修改方法，revise 主要执行对数据库的增删改查

import pymysql


class SqlUtil(object):

    def __init__(self,host,port,username,password,db):
        self.db = pymysql.connect(host=host, port=port, user=username, passwd=password,db=db, charset='utf8')

    def query(self, sql):
        cursor = self.db.cursor()
        cursor.execute(sql)
        return cursor.fetchall()

    def revise(self, sql):
        try:
            cursor = self.db.cursor()
            cursor.execute(sql)  # 执行SQL语句
            self.db.commit()  # 提交到数据库执行
            return True
        except:
            self.db.rollback()  # 发生错误时回滚
            return False

    def close(self):
        self.db.close()

使用

import SqlUtil
#  导入你的工具类文件，如在别的目录，使用from .. import SqlUtil


host='localhost'
port = 3306
username = 'root'
password = 'root'
db='db1'

su=SqlUtil(host,port,username,password,db)
sql='select * from t'
result = su.query(sql)
print(result)

sql1 = 'delete from t where id = 1'
result1=su.revise(sql1)
print(result1)

su.close()