#!/usr/bin/env python3 # -*- coding: utf-8 -*- """ 屏幕录制工具 支持录制当前屏幕并保存为视频文件 """ import cv2 import numpy as np import pyautogui import threading import time from datetime import datetime import os class ScreenRecorder: def __init__(self, output_dir="recordings", fps=20, quality=80, region=None): """ 初始化屏幕录制器 Args: output_dir (str): 输出目录 fps (int): 帧率 quality (int): 视频质量 (1-100) region (tuple): 录制区域 (x, y, width, height)，None表示全屏录制 """ self.output_dir = output_dir self.fps = fps self.quality = quality self.recording = False self.paused = False self.video_writer = None self.thread = None self.region = region # 创建输出目录 if not os.path.exists(output_dir): os.makedirs(output_dir) # 获取屏幕尺寸 self.screen_size = pyautogui.size() print(f"屏幕尺寸: {self.screen_size}") # 设置录制区域 if self.region: x, y, width, height = self.region # 确保区域在屏幕范围内 x = max(0, min(x, self.screen_size[0] - 1)) y = max(0, min(y, self.screen_size[1] - 1)) width = min(width, self.screen_size[0] - x) height = min(height, self.screen_size[1] - y) self.region = (x, y, width, height) self.record_size = (width, height) print(f"录制区域: {self.region}") else: self.record_size = self.screen_size print("录制模式: 全屏录制") def start_recording(self, filename=None): """ 开始录制 Args: filename (str): 输出文件名，如果为None则自动生成 """ if self.recording: print("录制已在进行中") return if filename is None: timestamp = datetime.now().strftime("%Y%m%d_%H%M%S") filename = f"screen_record_{timestamp}.mp4" self.output_path = os.path.join(self.output_dir, filename) # 设置视频编码器 fourcc = cv2.VideoWriter_fourcc(*'mp4v') self.video_writer = cv2.VideoWriter( self.output_path, fourcc, self.fps, self.record_size ) self.recording = True self.paused = False # 在新线程中开始录制 self.thread = threading.Thread(target=self._record_loop) self.thread.daemon = True self.thread.start() print(f"开始录制: {self.output_path}") def _record_loop(self): """ 录制循环 """ while self.recording: if not self.paused: # 截取屏幕 if self.region: # 区域录制 x, y, width, height = self.region screenshot = pyautogui.screenshot(region=(x, y, width, height)) else: # 全屏录制 screenshot = pyautogui.screenshot() # 转换为numpy数组 frame = np.array(screenshot) # 转换颜色格式 (RGB -> BGR) frame = cv2.cvtColor(frame, cv2.COLOR_RGB2BGR) # 写入视频文件 self.video_writer.write(frame) # 控制帧率 time.sleep(1.0 / self.fps) def pause_recording(self): """ 暂停录制 """ if not self.recording: print("当前没有在录制") return self.paused = not self.paused status = "暂停" if self.paused else "继续" print(f"录制{status}") def stop_recording(self): """ 停止录制 """ if not self.recording: print("当前没有在录制") return self.recording = False self.paused = False # 等待录制线程结束 if self.thread: self.thread.join() # 释放视频写入器 if self.video_writer: self.video_writer.release() self.video_writer = None print(f"录制完成: {self.output_path}") def set_region(self, region): """ 设置录制区域 Args: region (tuple): 录制区域 (x, y, width, height)，None表示全屏录制 """ if self.recording: print("录制进行中，无法更改区域设置") return False self.region = region # 重新计算录制区域 if self.region: x, y, width, height = self.region # 确保区域在屏幕范围内 x = max(0, min(x, self.screen_size[0] - 1)) y = max(0, min(y, self.screen_size[1] - 1)) width = min(width, self.screen_size[0] - x) height = min(height, self.screen_size[1] - y) self.region = (x, y, width, height) self.record_size = (width, height) print(f"录制区域已设置: {self.region}") else: self.record_size = self.screen_size print("录制模式已设置: 全屏录制") return True def get_status(self): """ 获取录制状态 Returns: dict: 包含录制状态信息的字典 """ return { 'recording': self.recording, 'paused': self.paused, 'output_path': getattr(self, 'output_path', None), 'screen_size': self.screen_size, 'record_size': self.record_size, 'region': self.region, 'fps': self.fps } def main(): """ 主函数 - 命令行界面 """ recorder = ScreenRecorder() print("\n=== 屏幕录制工具 ===") print("命令:") print(" start [filename] - 开始录制") print(" pause - 暂停/恢复录制") print(" stop - 停止录制") print(" status - 查看状态") print(" region x y w h - 设置录制区域 (x, y, 宽度, 高度)") print(" fullscreen - 设置全屏录制") print(" center w h - 设置居中区域录制 (宽度, 高度)") print(" quit - 退出程序") print("\n输入命令:") while True: try: command = input("> ").strip().split() if not command: continue cmd = command[0].lower() if cmd == "start": filename = command[1] if len(command) > 1 else None recorder.start_recording(filename) elif cmd == "pause": recorder.pause_recording() elif cmd == "stop": recorder.stop_recording() elif cmd == "status": status = recorder.get_status() print(f"录制状态: {'进行中' if status['recording'] else '已停止'}") print(f"暂停状态: {'是' if status['paused'] else '否'}") print(f"输出文件: {status['output_path']}") print(f"屏幕尺寸: {status['screen_size']}") print(f"录制尺寸: {status['record_size']}") print(f"录制区域: {status['region'] if status['region'] else '全屏'}") print(f"帧率: {status['fps']} FPS") elif cmd == "region": if len(command) != 5: print("用法: region x y width height") continue try: x, y, w, h = map(int, command[1:5]) if recorder.set_region((x, y, w, h)): print(f"录制区域已设置: ({x}, {y}, {w}, {h})") except ValueError: print("请输入有效的数字") elif cmd == "fullscreen": if recorder.set_region(None): print("已设置为全屏录制") elif cmd == "center": if len(command) != 3: print("用法: center width height") continue try: w, h = map(int, command[1:3]) screen_w, screen_h = recorder.screen_size x = (screen_w - w) // 2 y = (screen_h - h) // 2 if recorder.set_region((x, y, w, h)): print(f"居中录制区域已设置: ({x}, {y}, {w}, {h})") except ValueError: print("请输入有效的数字") elif cmd == "quit": if recorder.recording: recorder.stop_recording() print("程序退出") break else: print("未知命令") except KeyboardInterrupt: if recorder.recording: recorder.stop_recording() print("\n程序退出") break except Exception as e: print(f"错误: {e}") if __name__ == "__main__": main()