#!/usr/bin/python
# -*- coding: UTF-8 -*-
import config
import pymysql
from DBUtils.PooledDB import PooledDB

class mysql_obj():
    pool = None

    def __init__(self, config_temp=None):
        if config_temp is None:
            configs = {
                'host': config.mysql_host,
                'port': config.mysql_port,
                'user': config.mysql_user,
                'password': config.mysql_pass,
                'db': config.mysql_db,
                'charset': config.mysql_charset,
                'maxconnections': 70,  # 连接池最大连接数量
            }
        else:
            configs = config_temp
        configs['creator'] = pymysql
        configs['cursorclass'] = pymysql.cursors.DictCursor
        self.pool = PooledDB(**configs)

    def __enter__(self):
        self.conn = self.pool.connection()
        self.cursor = self.conn.cursor()
        return self

    def __exit__(self, type, value, trace):
        self.cursor.close()
        self.conn.close()

if __name__ == '__main__':

    pass