作用
对给定的列提取第1行,第27行,2-24行红色填充的数据,替换并生成sql语句,保存为output.txt
代码
要求安装openpyxl库
pip intall openpyxl
import openpyxl
### 对给定的列提取第1行,第27行,2-24行红色填充的数据,替换并生成sql语句,保存为output.txt #### 定义颜色填充的识别函数
def is_red_fill(cell):# 检查单元格是否有填充,并且填充颜色是否为红色(FFFF0000)return cell.fill.start_color.index == 'FFFF0000'# 读取Excel文件并获取指定列的数据,提取第1行,第27行,2-24行红色填充的数
def get_data_from_excel(file_path, column):wb = openpyxl.load_workbook(file_path)ws = wb.active# 获取第1行的文本first_row_text = ws[column + '1'].value if ws[column + '1'].value else ''# 获取第27行的文本twenty_seventh_row_text = ws[column + '27'].value if ws[column + '27'].value else ''# 获取第2行到第24行红色填充的数据red_filled_texts = []for row in range(2, 25): # 从第2行到第24行cell = ws[column + str(row)]if is_red_fill(cell) and cell.value: # 检查是否有红色填充并且单元格有值red_filled_texts.append(cell.value)return first_row_text, twenty_seventh_row_text, red_filled_texts# 调用函数并打印结果
def fanction(file_path, columns): result = []for column in columns:# column = 'AE' # 假设我们关注的是B列first_row_text, twenty_seventh_row_text, red_filled_texts = get_data_from_excel(file_path, column)red_filled_texts = tuple(red_filled_texts)red_filled_texts = str(red_filled_texts)# print(f"第1行的文本: {first_row_text}")# print(f"第27行的文本: {twenty_seventh_row_text}")# print(f"第2行到第24行红色填充的数据: {red_filled_texts}")if (first_row_text == "") | (twenty_seventh_row_text == "") | (len(red_filled_texts) < 3):print(column, "数据不全")continue # 数据不全跳过sql1 = f"""update t_LIMS_ORDERTASK OT
left join T_LIMS_TEST t on T.ID=OT.TESTID
set T.TESTUSERNAME='{twenty_seventh_row_text}'
where T.TESTREPORTFORMULA='{first_row_text}' and OT.TESTUSERNAME in {red_filled_texts};"""sql2 = f"""update T_LIMS_ORDERTASKLOG L
left join T_LIMS_ORDERTASK L on L.ORDERTASKID =OT.ID
left join T_LIMS_TEST t on T.ID=OT.TESTID
set L.CREATEDBYNAME ='{twenty_seventh_row_text}'
where T.TESTREPORTFORMULA='{first_row_text}' and L.CREATEDBYNAME in {red_filled_texts} and L.OPERATIONTYPE ='班长审核';"""result.append(sql1)result.append(sql2)with open('output.txt', 'w', encoding='utf-8') as f:for item in result:f.write("%s\n\n" % item)print("output.txt保存成功!")path = r"C:\Users\16071\Desktop\LIMS未持证上岗资质异常处理.xlsx"
cols = ['B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z', 'AA', 'AE']
fanction(path, cols)