python连接mysql

1、安装pymysql

pip install pymysql

2、连接数据库，创建游标

# 连接数据库
con = pymysql.connect(host='mysql数据库所在主机ip',
                      port=3306,
                      user='root',
                      password='123456',
                      database='test',
                      charset='utf8',
                      cursorclass=pymysql.cursors.DictCursor
                      )
# 创建游标
cur = con.cursor()

3、常用操作

3.1 执行sql

# 执行sql
cur.execute("select * from books")

3.2 获取数据

# 获取第一行数据
res = cur.fetchone()
print(res)

# 获取前n行数据
res2 = cur.fetchmany(5)
print(res2)

# 获取所有数据
res3 = cur.fetchall()
print(res3)

3.3 连接提交

con.commit()

4、关闭游标、断开连接

# 关闭游标
cur.close()
# 关闭连接
con.close()

封装成DB工具类

import pymysql

class DB:

    def __init__(self):
        """连接数据库，创建游标"""
        self.con = pymysql.connect(host='mysql数据库所在主机ip',
                                   port=3306,
                                   user='root',
                                   password='123456',
                                   database='test',
                                   charset='utf8',
                                   cursorclass=pymysql.cursors.DictCursor
                                   )
        self.cur = self.con.cursor()

    def query_sql(self, sql):
        """执行查询SQL"""
        self.cur.execute(sql)
        return self.cur.fetchall()

    def excute_sql(self, sql):
        """提交增删改SQL"""
        self.cur.execute(sql)
        self.con.commit()

    def close(self):
        """关闭游标、断开连接"""
        self.cur.close()
        self.con.close()