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()来提交更改。但在这个例子中,我们只处理查询操作,因此不需要提交更改。

相关推荐
易云码4 分钟前
信息安全建设方案,网络安全等保测评方案,等保技术解决方案,等保总体实施方案(Word原件)
数据库·物联网·安全·web安全·低代码
newxtc10 分钟前
【客观理性深入讨论国产中间件及数据库-科创基础软件】
数据库·中间件·国产数据库·国产中间件·科创
水月梦镜花12 分钟前
redis:list列表命令和内部编码
数据库·redis·list
yyfhq19 分钟前
sdnet
python
测试199826 分钟前
2024软件测试面试热点问题
自动化测试·软件测试·python·测试工具·面试·职场和发展·压力测试
love_and_hope26 分钟前
Pytorch学习--神经网络--搭建小实战(手撕CIFAR 10 model structure)和 Sequential 的使用
人工智能·pytorch·python·深度学习·学习
海阔天空_20131 小时前
Python pyautogui库:自动化操作的强大工具
运维·开发语言·python·青少年编程·自动化
MonkeyKing_sunyuhua1 小时前
ubuntu22.04 docker-compose安装postgresql数据库
数据库·docker·postgresql
天郁青1 小时前
数据库交互的本地项目:后台管理系统
数据库·交互
零意@1 小时前
ubuntu切换不同版本的python
windows·python·ubuntu