import os
import time
from pywinauto.application import Application
from pywinauto.keyboard import send_keys
import traceback
from pywinauto import mousedef get_app_and_main_window(app_path):app = Application(backend="uia").start(app_path)# app = Application(backend="uia").connect(path=app_path)# 等待应用程序加载time.sleep(1)# 获取主窗口main_window = app.window(title_re=".*Inpaint.*", control_type='Window')# main_window.print_control_identifiers()main_window.wait('visible', timeout=10)print("已连接到应用程序主窗口")return app, main_windowdef open_file_with_ctrl_o(app, file_path):"""使用 Ctrl+O 快捷键打开文件对话框并选择文件参数:app_path: 应用程序路径file_path: 要打开的文件路径"""try:# 激活窗口确保接收键盘输入main_window = app.top_window()main_window.set_focus()# # 使用 Ctrl+O 打开文件对话框# # 常用的控制快捷键说明 # SHIFT +,CTRL ^,ALT %,空格键 {SPACE},BACKSPACE {BACKSPACE}、# {BS} or {BKSP},BREAK {BREAK},CAPS LOCK {CAPSLOCK},# DEL or DELETE {DELETE} or {DEL},DOWN ARROW {DOWN},# END {END},ENTER {ENTER} or ~,ESC {ESC},HELP {HELP},# HOME {HOME},INS or INSERT {INSERT} or {INS},LEFT ARROW {LEFT},# NUM LOCK {NUMLOCK},PAGE DOWN {PGDN},# PAGE UP {PGUP},PRINT SCREEN {PRTSC},# RIGHT ARROW {RIGHT},SCROLL LOCK {SCROLLLOCK},# TAB {TAB},UP ARROW {UP},+ {ADD},- {SUBTRACT},# * {MULTIPLY},/ {DIVIDE},send_keys('^o') # ^ 代表 Ctrl 键print("已发送 Ctrl+O 打开文件对话框")# # 打印所有控件信息# # main_window.print_control_identifiers()# # 打开文件对话框，并点击打开main_window.child_window(title="文件名(N):", class_name="Edit").set_text(file_path)main_window.child_window(title="打开(O)", class_name="Button").click()print("已确认文件选择并打开")time.sleep(1)# 移动鼠标，左上角水印位置left_top_watermark_position = (742, 125)# 移动鼠标，右下角水印位置right_down_top_watermark_position = (824, 125)mouse.move(coords=left_top_watermark_position)# 指定位置，鼠标左击mouse.click(button='left', coords=left_top_watermark_position)# 将属性移动到（140，40）坐标处按下mouse.press(button='left', coords=left_top_watermark_position)# 将鼠标移动到（300，40）坐标处释放，mouse.release(button='left', coords=right_down_top_watermark_position)# 移动鼠标，左上角水印位置left_top_watermark_position_2 = (1133, 970)# 移动鼠标，右下角水印位置right_down_top_watermark_position_2 = (1250, 1000)mouse.move(coords=left_top_watermark_position_2)# 指定位置，鼠标左击mouse.click(button='left', coords=left_top_watermark_position_2)# 将属性移动到（140，40）坐标处按下mouse.press(button='left', coords=left_top_watermark_position_2)# 将鼠标移动到（300，40）坐标处释放，mouse.release(button='left', coords=right_down_top_watermark_position_2)print("水印区域框选完成")top_window=app.top_window()# top_window.print_control_identifiers()# 点击处理图像chuli_window=top_window.child_window(title="处理图像", control_type="Button")chuli_window.click()time.sleep(10)# 保存send_keys('^s') # ^ 代表 Ctrl 键print("已发送 Ctrl+s 保存文件对话框")# 另存为# top_window.child_window(title="另存为", control_type="Button").click() #这里执行click后，程序会阻塞，暂时忽略# top_window_2=app.top_window()# top_window_2.print_control_identifiers()## # # 打开文件对话框，并点击打开# top_window_2.set_focus()# top_window_2.child_window(title="文件名(N):", class_name="Edit").set_text(out_file_path)# top_window_2.child_window(title="保存(S)", class_name="Button").click()return Trueexcept Exception as e:# print(f"操作过程中出现错误: {e}")traceback.print_exc()return Falsedef batch_replace_image(app,main_window,file_path):try:for i in os.listdir(file_path):if i.endswith('.png') or i.endswith('.jpg'):image_path = os.path.join(file_path, i)open_file_with_ctrl_o(app,image_path)# print(i)except Exception as e:# print(f"操作过程中出现错误: {e}")traceback.print_exc()return Falsedef app_kill(app):try:app.kill()print("已关闭应用程序")except Exception as e:print(f"关闭应用程序时出现错误: {e}")# 使用示例
if __name__ == "__main__":inpaint_path = r'D:\exe安装包\Inpaint_v9.1_x64.exe'file_path = r'E:\AI生成\AI绘本\10月15日'app, main_window = get_app_and_main_window(inpaint_path)batch_replace_image(app,main_window,file_path)app_kill(app)
有问题可以随时反馈