python操作mysql上下文版(二)

"""中级封装python 封装 mysql"""
from pymysql import connect


class DB(object):
    def __init__(self, database):
        """初始化链接"""
        # 创建connection 链接
        self.conn = connect(host='localhost', port=3306, user='root', password='xxx', database=database, charset='utf8')
        # 获取cursor 对象
        self.cs = self.conn.cursor()

    def __enter__(self):
        """返回操作对象"""
        return self.conn, self.cs

    def __exit__(self, exc_type, exc_val, exc_tb):
        """关闭链接"""
        # print("close")
        self.cs.close()
        self.conn.close()


with DB("xxxxx") as (conn, cs):
    sql = """select * from goods"""
    cs.execute(sql)
    table_content = cs.fetchall()
    conn.commit()

# 打印结果
for temp in table_content:
    print(temp)