python连接mysql数据库

最新推荐文章于 2024-07-10 09:15:55 发布

monkey-怡宝

最新推荐文章于 2024-07-10 09:15:55 发布

阅读量93

点赞数

分类专栏：接口自动化系列文章标签： python Powered by 金山文档

本文链接：https://blog.csdn.net/weixin_42661815/article/details/128649765

版权

接口自动化系列专栏收录该内容

5 篇文章 0 订阅

订阅专栏

import pymysql


# 获取数据库连接
from service.vr_whms import mysql_info

db = pymysql.connect(
    host=mysql_info['host'],
    port=mysql_info['port'],
    user=mysql_info['user'],
    password=mysql_info['password'],
    charset='utf8',
    database=mysql_info['database']
    )

cursor = db.cursor()


# 查询一条数据
def query_one(sql):
    try:
        cursor.execute(sql)
        one = cursor.fetchone()
        return list(one)
    finally:
        cursor.close()
        db.close()


# 查询三条数据
def query_many(sql):
    try:
        cursor.execute(sql)
        one = cursor.fetchmany(size=3)
        return list(one)
    finally:
        cursor.close()
        db.close()


# 查询所有数据
def query_all(sql):
    try:
        cursor.execute(sql)
        one = cursor.fetchall()
        return list(one)
    finally:
        cursor.close()
        db.close()


# 执行插入、更新、删除操作
def update(sql):
    try:
        cursor.execute(sql)
        db.commit()
    except Exception as e:
        print(e)
        db.rollback()
    finally:
        cursor.close()
        db.close()