0
0

AI帮我写python脚本，效率提升N倍

当前在初创芯片公司，工作中经常需要通过脚本实现效率提升。因为公司人少，都需要自己空闲时间做一些脚本。最近使用AI做了两个尝试，明显能感觉到AI带给我的效率提升。以前是我需要自己梳理需求、写代码、调试。现在用上AI，我只需要自己提需求，然后迭代需求，测试和微调脚本即可。而且在使用AI的时候，也领会到一点和AI打交道的方法。

先介绍一下两个脚本的用途，最后会贴上可以使用的脚本代码。

脚本1：文件列表展开工具，命令行方式运行。因为芯片编译的filelist经常是嵌套的，且带环境变量的，但是有些工具（例如FPGA编译工具）需要读取打平的filelist。

脚本2：excel反标工具，带图形界面。在芯片开发末期，要做一些质量活动——就是确认下游是否覆盖了上游的需求，例如验证是否覆盖所有设计的功能点。经常需要在设计的功能点（excel文件A）上标注对应验证的测试点（excel文件B）的编号，这个工作就是需要从文件B中复制测试点编号的单元格内容，到文件A中。过程中有大量的ctrl+c和ctrl+v的操作，每次做下来，左手累的不行。希望有个脚本能实现ctrl+c和ctrl+v的动作。

1、文件列表展开脚本

要处理的文件层级如下：

顶层的filelist如下：

子级filelist内容如下：

AI提示词V1.0：

帮我写一个python脚本。工作环境为linux。filest文件中每一行是一个文件路径，把其中的环境变量展开，然后输出为一个新的文件。filelist中还会嵌套filelist，也需要展开。filelist的行前缀-f。删除文件路径两侧的空格。

生成的代码V1.0：

  1. import os
  3. def expand_env_variables(path):
  4.     """展开环境变量"""
  5.     return os.path.expandvars(path)
  7. def process_file_list(file_list_path, output_path):
  8.     """处理文件列表，展开环境变量，并输出到新文件"""
  9.     expanded_paths = []
  10.     with open(file_list_path, 'r') as file:
  11.         for line in file:
  12.             # 删除行首尾的空格
  13.             line = line.strip()
  14.             # 检查是否是文件列表的行
  15.             if line.startswith('-f'):
  16.                 # 递归处理嵌套的filelist
  17.                 nested_file_list_path = expand_env_variables(line[2:])
  18.                 nested_paths = process_file_list(nested_file_list_path, output_path)
  19.                 expanded_paths.extend(nested_paths)
  20.             else:
  21.                 # 展开环境变量并添加到列表
  22.                 expanded_path = expand_env_variables(line)
  23.                 expanded_paths.append(expanded_path)
  24.     
  25.     # 将展开后的路径写入到输出文件
  26.     with open(output_path, 'w') as output_file:
  27.         for path in expanded_paths:
  28.             output_file.write(path + '\n')
  30. # 脚本的入口点
  31. if __name__ == "__main__":
  32.     file_list_path = 'filelist'  # 假设你的文件名为filelist
  33.     output_file_path = 'expanded_filelist'  # 输出文件的名称
  34.     process_file_list(file_list_path, output_file_path)

这个脚本有3个问题：

1、就是process_file_list每次调用都会打开文件。但是递归调用的时候，写入有些问题

2、嵌套的filelist路径还会包含空格，会出错。看来去掉空格这个命令，AI理解的和我不同。

3、只处理嵌套的filelist，而且递归调用处理二级fielist时，会返回空（因为只处理前缀-f的行），导致运行出错。

AI提示词V2.0

帮我写一个python脚本。工作环境为linux。filest文件中每一行是一个文件路径，把其中的环境变量展开，然后输出为一个新的文件。filelist中还会嵌套filelist，也需要展开。filelist中嵌套的filelist的行前缀-f，其他的文件没有前缀。删除文件路径两侧的空格。文件路径提取后先保存在一个列表中，读取完成后，再一次性输出到文件。递归执行时，返回列表，不要直接写入文件。先把路径两侧可能存在的空格去掉，再展开环境变量。

生成的代码V2.0

  1. import os
  3. def expand_path(path):
  4.     # 去除路径两侧的空格并展开环境变量
  5.     return os.path.expandvars(os.path.expanduser(path.strip()))
  7. def process_filelist(filelist_path, expanded_paths):
  8.     # 读取filelist文件
  9.     with open(filelist_path, 'r') as file:
  10.         for line in file:
  11.             original_line = line
  12.             # 去除行首尾的空格
  13.             stripped_line = line.strip()
  14.             # 检查是否是嵌套的filelist文件
  15.             if stripped_line.startswith('-f'):
  16.                 # 去除'-f'前缀并处理嵌套的filelist
  17.                 nested_filelist_path = expand_path(stripped_line[2:])
  18.                 if os.path.isfile(nested_filelist_path):
  19.                     # 递归处理嵌套的filelist文件
  20.                     process_filelist(nested_filelist_path, expanded_paths)
  21.             else:
  22.                 # 展开环境变量
  23.                 expanded_path = expand_path(stripped_line)
  24.                 # 确保路径是文件并且存在，避免重复添加
  25.                 if os.path.isfile(expanded_path) and expanded_path not in expanded_paths:
  26.                     expanded_paths.append(expanded_path)
  28.     return expanded_paths
  30. def main():
  31.     # filelist文件的路径
  32.     filelist_path = 'filelist'
  33.     # 用于存储所有文件路径的列表
  34.     expanded_paths = []
  35.     # 处理filelist文件并获取所有路径
  36.     expanded_paths = process_filelist(filelist_path, expanded_paths)
  37.     # 将路径列表写入到新文件
  38.     with open('expanded_file_paths.txt', 'w') as output_file:
  39.         for path in expanded_paths:
  40.             output_file.write(path + '\n')
  41.     print(f"Expanded paths have been written to 'expanded_file_paths.txt'")
  43. if __name__ == "__main__":
  44.     main()

运行结果

2、excel反标工具

由于生成的代码比较长，我就每个都贴了。只是贴一下提示词，以及生成的脚本的缺点。

提示词V1.0

帮我写一个python脚本。使用pywin32和pyside6。需求如下：1、支持从当前已经打开的excel文件中选择excel文件；2、支持提取两个excel文件中选中的单元格的内容；3、提供界面，界面有2个下拉菜单，下拉菜单中为当前打开的excel文件名，支持从其中选择excel文件；还需要一个刷新按钮，更新下拉菜单； 4、支持界面置顶显示，也可以取消置顶显示，置顶选择按钮按下表示置顶，弹起表示取消置顶；5、支持通过一个按键（颜色为蓝色），关联两个单元格的内容——关联方式为，将其中一个文件单元格拷贝另一个文件的同一行的某列，列的位置，可以通过界面下拉菜单选择。6、界面有一个刷新按钮，用于更新excel文件列表。7、界面还支持一个按钮来表示复制的方向，按钮弹起时，表示从第一个文件复制到第二个文件的某个单元格，按键按下时表示从第二个文件复制到第一个文件的某个单元格，此单元格与excel中已选中的单元格同行，列的位置由界面选择的列位置来决定。

生成的GUI界面：

问题：可以生成图形界面每个按钮有对应的调用函数，但是调用函数里面只有一些打印提示，没有对应的功能（虽然我在提示词中对每个按钮的功能有描述，但是AI看起来只理解到对界面的诉求，没有理解按钮背后要做的动作。

提示词V2.0

提示词改进点：增加功能需求概述，放在提示词最前面；对界面显示区域进行划分。

帮我写一个python脚本。使用pywin32和pyside6。脚本支持从当前已经打开的excel文件中选择excel文件，支持复制其中一个excel文件中选中的单元格的内容，粘贴到另外一个excel选中的单元格中；脚本还支持界面。

脚本的界面需要分3个区域。

区域1包含如下控件：静态文本，内容为”Excel文件1”，按照最小长度显示；下拉菜单A，显示当前已经打开的excel的文件名，下拉菜单通过“更新”按钮刷新下拉选项。

区域2包含如下控件：静态文本，内容为”Excel文件2”，按照最小长度显示；下拉菜单A，显示当前已经打开的excel的文件名，下拉菜单通过“更新”按钮刷新下拉选项。

区域3包含控件：刷新按钮，按钮按下时，重新检测已经打开的excel文件，更新区域1和2中的下拉菜单，刷新前保留之前选中的文件名，更新下来菜单后，如果之前下拉菜单的选中的文件还在下来选项中，则还是选择和之前一样的文件名；方向按钮，弹起时表示从区域1的菜单A中选中的单元格，复制到区域2的菜单A文件当前激活的sheet的单元格，按下时，则相反，按钮支持check；关联按钮，无需check，填充浅蓝色，此按钮按下时，按照方向按钮的选择，将选中的单元格复制到另一个excel文件选中的单元格；置顶按钮，有check功能，按下时，表示置顶，此按钮填充为黄色，弹起时表示不置顶，无填充，默认为置顶。

运行效果：

生成的脚本还是有一点问题，修改后可以使用的代码如下：

  1. import sys
  2. import win32com.client
  3. from PySide6.QtWidgets import (QApplication, QWidget, QVBoxLayout, QHBoxLayout,
  4.                                QLabel, QComboBox, QPushButton, QCheckBox)
  5. from PySide6.QtCore import Qt
  6. from PySide6.QtGui import QColor,  QShortcut, QKeySequence
  7. import os
  9. class ExcelManager(QWidget):
  10.     def __init__(self):
  11.         super().__init__()
  12.         self.initUI()
  13.         self.direction = 'to2'  # 'to1' or 'to2'
  15.     def initUI(self):
  16.         # 设置窗口标题和初始大小
  17.         self.setWindowTitle('Excel Manager')
  18.         self.setGeometry(100, 100, 800, 200)
  20.         # 创建垂直布局
  21.         layout = QVBoxLayout(self)
  23.         # 区域1
  24.         self.region1_layout = QHBoxLayout()
  25.         self.label1 = QLabel('Excel文件1')
  26.         self.label1.setMinimumWidth(self.label1.sizeHint().width())
  27.         self.comboBoxA1 = QComboBox()
  28.         self.region1_layout.addWidget(self.label1)
  29.         self.region1_layout.addWidget(self.comboBoxA1, stretch=5)
  30.         layout.addLayout(self.region1_layout)
  32.         # 区域2
  33.         self.region2_layout = QHBoxLayout()
  34.         self.label2 = QLabel('Excel文件2')
  35.         self.label2.setMinimumWidth(self.label2.sizeHint().width())
  36.         self.comboBoxA2 = QComboBox()
  37.         self.region2_layout.addWidget(self.label2)
  38.         self.region2_layout.addWidget(self.comboBoxA2, stretch=5)
  39.         layout.addLayout(self.region2_layout)
  41.         # 区域3
  42.         self.region3_layout = QHBoxLayout()
  43.         self.refreshButton = QPushButton('刷新')
  44.         self.directionButton = QPushButton('方向:1-->2')
  45.         self.directionButton.setCheckable(True)
  46.         self.assocButton = QPushButton('关联')
  47.         self.assocButton.setStyleSheet("background-color: lightblue")
  48.         self.stickyButton = QCheckBox('置顶')
  49.         self.region3_layout.addWidget(self.refreshButton)
  50.         self.region3_layout.addWidget(self.directionButton)
  51.         self.region3_layout.addWidget(self.assocButton)
  52.         self.region3_layout.addWidget(self.stickyButton)
  54.         layout.addLayout(self.region3_layout)
  56.         # # 创建一个全局快捷键 F4
  57.         # self.shortcut = QShortcut(QKeySequence(Qt.Key_F4), self, context=Qt.ApplicationShortcut)
  58.         # self.shortcut.activated.connect(self.copy_cells)
  60.         # 连接信号和槽
  61.         self.refreshButton.clicked.connect(self.refresh_comboboxes)
  62.         self.directionButton.clicked.connect(self.toggle_direction)
  63.         self.assocButton.clicked.connect(self.copy_cells)
  64.         self.stickyButton.stateChanged.connect(self.sticky_logic)
  65.         self.stickyButton.setChecked(True) # 默认选中
  67.         # 刷新下拉菜单
  68.         self.refresh_comboboxes()
  69.         # 设置下拉菜单的默认值
  71.     def refresh_comboboxes(self):
  72.         # 使用pywin32获取打开的Excel文件列表
  73.         app = win32com.client.Dispatch("Excel.Application")
  74.         excel_files = []
  75.         
  76.         try:
  77.             for book in app.Workbooks:
  78.                 excel_files.append(book.FullName)
  79.         except (IndexError, AttributeError):
  80.             pass
  81.         current_selection1 = self.comboBoxA1.currentText()
  82.         current_selection2 = self.comboBoxA2.currentText()
  84.         self.comboBoxA1.clear()
  85.         self.comboBoxA2.clear()
  86.         self.comboBoxA1.addItems(excel_files)
  87.         self.comboBoxA2.addItems(excel_files)
  89.         if current_selection1 in excel_files:
  90.             self.comboBoxA1.setCurrentText(current_selection1)
  91.         if current_selection2 in excel_files:
  92.             self.comboBoxA2.setCurrentText(current_selection2)
  94.     def toggle_direction(self):
  95.         # 切换方向
  96.         if self.direction == 'to2' :
  97.             self.direction = 'to1' 
  98.             self.directionButton.setText('方向：文件1<--2')
  99.         else: 
  100.             self.direction = 'to2'
  101.             self.directionButton.setText('方向：文件1-->2')
  102.         print(f"方向已切换到: {self.direction}")
  103.         
  105.     def copy_cells(self):
  106.         # 复制单元格内容
  107.         if self.comboBoxA1.currentIndex() == -1 or self.comboBoxA2.currentIndex() == -1:
  108.             print("请先选择两个Excel文件")
  109.             return
  110.         if(self.comboBoxA1.currentText() == self.comboBoxA2.currentText()):
  111.             print("请先选择两个不同的Excel文件")
  112.             return            
  113.         xls_app = win32com.client.Dispatch("Excel.Application")
  114.         original_display_alerts = xls_app.DisplayAlerts  # 保存原始状态
  115.         # # 关闭警告，以便在操作过程中不显示任何对话框
  116.         # xls_app.DisplayAlerts = False
  117.         comboBoxA1_filename = os.path.basename(self.comboBoxA1.currentText())
  118.         comboBoxA2_filename = os.path.basename(self.comboBoxA2.currentText())
  119.         for wb in xls_app.Workbooks:
  121.             if(wb.Name == comboBoxA1_filename):
  122.                 wb1=wb
  123.             if(wb.Name == comboBoxA2_filename):
  124.                 wb2 = wb
  126.         try:
  128.             if self.direction == 'to2':
  129.                 wb1.Activate()
  130.                 selected_range1 = wb1.Application.Selection
  131.                 value1 = selected_range1.Value
  132.                 wb2.Activate()
  133.                 selected_range2 = wb2.Application.Selection
  134.                 if(selected_range2.Text != ""):
  135.                     selected_range2.Value +='\n'+ value1
  136.                 else:
  137.                     selected_range2.Value = value1
  138.                 # print(wb1.Name, selected_range1.Row, selected_range1.Column, value1)
  139.                 # print(wb2.Name, selected_range2.Row, selected_range2.Column, selected_range2.Value)
  140.                 # print('\n')
  141.             else:
  142.                 wb2.Activate()
  143.                 selected_range2 = wb2.Application.Selection
  144.                 value2 = selected_range2.Value 
  145.                 wb1.Activate()
  146.                 selected_range1 = wb1.Application.Selection
  147.                 if(selected_range1.Text != ""):
  148.                     selected_range1.Value +='\n'+ value2
  149.                 else:
  150.                     selected_range1.Value = value2
  152.         except Exception as e:
  153.             print(f"复制出错: {e}")
  154.         # finally:
  155.         #     excel1.Close(SaveChanges=False)
  156.         #     excel2.Close(SaveChanges=False)
  158.     def sticky_logic(self, state):
  159.         if state == Qt.Checked:
  160.             self.stickyButton.setStyleSheet("background-color: yellow")
  161.             self.setWindowFlags(Qt.WindowStaysOnTopHint)
  162.         else:
  163.             self.stickyButton.setStyleSheet("")
  164.             self.setWindowFlags(Qt.Window)
  165.         self.show()
  167. def main():
  168.     app = QApplication(sys.argv)
  169.     excel_manager = ExcelManager()
  170.     excel_manager.show()
  171.     sys.exit(app.exec())
  173. if __name__ == '__main__':
  174.     main()

copy_cells函数调整过，和我想要的有差异。但是感觉文字不好描述。

3、总结

避免错误的提示词

filelist的提示词中的

先把路径两侧可能存在的空格去掉，再展开环境变量。

我也是试过另一个错误的表述：

替换环境变量前，先去掉路径的空格

这个“前”被AI理解为位置关系，而不是操作顺序关系。

推荐使用总分结构组织提示词

建议使用总分结构。就是先描述整体的功能，然后再描述细分的功能。只描述细节的功能，感觉AI就像一个新同事一样，不知道你在说什么。先要给它一个概括的描述，然后再层层展开。

效率提升效果

首先，我自己经常使用pywin32来操作excel，对于这个库比较熟悉，所以改起来就很快。如果你对这个库一无所知，AI还很难一次做对。

但是对于第二个脚本一共接近180行，如果我自己写、调试至少需要1天的时间。但是用AI生成，我自己只是改脚本，就变得轻松狠多。

所以AI对于有一定编程经验（但不需要达到高手的水平）的人来说，提升效果很明显。

标签： 人工智能 excel 自动化
本文转载自: https://blog.csdn.net/flashman911/article/details/142416181
版权归原作者 flashman911 所有， 如有侵权，请联系我们删除。
登录后发布评论

“AI帮我写python脚本，效率提升N倍”的评论:

还没有评论
关于作者
...
overfit同步小助手
文章同步
相关阅读

网络安全法-网络运行安全

使用selenium/drissionpage时如何阻止chrome自动跳转http到https

docker desktop 里部署的Open WebUI 管理员密码忘记了的处理方法

在ubuntu20.04中搭建onsite比赛运行环境

利用开源的低代码表单设计器FcDesigner高效管理和渲染复杂表单结构

Kafka学习笔记

【前端】浏览器输入url到页面呈现发生了什么？