python使用pymysql

为了封装这个数据库操作为一个通用方法,我们可以创建一个函数,该函数接受数据库连接参数(如主机名、用户名、密码、数据库名)、SQL语句以及必要的参数(用于参数化查询)。下面是一个简单的封装示例:

python 复制代码
import pymysql

def execute_query(host, user, password, db, query, params=None, charset='utf8mb4'):
    """
    执行SQL查询并返回结果。

    :param host: 数据库主机名
    :param user: 数据库用户名
    :param password: 数据库密码
    :param db: 数据库名
    :param query: SQL查询语句
    :param params: 用于参数化查询的参数元组,默认为None
    :param charset: 字符集,默认为'utf8mb4'
    :return: 查询结果
    """
    try:
        # 建立数据库连接
        connection = pymysql.connect(host=host, user=user, password=password, db=db, charset=charset)
        
        with connection.cursor() as cursor:
            # 执行SQL语句
            cursor.execute(query, params)
            
            # 获取查询结果
            result = cursor.fetchall()
            
            # 返回查询结果
            return result
    
    except pymysql.MySQLError as e:
        print(f"数据库错误: {e}")
        return None
    
    finally:
        # 关闭数据库连接
        if connection:
            connection.close()

# 使用示例
if __name__ == "__main__":
    host = 'localhost'
    user = 'user'
    password = 'passwd'
    db = 'db'
    query = "SELECT `id`, `name` FROM `users` WHERE `age` > %s"
    params = (18,)
    
    result = execute_query(host, user, password, db, query, params)
    if result:
        for row in result:
            print(row)
    else:
        print("没有查询到结果")

这个函数execute_query接受必要的数据库连接信息和SQL查询语句,以及可选的参数元组用于参数化查询。它执行查询并返回结果。如果发生数据库错误(如连接问题或SQL语句错误),它会打印错误消息并返回None。最后,它确保数据库连接在使用后被正确关闭。

请注意,在实际应用中,您可能还需要处理其他类型的异常,比如网络问题导致的连接中断等。此外,对于需要提交更改的SQL语句(如INSERT、UPDATE、DELETE),您可能需要在with块内调用connection.commit()来提交更改。但在这个例子中,我们只处理查询操作,因此不需要提交更改。

相关推荐
SelectDB技术团队6 分钟前
兼顾高性能与低成本,浅析 Apache Doris 异步物化视图原理及典型场景
大数据·数据库·数据仓库·数据分析·doris
nuclear201118 分钟前
使用Python 在Excel中创建和取消数据分组 - 详解
python·excel数据分组·创建excel分组·excel分类汇总·excel嵌套分组·excel大纲级别·取消excel分组
inventecsh22 分钟前
mongodb基础操作
数据库·mongodb
白云如幻26 分钟前
SQL99版链接查询语法
数据库·sql·mysql
Lucky小小吴33 分钟前
有关django、python版本、sqlite3版本冲突问题
python·django·sqlite
爱吃烤鸡翅的酸菜鱼1 小时前
MySQL初学之旅(4)表的设计
数据库·sql·mysql·database
GIS 数据栈1 小时前
每日一书 《基于ArcGIS的Python编程秘笈》
开发语言·python·arcgis
爱分享的码瑞哥1 小时前
Python爬虫中的IP封禁问题及其解决方案
爬虫·python·tcp/ip
The_Ticker2 小时前
CFD平台如何接入实时行情源
java·大数据·数据库·人工智能·算法·区块链·软件工程
Elastic 中国社区官方博客2 小时前
Elasticsearch 开放推理 API 增加了对 IBM watsonx.ai Slate 嵌入模型的支持
大数据·数据库·人工智能·elasticsearch·搜索引擎·ai·全文检索