欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

python-pymysql 增删改查 简单的函数封装

程序员文章站 2022-05-30 12:27:33
...

python-pymysql 增删改查 简单的函数封装

pytho用于连接Mysql的第三方库:
pip install pymysql

基础信息

# coding:utf-8
import datetime
import random
import string

import pymysql
import pytz

基础配置 = {
    'mysql地址': "localhost",
    'mysql端口': 3306,
    'mysql账号': "root",
    'mysql密码': "a123456",
    'mysql库名': "finerain",
}

sql查询

输入SQL查询语句,返回一个查询结果的列表
注意:当有时间类型的数据需要通过str()函数转换,否则会出错

# 数据库查询
def sql查(sql):
    try:
        # 创建链接
        conn = pymysql.connect(
            host=基础配置['mysql地址'],
            port=基础配置['mysql端口'],
            user=基础配置['mysql账号'],
            password=基础配置['mysql密码'],
            db=基础配置['mysql库名'],
            charset='utf8')

        # 创建游标
        cursor = conn.cursor(cursor=pymysql.cursors.DictCursor)

        # 执行sql语句
        cursor.execute(sql)

        # 关闭连接,游标和连接都要关闭
        cursor.close()
        conn.close()

        # 获取结果
        result = cursor.fetchall()

        # 判断结果是否为空
        if result == ():
            return "空"
        else:
            data_dict = []
            for i in result:
                data_dict.append(i)
            return data_dict
    except:
        return "空"

sql修改

输入SQL更新,新增或删除语句,返回一个执行后的新增id

# 数据库查询
def sql改(sql):
    try:
        # 创建链接
        conn = pymysql.connect(
            host=基础配置['mysql地址'],
            port=基础配置['mysql端口'],
            user=基础配置['mysql账号'],
            password=基础配置['mysql密码'],
            db=基础配置['mysql库名'],
            charset='utf8')

        cursor = conn.cursor()
        cursor.execute(sql)  # 执行sql语句
        结果 = str(conn.insert_id())
        conn.commit()  # 执行update操作时需要写这个,否则就会更新不成功

        cursor.close()
        conn.close()
        return 结果
    except:
        return "空"