lib/dbi/_dbi.py

# encoding: utf-8


from contextlib import closing

DEFAULTBUFFERSIZE = 8192

class _Buffer(object):
    ''' 提供简化的缓冲批量处理方式 '''
    def __init__(self, sql, size=DEFAULTBUFFERSIZE):
       ''' 初始化时指定缓冲操作SQL '''
       self.sql = sql
       self.size = size
       self.data = []

    def connect(self, conn):
       ''' 绑定数据库连接 '''
       self.conn = conn

    def flush(self):
       ''' flush self's buffer '''
       buffer, self.data = self.data[:], []
        with closing(self.conn.cursor()) as curr:
           curr.executemany(self.sql, buffer)
          self.conn.commit()
         
    def append(self, o):
        ''' 向缓冲中添加数据 '''
        self.data.append(o)
        if len(self.data) >= self.size:
            self.flush()

    def __lshift__(self, o):
         ''' 重载 << 操作符于append方法 '''
         self.append(o)

    def close(self):
        ''' 完成关闭 '''
       self.flush()


class _DBI(object):

    def __init__(self, **kwg):
        ''' 初始化时指定参数 '''
       self.setting = dict(kwg)
       self.conn = None

    def open(self):
       ''' 用专有的驱动程序依据初始化参数创建数据库连接 '''
       if not self.conn:
          #~ self.conn = ..(**self.setting)
            pass
        return

    def close(self):
         ''' 关闭连接 '''
        self.conn.close()
       self.conn = None

    def __enter__(self):
       ''' 正文模式的进入处理 '''
       self.open()
       return self

    def __exit__(self, *args):
        ''' 正文模式的退出处理 '''
       self.close()
       return


    def execute(self, sql, *args):
        ''' 执行给定的SQL, 并传入给定的参数 '''
        with closing(self.conn.cursor()) as curr:
            curr.execute(sql, args)
           self.conn.commit()

    def query(self, sql, *args, **kwg):
        ''' 按指定的SQL及参数查询, 
       命名参数中可以指定rowfactory='dict'要求以字典形式返回查询结果,
       或以其提供的方法代入查询结果的列名-值, 生产指定的对象实例,
       显示指定rowfactory或以默认的形式, 将直接返回查询结果集中的元组'''
       setting = dict(kwg)
       rowfactory = setting.get('rowfactory','row')
        with closing(self.conn.cursor()) as curr:
           curr.execute(sql, args)
           fields = [f[0] for f in curr.description]
           for r in curr:
               if rowfactory == 'row':
                  yield r
               elif rowfactory == 'dict':
                  yield dict(map(None, fields, r))
             else:
                  yield rowfactory(**dict(map(None, fields, r)))

    def bindBuffer(self, buffer):
        ''' 绑定一个已经定义了的_Buffer实例 '''
       self.buffer = buffer
       buffer.connect(self.conn)

    def initBuffer(self, sql, size=DEFAULTBUFFERSIZE):
        ''' 指定SQL, 初始化一个被绑定的_Buffer '''

       self.bindBuffer(_Buffer(sql, size))