怎么建一个小说网站,一个人免费观看在线高清国语,开发网站,怎么制作网页链接春节假期在家无聊#xff0c;撸了三个小工具#xff1a;PC截图编辑/PC录屏(用于meeting录屏)/PC截屏文字识别。因为感觉这三个小工具是工作中常常需要用到的#xff0c;github上也有很多开源的#xff0c;不过总有点或多或少的小问题#xff0c;不利于自己的使用。脚本的编…春节假期在家无聊撸了三个小工具PC截图编辑/PC录屏(用于meeting录屏)/PC截屏文字识别。因为感觉这三个小工具是工作中常常需要用到的github上也有很多开源的不过总有点或多或少的小问题不利于自己的使用。脚本的编写尽量减少对三方库的使用。
已全部完成这是其中的一个后续将三个集成在在一个工具中。
import tkinter as tk
from tkinter import ttk, messagebox, font, filedialog
from PIL import Image, ImageTk, ImageGrab
import sys
import tempfile
import threading
from pathlib import Path
import ctypes
import logging.handlers
from datetime import datetime# 最小化控制台窗口
def minimize_console():ctypes.windll.user32.ShowWindow(ctypes.windll.kernel32.GetConsoleWindow(), 6)minimize_console() # 调用最小化函数# 获取脚本所在目录路径
def get_script_directory():return Path(__file__).parent# 配置日志文件路径和日志级别
log_file_path get_script_directory() / ocr_errors.log
logging.basicConfig(filenamelog_file_path,levellogging.DEBUG,format%(asctime)s - %(levelname)s - %(message)s
)
# 添加日志轮转
handler logging.handlers.RotatingFileHandler(log_file_path, maxBytes1024*1024*5, backupCount3)
logger logging.getLogger()
logger.addHandler(handler)# 保存临时图片到磁盘
def save_temp_image(image, suffix.png):with tempfile.NamedTemporaryFile(deleteFalse, suffixsuffix) as temp_file:image.save(temp_file.name)return Path(temp_file.name)class OCRApp:def __init__(self):try:self.root tk.Tk()self.root.withdraw()# 禁用最大化按钮self.root.resizable(False, True)self.screenshot Noneself.ocr_model None # 延迟初始化self.recognized_text self.main_frame Noneself.load_win None # 初始化 load_win 为 None# 启动后台线程加载OCR模型以优化性能使run脚本后能马上进入截图状态threading.Thread(targetself.load_ocr_model, daemonTrue).start()# 立即开始截图选择self.start_selection()except Exception as e:self.show_crash_message(f程序启动失败: {str(e)})sys.exit(1)def load_ocr_model(self):from paddleocr import PaddleOCRtry:self.ocr_model PaddleOCR(use_angle_clsTrue, show_logFalse, langch)except Exception as e:logger.error(fOCR模型加载失败: {str(e)})# 开始截图选择区域def start_selection(self):self.selection_win tk.Toplevel()self.selection_win.attributes(-fullscreen, True)self.selection_win.attributes(-alpha, 0.3)# 绑定整个窗口的 ESC 键事件self.selection_win.bind(Escape, self.on_escape)self.canvas tk.Canvas(self.selection_win,cursorcross,bggray30,highlightthickness0)self.canvas.pack(filltk.BOTH, expandTrue)self.start_x self.start_y 0self.rect_id Noneself.crosshair_ids []self.canvas.bind(Button-1, self.on_mouse_down)self.canvas.bind(B1-Motion, self.on_mouse_drag)self.canvas.bind(ButtonRelease-1, self.on_mouse_up)self.canvas.bind(Motion, self.on_mouse_move)self.escape_label tk.Label(self.selection_win,text按ESC键退出截图,fgyellow,bggray20,font(Helvetica, 12, bold))self.escape_label.place(x10, y10)self.update_crosshair(0, 0)# 鼠标按下事件处理def on_mouse_down(self, event):self.start_x event.xself.start_y event.yself.clear_crosshair()if self.rect_id:self.canvas.delete(self.rect_id)self.rect_id None# 鼠标拖动事件处理def on_mouse_drag(self, event):current_x event.xcurrent_y event.yif self.rect_id:self.canvas.coords(self.rect_id, self.start_x, self.start_y, current_x, current_y)else:self.rect_id self.canvas.create_rectangle(self.start_x, self.start_y,current_x, current_y,outlineblue, width2, fillgray75, tagsrect)# 鼠标释放事件处理def on_mouse_up(self, event):try:x1 min(self.start_x, event.x)y1 min(self.start_y, event.y)x2 max(self.start_x, event.x)y2 max(self.start_y, event.y)if (x2 - x1) 10 or (y2 - y1) 10:raise ValueError(选区过小请选择更大的区域)if (x2 - x1) self.canvas.winfo_width() or (y2 - y1) self.canvas.winfo_height():raise ValueError(选区过大请选择更小的区域)self.screenshot ImageGrab.grab(bbox(x1, y1, x2, y2))self.selection_win.destroy()self.initialize_ocr_and_process()except Exception as e:logger.error(f截图错误: {str(e)})messagebox.showerror(截图错误, str(e))self.restart_selection()# 初始化OCR引擎并处理截图def initialize_ocr_and_process(self):try:if self.ocr_model is None:self.load_win self.show_loading(OCR模型正在加载中请稍后...)self.root.after(100, self.check_ocr_model) # 每100毫秒检查一次else:self.process_ocr()self.setup_main_ui()self.root.deiconify()except Exception as e:logger.error(fOCR初始化失败: {str(e)})if self.load_win:self.load_win.destroy()self.handle_ocr_init_error(str(e))def check_ocr_model(self):if self.ocr_model is None:self.root.after(100, self.check_ocr_model) # 每100毫秒检查一次else:if self.load_win:self.load_win.destroy()self.process_ocr()self.setup_main_ui()self.root.deiconify()# 执行OCR处理def process_ocr(self):try:temp_image_path save_temp_image(self.screenshot)result self.ocr_model.ocr(str(temp_image_path), clsTrue)temp_image_path.unlink() # 确保临时文件被删除# 后处理识别结果合并同一行的文字merged_text self.merge_lines(result[0])self.recognized_text merged_textexcept Exception as e:logger.error(fOCR处理失败: {str(e)})messagebox.showerror(识别错误, fOCR处理失败: {str(e)})self.restart_selection()# 合并同一行的文字def merge_lines(self, ocr_result):merged_text []current_line []current_y1 Nonecurrent_y2 Noneline_threshold 5 # 设置行间距阈值可以根据需要调整for line in ocr_result:# 提取坐标点x1, y1 line[0][0] # 第一个坐标点x2, y2 line[0][2] # 第三个坐标点text line[1][0] # 提取文本if current_y1 is None or current_y2 is None:current_y1 y1current_y2 y2current_line.append(text)elif abs(y1 - current_y1) line_threshold and abs(y2 - current_y2) line_threshold:current_line.append(text)else:merged_text.append( .join(current_line))current_line [text]current_y1 y1current_y2 y2if current_line:merged_text.append( .join(current_line))return \n.join(merged_text)# 设置主界面UIdef setup_main_ui(self):if self.main_frame is None:self.main_frame ttk.Frame(self.root, padding20)self.main_frame.grid(row0, column0, stickynsew)self.root.grid_rowconfigure(0, weight1)self.root.grid_columnconfigure(0, weight1)# 使用 PanedWindow 来分割图片框和文本框self.paned_window ttk.PanedWindow(self.main_frame, orienttk.VERTICAL)self.paned_window.grid(row0, column0, stickynsew)# 创建一个 Frame 来包含图片和滚动条self.image_frame ttk.Frame(self.paned_window)self.image_frame.pack(filltk.BOTH, expandTrue)# 使用 Canvas 来显示图片并添加滚动条self.image_canvas tk.Canvas(self.image_frame, highlightbackgroundself.root.cget(bg), highlightthickness0)self.image_canvas.pack(sidetk.LEFT, filltk.BOTH, expandTrue)self.image_scrollbar ttk.Scrollbar(self.image_frame, orienttk.VERTICAL, commandself.image_canvas.yview)self.image_scrollbar.pack(sidetk.RIGHT, filltk.Y)self.image_canvas.config(yscrollcommandself.image_scrollbar.set)self.image_canvas.bind(Configure, self.on_canvas_configure)self.image_container ttk.Frame(self.image_canvas)self.image_container_id self.image_canvas.create_window((0, 0), windowself.image_container, anchornw)self.img_label ttk.Label(self.image_container)self.img_label.pack(filltk.BOTH, expandTrue)# 定义字体custom_font font.Font(familyMicrosoft YaHei, size9)self.text_area tk.Text(self.paned_window,wraptk.WORD,fontcustom_font, # 设置字体height15 # 初始高度设置为15行)self.text_area.pack(filltk.BOTH, expandTrue)self.paned_window.add(self.image_frame)self.paned_window.add(self.text_area)btn_frame ttk.Frame(self.main_frame)btn_frame.grid(row1, column0, stickyew, pady10)# 确保按钮行不会被压缩self.main_frame.grid_rowconfigure(0, weight1)self.main_frame.grid_rowconfigure(1, weight0)ttk.Button(btn_frame,text重新选择,commandself.restart_selection).pack(sidetk.LEFT, padx5)ttk.Button(btn_frame,text复制文本,commandself.copy_result).pack(sidetk.LEFT, padx5)ttk.Button(btn_frame,text保存图片,commandself.save_image).pack(sidetk.LEFT, padx5)ttk.Button(btn_frame,text退出,commandself.safe_exit).pack(sidetk.RIGHT, padx5)# 设置窗口标题self.root.title(文字识别PDM3)self.update_image_display()self.text_area.delete(1.0, tk.END)self.text_area.insert(tk.END, self.recognized_text.strip())self.update_text_area_height() # 更新文本框高度# 设置窗口总是最顶层self.root.attributes(-topmost, True)# 更新图片显示def update_image_display(self):if self.screenshot:photo ImageTk.PhotoImage(self.screenshot)self.img_label.config(imagephoto)self.img_label.image photo# 获取图片的实际大小img_width, img_height self.screenshot.size# 获取屏幕高度screen_height self.root.winfo_screenheight()# 计算图片框的最大高度max_image_height screen_height // 2# 设置 Canvas 的滚动区域self.image_canvas.config(scrollregion(0, 0, img_width, img_height))# 调整 image_canvas 的高度if img_height max_image_height:self.image_canvas.config(heightmax_image_height)else:self.image_canvas.config(heightimg_height)# 配置 Canvas 大小def on_canvas_configure(self, event):# 更新 Canvas 的滚动区域self.image_canvas.config(scrollregionself.image_canvas.bbox(all))# 显示加载中的窗口def show_loading(self, message):load_win tk.Toplevel()load_win.title(请稍候)frame ttk.Frame(load_win, padding20)frame.pack()ttk.Label(frame, textmessage).pack(pady10)progress ttk.Progressbar(frame, modeindeterminate)progress.pack(pady5)progress.start()return load_win# 处理OCR初始化错误def handle_ocr_init_error(self, error_msg):choice messagebox.askretrycancel(OCR初始化失败,f{error_msg}\n\n是否重试,iconerror)if choice:threading.Thread(targetself.initialize_ocr_and_process).start()else:self.safe_exit()# 重新开始截图选择def restart_selection(self):if self.root.winfo_exists():self.root.withdraw()self.screenshot Noneself.recognized_text self.clear_ui()self.start_selection()# 清理UI界面def clear_ui(self):if hasattr(self, img_label):self.img_label.config(image)self.img_label.image Noneif hasattr(self, text_area):self.text_area.delete(1.0, tk.END)# 复制识别结果到剪贴板def copy_result(self):self.root.clipboard_clear()self.root.clipboard_append(self.recognized_text)messagebox.showinfo(成功, 已复制到剪贴板)# 安全退出程序def safe_exit(self):if self.root.winfo_exists():self.root.destroy()sys.exit(0)# 显示程序崩溃错误信息def show_crash_message(self, message):crash_win tk.Tk()crash_win.withdraw()messagebox.showerror(致命错误, message)crash_win.destroy()# 按下ESC键时退出程序def on_escape(self, event):self.selection_win.destroy()self.safe_exit()# 鼠标移动事件处理def on_mouse_move(self, event):current_x event.xcurrent_y event.yself.update_crosshair(current_x, current_y)# 更新十字线位置def update_crosshair(self, x, y):self.clear_crosshair()self.crosshair_ids.append(self.canvas.create_line(0, y, self.canvas.winfo_width(), y,tagscrosshair, fillyellow, width2))self.crosshair_ids.append(self.canvas.create_line(x, 0, x, self.canvas.winfo_height(),tagscrosshair, fillyellow, width2))# 清除十字线def clear_crosshair(self):for crosshair_id in self.crosshair_ids:self.canvas.delete(crosshair_id)self.crosshair_ids []# 保存图片def save_image(self):if self.screenshot:# 获取用户桌面路径desktop_path Path.home() / Desktop# 生成当前日期和时间的字符串current_datetime datetime.now().strftime(%Y%m%d_%H%M%S)default_filename fscreenshot_{current_datetime}.pngfile_path filedialog.asksaveasfilename(initialdirdesktop_path, # 设置初始目录为用户桌面initialfiledefault_filename, # 设置默认文件名defaultextension.png,filetypes[(PNG files, *.png), (JPEG files, *.jpg), (All files, *.*)])if file_path:self.screenshot.save(file_path)messagebox.showinfo(保存成功, f图片已保存到 {file_path})# 更新文本框高度def update_text_area_height(self):# 计算当前文本行数line_count int(self.text_area.index(end-1c).split(.)[0])if line_count 15:self.text_area.config(height15) # 如果行数超过15行固定高度为15行else:self.text_area.config(heightline_count) # 否则根据内容调整高度# 运行主循环def run(self):self.root.mainloop()if __name__ __main__:app OCRApp()app.run()
