add: common.py 新增批量插入方法 insert_batch(table_name: str, dict_list: list) -> None

datacenter/common.py

@@ -19,6 +19,28 @@ def query_json(sql):
     return result
 
 
+def insert_batch(table_name: str, dict_list: list) -> None:
+    """
+    向{table_name}表批量插入数据,
+    {table_name}: str，表名，建议带上schema，如label.t1027
+    {dict_list}: dict的list，欲插入数据的列表，如[{"id": "1", "name": "Chris"}, {"id": "2", "name": "Harris"}]
+    """
+    if 0 == len(dict_list):
+        return
+    # 获得一个游标(cursor)对象
+    cursor = connection.cursor()
+    # 构建sql语句
+    keys = ', '.join(dict_list[0].keys())
+    values = []
+    for val in dict_list:
+        value = '(' + str(val.values()).lstrip('dict_values([').rstrip('])') + ')'
+        values.append(value)
+    sql = "INSERT INTO {table_name}({keys}) VALUES {values}".format(table_name=table_name, keys=keys,
+                                                                    values=', '.join(values))
+    cursor.execute(sql)
+    connection.commit()
+
+
 def get_label_json(label_code):
     tblName = "t" + str(label_code).strip(' ')[0:4]
     sql = "SELECT * FROM label." + tblName + f" WHERE code = '{label_code}' ORDER BY tm DESC"