python使用上下文管理器操作数据库

最新推荐文章于 2025-11-18 14:03:03 发布

原创最新推荐文章于 2025-11-18 14:03:03 发布 · 1.2k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#python #数据库 #上下文管理器

python 高级同时被 2 个专栏收录

2 篇文章

订阅专栏

python

1 篇文章

订阅专栏

本文介绍如何利用Python的with语法结合自定义上下文管理器简化数据库操作流程，包括连接数据库、执行SQL语句及关闭连接等步骤。

with 语法用于简化资源操作的后续清除操作，是 try/finally 的替代方法，实现原理建立在上下文管理器之上

代码如下，数据库相关操作依库

from pymysql import *

class DB():

    def __init__(self, my_database1, my_password):

        self.conn = connect(host='localhost', port=3306, database=str(my_database1), user='root', password=str(my_password),
                           charset='utf8')
        self.cs1 = self.conn.cursor()


    def __enter__(self):
        return (self.cs1, self.conn)

    def __exit__(self, exc_type, exc_val, exc_tb):
        self.conn.close()
        self.cs1.close()

with DB('jing_dong','mysql') as (db, dbcoon):
    count = db.execute('''select * from goods;''')
    content = db.fetchall()
    print(content)
    for i in range(len(content)):
        print(content[i])
        i += 1
    db.execute('''insert into goods(name,price) values('鞋子','200');''')
    dbcoon.commit()