Python连接mysql数据库

Posted 2021-02-10 帅瓶子

from mysql import connector
from Agin_Project.unittest_again.common.file_path import FilePath
from Agin_Project.unittest_again.common.read_yaml import Read_Yaml
class MySql_db_Connect:

    def mysql_connect(self, sql):
        # 获取数据库的文件路径
        file = FilePath().file_path("mysql.yaml")
        # 读取数据
        mysql_data = Read_Yaml().read_yaml(file)
        # 建立数据库连接
        mysql = connector.connect(**mysql_data)
        # 新建一个游标，用来执行sql语句
        cursor = mysql.cursor()
        # 执行sql语句
        cursor.execute(sql)
        # 获取结果
        #data =cursor.fetchall() # 返回的数据类型为嵌套元祖的列表
        data =cursor.fetchone() # 返回的数据类型为元祖
        # 关闭游标
        cursor.close()
        # 关闭数据库
        mysql.close()

　　  return data

　　

这里我使用的yaml配置文件来存放数据的连接信息，这样即使我们数据有所改动，也不需要改代码，直接修改配置文件就可以了。
安装mysql模块用到的命令： pip install mysql-connector
欢迎大家一起交流学习，群号 490327371