context mysql_python使用上下文处理器来管理mysql连接

一，使用contextmanager

from contextlib import contextmanager

import MySQLdb

DB_CONFIG = {

'host': '192.168.1.253',

'user': 'pythondb',

'passwd': 'python123',

'port': 20002,

'db': 'xw',

'charset': 'utf8'

}

@contextmanager

def open_mysql(db_conf):

try:

conn = MySQLdb.connect(**db_conf)

if conn:

yield conn.cursor(cursorclass=MySQLdb.cursors.DictCursor)

except Exception as e:

print(e)

finally:

conn.close()

if __name__ == '__main__':

with open_mysql(DB_CONFIG) as con:

sql = 'select 1 as c'

con.execute(sql)

rv = con.fetchall()

print(rv)

二，使用__enter__跟__exit__

class OpenMysqlConn(object):

def __init__(self, db_conf):

self.db_conf = db_conf

self.conn = None

def __enter__(self):

try:

self.conn = MySQLdb.connect(**self.db_conf)

if self.conn:

return self.conn.cursor(cursorclass=MySQLdb.cursors.DictCursor)

except (AttributeError, MySQLdb.OperationalError):

current_app.logger.error("连接数据库失败")

def __exit__(self, exc_type, exc_val, exc_tb):

if self.conn:

self.conn.close()