python-enter和exit的使用

python-enter和exit的使用

可使用with语句，结束时调用__exit__方法

import pymongo
import sys
import traceback
import logging

class ConnectMongo(object):
    """
    连接 MongoDB 数据库
    :param dbHost:                  数据库IP地址
    :param dbPort:                  数据库Port端口

    """

    def __init__(self, dbHost, dbPort):
        self.host = dbHost
        self.port = int(dbPort)
        self.conn = None

    def __enter__(self):
        self.conn = pymongo.MongoClient(host=self.host, port=self.port)
        return self.conn

    def __exit__(self, exc_type, exc_val, exc_tb):
        if self.conn is not None:
            self.conn.close()

def insertData(self, data):
	"""

	:param kwargs:
	:return:
	"""
	try:

		with ConnectMongo(dbHost="", dbPort="") as conn:
            conn.get_database().get_collection().insert()
        # with结束后会断开数据库连接

	except Exception as er:
		logging.error(f"** Meet exception ** :{er}")
		logging.error("".join(traceback.format_exception(*sys.exc_info())))