python 数据库连接池

python 数据库连接池可以使用DBUtils模块。

示例程序：

#!/usr/bin/env python

# encoding:utf8



import sys

reload(sys)

sys.setdefaultencoding("utf-8")

import time



import MySQLdb

from DBUtils.PooledDB import PooledDB



import lsconst



SLEEP_FOR_NEXT_TRY = 3



class DBTools:

    def __init__(self, host, user, passwd, db):

        self.host = host

        self.user = user

        self.passwd = passwd

        self.db = db

        self.conn = None



    def connect(self):

        if self.conn:

            return



        while True:

            try:

                self.conn = PooledDB(MySQLdb, maxusage=10, db=self.db, host=self.host, user=self.user, passwd=self.passwd, charset='utf8')

                break

            except Exception, e:

                print lsconst.BRO, 'DBTools.connect', e, lsconst.NOR

                print 'Sleep %d seconds for next try.' % SLEEP_FOR_NEXT_TRY

                time.sleep(SLEEP_FOR_NEXT_TRY)



    def disconnect(self):

        if not self.conn:

            return

        try:

            #PooledDB instance has no attribute 'commit'

            #self.conn.commit()

            self.conn.close()

            self.conn = None



        except Exception, e:

            print lsconst.BRO, e, lsconst.NOR



    def operateDB(self, type, *sqls):

        if not self.conn:

            self.connect()

        try:

            pool = self.conn.connection()

            cur = pool.cursor()

            result = []

            if type == 'select':

                if cur.execute(sqls[0]) > 0:

                    #print sqls[0]

                    result = cur.fetchall()

                    pool.close()

                    return result

            else:

                for sql in sqls:

                    #print sql

                    cur.execute(sql)

                    pool.close()

                    return True

        except Exception, e:

            print lsconst.RED, e, lsconst.NOR

            print

            print lsconst.RED, sqls, lsconst.NOR



        #self.disconnect()

 

参考：

http://blog.csdn.net/gashero/article/details/1577187

 

原文：http://www.cnblogs.com/congbo/archive/2012/08/27/2658518.html