|
|
import json
|
|
|
import os
|
|
|
import sys
|
|
|
|
|
|
from natsort import natsorted
|
|
|
from openpyxl import Workbook
|
|
|
from openpyxl.styles import Alignment
|
|
|
from openpyxl.styles import Border, Side, PatternFill
|
|
|
from openpyxl.styles import Font
|
|
|
from openpyxl.utils import get_column_letter
|
|
|
|
|
|
from Util.PgUtil import *
|
|
|
from openpyxl import load_workbook
|
|
|
|
|
|
# pip install natsort
|
|
|
# pip install openpyxl
|
|
|
from Util.PgUtil import *
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
file_path = '绘鸭AI模板ForFillSql.xlsx'
|
|
|
|
|
|
sql = 'truncate table t_hy_model_prompt'
|
|
|
execute_modify(sql)
|
|
|
|
|
|
# 加载工作簿
|
|
|
wb = load_workbook(file_path)
|
|
|
|
|
|
# 选择活动的工作表
|
|
|
ws = wb.active
|
|
|
# 初始化上一行的值列表，初始为空
|
|
|
last_row_values = []
|
|
|
|
|
|
# 遍历工作表中的所有行
|
|
|
idx = 0
|
|
|
for row in ws.iter_rows(values_only=True):
|
|
|
idx = idx + 1
|
|
|
if idx == 1:
|
|
|
continue
|
|
|
# 为当前行创建一个新列表
|
|
|
current_row_values = []
|
|
|
|
|
|
# 遍历当前行的每个单元格
|
|
|
for col_idx, cell_value in enumerate(row):
|
|
|
# 如果单元格的值是None，并且不是第一行（第一行没有上一行的值）
|
|
|
if cell_value is None and last_row_values:
|
|
|
# 使用上一行相同列的值替换当前单元格的值
|
|
|
current_row_values.append(last_row_values[col_idx])
|
|
|
else:
|
|
|
# 如果单元格的值不是None，添加到当前行的值列表中
|
|
|
current_row_values.append(cell_value)
|
|
|
|
|
|
model_id = int(current_row_values[0])
|
|
|
prompt_template = current_row_values[3]
|
|
|
prompt_en = current_row_values[4]
|
|
|
# 先检查有几个，然后+1就是应该是第几个
|
|
|
sql = 'select * from t_hy_model_prompt where model_id=' + str(model_id)
|
|
|
results = execute_query(sql)
|
|
|
prompt_id = len(results) + 1
|
|
|
|
|
|
sql = "insert into t_hy_model_prompt(model_id,prompt_id,prompt_template,prompt_en) values(%s,%s,%s,%s)"
|
|
|
params = (model_id, prompt_id, prompt_template, prompt_en)
|
|
|
execute_modify(sql, params)
|
|
|
# 更新上一行的值
|
|
|
last_row_values = current_row_values
|
|
|
|
|
|
wb.close()
|
