第一步: pip install pymysql
第二步: 如下是连接数据库代码:
import pymysql
class ConnectDatabase:
# 封装一个数据库类
# 用于数据库查询/删除/更新/插入
def __init__(self,
host="xxx.xx.xx.xx", # 数据库地址
port=3306, # 端口
user='xxxx',
password='xxxx', # 登录数据库密码
database='xxxx', # 数据库名
charset='utf8'): # 编码
# 第一步:连接数据库
self.connect = pymysql.connect(host=host, user=user, password=password,
database=database, port=int(port), charset=charset)
# 第二步:创建游标
self.cur = self.connect.cursor(cursor=pymysql.cursors.DictCursor)
# 执行sql语句,如果是查询语句,则返回第一条
def run_sql(self, sql_param):
if sql_param.strip().upper().startswith('SELECT'): # 判断是否是select查询语句
self.cur.execute(sql_param)
# fetchone只返回一条数据 ;fetchmany(5) 只返回5条数据 ;fetchall() 返回所有数据
alldata = self.cur.fetchone()
# 此处并非是提交事务,主要是用于数据库数据刷新,源于接口请求之后,数据库可能会数据更新,如果此时不更新数据表,
# 那么查询到的数据依旧是老数据,非更新数据
self.connect.commit()
return alldata
else:
# 如果是insert,update,delete语句,那么需要进行提交事务,否则操作失败
self.cur.execute(sql_param)
self.connect.commit() #
return None
def colse(self):
self.cur.close()
self.connect.close()