iis服务器助手广告广告
返回顶部
首页 > 资讯 > 后端开发 > Python >Python自动生成sql语句
  • 908
分享到

Python自动生成sql语句

自动生成语句Python 2023-01-31 02:01:49 908人浏览 安东尼

Python 官方文档:入门教程 => 点击学习

摘要

#!usr/bin/env python # coding=utf-8 def auto_insert_sql(objs, table=None, charset='UTF-8'): """ 自动生成insert

#!usr/bin/env python
# coding=utf-8

def auto_insert_sql(objs, table=None, charset='UTF-8'):
    """
    自动生成insert SQL语句,如果obj为对象列表,则根据第一个对象生成SQL语句。
    @objs 对象或对象列表
    @table 数据表名
    @charset 数据库编码
    @return (SQL语句,转换后的对象或对象列表)
    """
    is_batch = isinstance(objs, list)
    if not objs:
        return (None, []) if is_batch else (None, None)
    obj = objs[0] if is_batch else objs
    obj_li = objs if is_batch else [objs]
    # 解析表名
    if not table:
        table = obj.__class__.__name__
    # 解析字段 和 转换数据
    cols = []
    row_li = []
    if hasattr(obj, '__iter__'):  # dict
        # 解析字段
        for k in obj:
            cols.append(k)
        # 转换数据
        for obj in obj_li:
            row = {}
            for k in cols:
                v = obj[k]
                row[k] = v.encode(charset) if isinstance(v, unicode) else v
            row_li.append(row)
    elif hasattr(obj, '__dict__'):  # object
        for i, (k, v) in enumerate(obj.__dict__.items()):
            cols.append(k)
        for obj in obj_li:
            row = {}
            for k in cols:
                v = getattr(obj, k)
                row[k] = v.encode(charset) if isinstance(v, unicode) else v
            row_li.append(row)
    elif hasattr(obj, 'ListFields'):  # Google Protocol Buffer
        for field, v in obj.ListFields():
            cols.append(field.name)
        for obj in obj_li:
            row = {}
            for k in cols:
                v = getattr(obj, k) if obj.HasField(k) else None
                row[k] = v.encode(charset) if isinstance(v, unicode) else v
            row_li.append(row)
    else:
        raise Exception("couldn't parse object: %s" % obj)
    # 生成SQL语句
    if not cols:
        raise Exception("no colume found in object: %s" % obj)
    sql_prefix = ['insert into ', table, ' (']
    sql_suffix = [') values(']
    for i, k in enumerate(cols):
        sql_prefix.append(', %s' % k if i > 0 else k)
        sql_suffix.append(', :%s' % k if i > 0 else ':%s' % k)
    sql_suffix.append(')')
    sql_prefix.extend(sql_suffix)
    sql = ''.join(sql_prefix)
    # --------------------------------------------------
    return (sql, row_li) if is_batch else (sql, row_li[0])


class A(object):
    def __init__(self):
        self.c1 = 0
        self.c2 = 0
        self.c3 = 3


def main():
    # a = A()
    a = {'a': 1, 'b': 2}
    print auto_insert_sql([a], 'test_table')
if __name__ == '__main__':
    main()

--结束END--

本文标题: Python自动生成sql语句

本文链接: https://www.lsjlt.com/news/185439.html(转载时请注明来源链接)

有问题或投稿请发送至: 邮箱/279061341@qq.com    QQ/279061341

本篇文章演示代码以及资料文档资料下载

下载Word文档到电脑,方便收藏和打印~

下载Word文档
软考高级职称资格查询
编程网,编程工程师的家园,是目前国内优秀的开源技术社区之一,形成了由开源软件库、代码分享、资讯、协作翻译、讨论区和博客等几大频道内容,为IT开发者提供了一个发现、使用、并交流开源技术的平台。
  • 官方手机版

  • 微信公众号

  • 商务合作