common iter query

下面是脚本之家 jb51.cc 通过网络收集整理的代码片段。

脚本之家小编现在分享给大家，也给大家做个参考。

def iquery(conn,sql,rowfactory=tuple,batch=1,*args,**kwgs):
    """在给定的数据库连接conn上执行sql(可带有参数*args或命名参数**kwgs)
    rowfactory指定输出物为元组/字典(或字典的扩展类)
        /或使用用命名参数的函数的结果
    batch <= 0时,直接全部输出
    batch == 1时,逐记录输出
    batch >= 1时,按指定的数量形成批次列表
    """
    from itertools import imap
    from contextlib import closing

    with closing(conn.cursor()) as curr:
        if curr.execute(sql,args or kwgs):
            if rowfactory == tuple:
                factory = tuple
            elif issubclass(rowfactory,dict):
                fields = [f[0].lower() for f in curr.description]
                factory = lambda x: rowfactory(zip(fields,x))
            else:
                fields = [f[0].lower() for f in curr.description]
                factory = lambda x: rowfactory(**dict(zip(fields,x)))

            if batch <= 0:
                yield imap(factory,curr.fetchall())
            elif batch == 1:
                for row in imap(factory,curr):
                    yield row
            else:
                rows = curr.fetchmany(batch)
                while rows:
                    yield map(factory,rows)
                    rows = curr.fetchmany(batch)

以上是脚本之家(jb51.cc)为你收集整理的全部代码内容，希望文章能够帮你解决所遇到的程序开发问题。

如果觉得脚本之家网站内容还不错，欢迎将脚本之家网站推荐给程序员好友。

（编辑：安卓应用网）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!