1.首先先安装模块:
pip3 install pymysql
2.导入模块,定义函数:
vim pymysql.py
#/usr/bin/python3
#encoding:utf-8
import pymysql
# 封装SQL语句函数
def db_sql(sql,m='r'): #这里默认是读
#这里填写连接数据库的地址,用户,密码,和所连接的库
db = pymysql.connect(host='127.0.0.1', user='root', passwd='123', db='test')
cursor = db.cursor()
try:
cursor.execute(sql)
if m == 'r':
data = cursor.fetchall ()
elif m == 'w':
db.commit ()
data = cursor.rowcount
except:
data = False
db.rollback()
db.close()
return data
3.读取数据库内容:
name = 填写name字段信息
table = 填写表名
#将查询内容存到变量data中,f能将table和name读取到变量信息,而不是字符串,m的值没有填写,默认为读
data = db_sql(f'select id,name from {table} where name={name}')
#打印查询内容
print(data)
4.修改数据库内容:
idd = 1
name = 小明
table = test
#这次使用format读取变量,m=‘w’代表是写入操作,将test表中id为1的name字段改成小明,
res = db_sql('update {} set name={} where id={}'.format(fname,name,idd),m='w')
#判断是否成功
if res:
print("更新成功")
else:
print("未更新")