pymysql编写脚本

import pymysql

class DataBaseHandle(object):
''' 定义一个 MySQL 操作类'''

def __init__(self, host, port, user, password, database):
'''初始化数据库信息并创建数据库连接'''
# 下面的赋值其实可以省略，connect 时 直接使用形参即可
self.connnect = pymysql.connect(
host=host,
user=user,
password=password,
db=database,
port=port,
charset="utf8"
)
self.cursor = self.connnect.cursor(pymysql.cursors.DictCursor)

# 插入操作
def insertDB(self, sql):
''' 插入数据库操作 '''
try:
# 执行sql
self.cursor.execute(sql)
# tt = self.cursor.execute(sql) # 返回 插入数据 条数 可以根据 返回值 判定处理结果
# print(tt)
self.connnect.commit()

# query = 'insert into 表名(列名1, 列名2, 列名3, 列名4, 列名5, 列名6) values(%s, %s, %s, %s, %s, %s)'
# self.cursor.execute(sql)

except:
# 发生错误时回滚
self.connnect.rollback()
finally:
self.cursor.close()

# 删除操作
def deleteDB(self, sql):
''' 操作数据库数据删除 '''

try:
# 执行sql
self.cursor.execute(sql)
# tt = self.cursor.execute(sql) # 返回 删除数据 条数 可以根据 返回值 判定处理结果
# print(tt)
self.connnect.commit()
except:
# 发生错误时回滚
self.connnect.rollback()
finally:
self.cursor.close()

# 修改操作
def updateDb(self, sql):
''' 更新数据库操作 '''
try:
# 执行sql
self.cursor.execute(sql)
# tt = self.cursor.execute(sql) # 返回 更新数据 条数 可以根据 返回值 判定处理结果
# print(tt)
self.connnect.commit()
except:
# 发生错误时回滚
self.connnect.rollback()
finally:
self.cursor.close()

# 查询操作
def selectDb(self, sql):
''' 数据库查询 '''
try:
self.cursor.execute(sql) # 返回 查询数据 条数 可以根据 返回值 判定处理结果
data = self.cursor.fetchall() # 返回所有记录列表
# print(data)
return data