当前位置: 首页 > news >正文

pymysql的上下文管理器:简化数据库操作

pymysql的上下文管理器:简化数据库操作

当我们使用 pymysql 操作数据库时,管理数据库连接和游标的生命周期是一项重要的任务。Python 的上下文管理器提供了一种优雅的方式来处理资源的获取和释放。在本文中,我们将探索如何创建一个简单的 pymysql 上下文管理器,以简化数据库操作。

为什么使用上下文管理器

  • 自动资源管理:自动关闭数据库连接和游标,即使发生异常也能确保资源被释放。
  • 代码可读性:使代码更加简洁,易于理解和维护。
  • 异常处理:可以在退出代码块时统一处理异常。

代码实现

import pymysqlclass Connect(object):def __init__(self, host, user, password, db, charset='utf8mb4',cursor_class=pymysql.cursors.DictCursor):self.config = {'host': host,'user': user,'password': password,'database': db,'charset': charset,'cursorclass': cursor_class}self.conn = pymysql.connect(self.config)self.cursor = self.conn.cursor(pymysql.cursors.DictCursor)def __enter__(self):# 返回自身以便在 with 块中使用return selfdef __exit__(self, exc_type, exc_val, exc_tb):# 检查游标和连接是否打开,然后关闭它们if self.cursor:self.cursor.close()if self.conn:self.conn.close()# 处理异常if exc_type:print(f"An error occurred: {exc_val}")def exec_query(self, sql, **kwargs):# 执行查询并返回结果集self.cursor.execute(sql, kwargs)return self.cursor.fetchall()def exec_command(self, sql, **kwargs):# 执行非查询命令并返回影响的行数self.cursor.execute(sql, kwargs)return self.conn.commit()  # commit() 返回 None,你可能想要返回影响的行数

简单使用

if __name__ == "__main__":settings = {'host': 'localhost','user': 'your_username','password': 'your_password','database': 'your_database','charset': 'utf8mb4',}with Connect(**settings) as db:results = db.exec_query("SELECT * FROM your_table")for result in results:print(result)affected_rows = db.exec_command("INSERT INTO your_table (column) VALUES (value)")print(f"受影响的行数: {affected_rows}")

若有错误与不足请指出,关注DPT一起进步吧!!!

http://www.lryc.cn/news/424200.html

相关文章:

  • AI秘境-墨小黑奇遇记 - 修炼成神经(二)
  • 计算机网络之分组交换时延的计算
  • 虚幻5|入门AI行为树,建立敌人
  • ARM处理架构中的PMU(Performance Monitoring Unit)和 AMU(Activity Monitors Unit)简介
  • Service服务在Android中的使用
  • 浅谈C语言位段
  • arcgisserver登陆信息不正确
  • KOLA: CAREFULLY BENCHMARKING WORLD KNOWLEDGE OF LARGE LANGUAGE MODELS
  • Robot Operating System——机器人关节的角度、速度和力矩
  • 一分钟掌握java9新特性
  • 89. UE5 RPG 实现伤害 冷却 消耗技能描述
  • el-tree树状控件,定位到选中的节点的位置
  • YOLO目标检测的单目(多目标测距),使用相机光学模型,支持目标检测模型训练,可输出目标位置和距离信息并可视化
  • unity简易lua文件迁移工具
  • Elasticsearch中的自动补全功能详解与实践
  • 前端如何使用Nginx代理dist网页,代理websocket,代理后端
  • Cannot connect to the Docker daemon at unix:///var/run/docker.sock. 问题解决
  • 零基础学习Redis(2) -- Redis安装与配置
  • UniApp第一天
  • TLE4966-3G带方向检测功能的高灵敏度汽车霍尔开关
  • Github 2024-08-14 C开源项目日报Top10
  • 飞桨Paddle API index_add 详解
  • 后端代码练习1——加法计算器
  • 观察者模式和MQ是什么关系
  • JDK动态代理和CGLIB动态代理案例分析
  • 【数据结构-前缀哈希】力扣1124. 表现良好的最长时间段
  • 电商平台产品ID|CDN与预渲染|前端边缘计算
  • LATTICE进阶篇DDR2--(4)DDR2 IP核总结
  • windows下php安装kafka
  • 【wiki知识库】09.欢迎页面展示(浏览量统计)SpringBoot部分