common iter query

以下代码由PHP站长网 52php.cn收集自互联网

现在PHP站长网小编把它分享给大家，仅供参考

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)

以上内容由PHP站长网【52php.cn】收集整理供大家参考研究

如果以上内容对您有帮助，欢迎收藏、点赞、推荐、分享。

（编辑：李大同）

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