# -*- coding: utf-8 -*- # Form implementation generated from reading ui file 'Screenshot_GUI.ui' # # Created by: PyQt5 UI code generator 5.15.4 # # WARNING: Any manual changes made to this file will be lost when pyuic5 is # run again. Do not edit this file unless you know what you are doing. import shutil import sys from datetime import datetime from pathlib import Path from PIL import ImageGrab from PyQt5.QtCore import QRect, Qt, QPoint, QMetaObject, QThread from PyQt5.QtGui import QIcon, QFont, QColor, QImage, QPixmap, QPen, QPainter from PyQt5.QtWidgets import (QMainWindow, QApplication, QDesktopWidget, QWidget, QFrame, QLabel, QPushButton, QComboBox, QCheckBox, QSpinBox, QDialog, QFileDialog, QMessageBox) from py7zr import pack_7zarchive, unpack_7zarchive from Screenshot.Audio_record import AudioRecord from Screenshot.Screenshot_record import Screenshot, file_path def unpack_7zip_to_exe(filename): file_basename = Path(file_path(f'{filename}.exe')) if file_basename.exists(): return True elif file_basename.with_suffix('.7z').exists(): shutil.unpack_archive(file_basename.with_suffix('.7z'), file_basename.parent) return unpack_7zip_to_exe(file_basename.stem) else: return False class Thread_screenshot(QThread): def __init__(self): super().__init__() self.offset_x = 0 self.offset_y = 0 self.draw_mouse = 1 self.filename = '' self.screen = Screenshot() def config_modify(self, **kwargs): self.screen.width = kwargs.get('width', self.screen.width) self.screen.height = kwargs.get('height', self.screen.height) self.screen.fps = kwargs.get('fps', self.screen.fps) self.offset_x = kwargs.get('offset_x', self.offset_x) self.offset_y = kwargs.get('offset_y', self.offset_y) self.draw_mouse = kwargs.get('draw_mouse', self.draw_mouse) self.filename = kwargs.get('filename', self.filename) def compose(self): video = self.filename + '.mp4' audio = self.filename + '.mp3' self.screen.compose_audio(video, audio, video.replace('缓存', '')) QThread.sleep(1) self.stop() Path(audio).unlink() Path(video).unlink() def run(self): if not self.filename: raise Exception('请输入filename!') filename = self.filename + '.mp4' self.screen.record(filename, self.offset_x, self.offset_y, self.draw_mouse) def stop(self): self.screen.terminate() class Ui_MainWindow1(QMainWindow): """主体窗口设置""" def __init__(self): super().__init__() self.setObjectName("MainWindow") self.setWindowTitle('屏幕录制') windowico = file_path('gui_svg.svg') self.setWindowIcon(QIcon(windowico)) self.resize(512, 352) self.setFixedSize(512, 352) self._dir = '.' self.fname = '' self.offset_x, self.offset_y = 0, 0 self.screen_x, self.screen_y = ImageGrab.grab().size self.audio = AudioRecord() self.audio.get_in_out_devices() self.device_name = None self.record_audio = True if self.audio.input_dict else False self.hotkey_start = False self.screenshot = Thread_screenshot() self.setupUi() ffmpeg_exists = unpack_7zip_to_exe('ffmpeg') if not ffmpeg_exists: QMessageBox.warning(self, '请下载ffmpeg！', '请将ffmpeg.exe放在resource目录下再录制！') self.show() def setupUi(self): self.centralwidget = QWidget(self) self.centralwidget.setObjectName("centralwidget") self.label0 = QLabel(self.centralwidget) self.label0.setGeometry(QRect(0, 0, 421, 351)) self.label0.setObjectName("scrollArea") # 添加边框 self.label0.setFrameShape(QFrame.Box) self.label0.setFrameShadow(QFrame.Sunken) self.get_screen_grab((0, 0, self.screen_x, self.screen_y)) self.label0.setAlignment(Qt.AlignCenter) self.label_1 = QLabel(self.centralwidget) font = QFont() font.setBold(True) self.label_1.setGeometry(QRect(430, 0, 85, 31)) self.label_1.setTextFormat(Qt.AutoText) self.label_1.setFont(font) self.label_1.setScaledContents(True) self.label_1.setObjectName("屏幕录制器") self.label_6 = QLabel(self.centralwidget) self.label_6.setGeometry(QRect(430, 20, 85, 31)) self.label_6.setTextFormat(Qt.AutoText) self.label_6.setScaledContents(True) self.label_6.setObjectName("注释1") self.checkbox = QCheckBox(self.centralwidget) self.checkbox.setGeometry(QRect(430, 100, 85, 20)) self.checkbox.setChecked(True) self.checkbox.stateChanged.connect(self.mouse_draw) self.checkbox.setObjectName("录制鼠标") self.comboBox = QComboBox(self.centralwidget) self.comboBox.setGeometry(QRect(430, 200, 85, 20)) self.comboBox.setEditable(False) self.comboBox.addItems(self.audio.input_dict.keys()) self.comboBox.addItems(['无']) self.comboBox.setMaxVisibleItems(9) self.comboBox.currentTextChanged.connect(self.change_combox) self.comboBox.setObjectName("音频来源") self.comboBox_3 = QComboBox(self.centralwidget) self.comboBox_3.setGeometry(QRect(430, 260, 80, 20)) self.comboBox_3.addItems(['双声道', '单声道']) self.comboBox_3.currentIndexChanged.connect(self.change_combox) self.comboBox_3.setObjectName("声道设置") self.pushButton = QPushButton(self.centralwidget) self.pushButton.setGeometry(QRect(470, 320, 41, 31)) self.pushButton.clicked.connect(self.stop) self.pushButton.setObjectName("停止") self.pushButton_2 = QPushButton(self.centralwidget) self.pushButton_2.setGeometry(QRect(430, 320, 41, 31)) self.pushButton_2.setObjectName("开始") self.pushButton_2.clicked.connect(self.start) self.pushButton_3 = QPushButton(self.centralwidget) self.pushButton_3.setGeometry(QRect(430, 290, 80, 23)) self.pushButton_3.setObjectName("保存目录") self.pushButton_3.clicked.connect(self.open_dirpath) self.pushButton_4 = QPushButton(self.centralwidget) self.pushButton_4.setGeometry(QRect(430, 70, 80, 23)) self.pushButton_4.setCheckable(True) self.pushButton_4.setObjectName("Size") self.pushButton_4.clicked.connect(self.get_screen_area) self.spinBox = QSpinBox(self.centralwidget) self.spinBox.setValue(15) self.spinBox.setRange(10, 60) self.spinBox.setGeometry(QRect(430, 145, 80, 20)) self.spinBox.valueChanged.connect(self.screen_fps_mofify) self.spinBox.setObjectName("spinBox") self.label_2 = QLabel(self.centralwidget) self.label_2.setGeometry(QRect(430, 50, 54, 16)) self.label_2.setTextFormat(Qt.AutoText) self.label_2.setObjectName("选择区域") self.label_3 = QLabel(self.centralwidget) self.label_3.setGeometry(QRect(430, 125, 54, 16)) self.label_3.setObjectName("帧率") self.label_4 = QLabel(self.centralwidget) self.label_4.setGeometry(QRect(430, 240, 54, 16)) self.label_4.setObjectName("声道设置") self.label_5 = QLabel(self.centralwidget) self.label_5.setGeometry(QRect(430, 180, 54, 16)) self.label_5.setObjectName("音频来源") self.setCentralWidget(self.centralwidget) self.retranslateUi() QMetaObject.connectSlotsByName(self) def retranslateUi(self): self.pushButton.setText("停止") self.pushButton_2.setText("开始") self.pushButton_3.setText("保存目录") self.pushButton_4.setText("Size*Size") self.checkbox.setText("录制鼠标") self.label_1.setText("屏幕录制器") self.label_2.setText("选择区域：") self.label_3.setText("帧率：") self.label_4.setText("声道设置：") self.label_5.setText("音频来源：") self.label_6.setText("F7：开始/停止") def keyPressEvent(self, event): """监测键盘是否按下F3，如果已开始录制则停止，否则开始录制""" if event.key() == Qt.Key_F7: if not self.hotkey_start: self.start() else: self.stop() def change_combox(self, event): sendername = self.sender().objectName() if sendername == '声道设置': self.channels = 2 if event == 0 else 1 elif sendername == '音频来源': if event == '无': self.record_audio = False else: self.record_audio = True self.device_name = event def get_screen_grab(self, crop_size): # 获取屏幕截图并更新至self.label0中 screen = ImageGrab.grab().convert('RGBA') screen = screen.crop(box=crop_size) # box=(left, upper, right, lower) data = screen.tobytes("raw", "RGBA") qim = QImage(data, screen.size[0], screen.size[1], QImage.Format_RGBA8888) pix = QPixmap.fromImage(qim).scaled(self.label0.size(), aspectRatioMode=Qt.KeepAspectRatio) pix.detach() # QPixmap为另开线程操作的，避免程序崩溃，使用detach方法使用一个QImage对象直到QPixmap对象销毁 self.label0.setPixmap(pix) self.label0.repaint() return screen.size def get_screen_area(self): # 获取截图区间 SizeScreen = MousePaint() SizeScreen.exec_() self.offset_x, self.offset_y = SizeScreen.lastpoint end_x, end_y = SizeScreen.endpoint self.screen_x, self.screen_y = self.get_screen_grab((self.offset_x, self.offset_y, end_x, end_y)) self.screenshot.config_modify(offset_x=self.offset_x, offset_y=self.offset_y, width=self.screen_x, height=self.screen_y) self.pushButton_4.setText(f'{self.screen_x}*{self.screen_y}') def screen_fps_mofify(self, fps): self.screenshot.config_modify(fps=fps) def mouse_draw(self, checked): draw_mouse = 1 if checked else 0 self.screenshot.config_modify(draw_mouse=draw_mouse) def open_dirpath(self): fdir = QFileDialog.getExistingDirectory(self, '选择目录', self._dir) if not fdir: return None self._dir = fdir self.set_filename() def set_filename(self): self.fname = f'{self._dir}/屏幕录制缓存_{int(datetime.now().replace(microsecond=0).timestamp())}' self.screenshot.config_modify(filename=self.fname) self.audio = AudioRecord() self.setWindowTitle(f'屏幕录制 {self.fname}') def start(self): if self.hotkey_start: return if not self.fname: QMessageBox.warning(self, '请选择保存目录！', '请选择保存目录！') return self.screenshot.start() if self.record_audio: self.audio.stop_flag = False self.audio.run(filename=self.fname + '.mp3', name=self.device_name) self.hotkey_start = True def stop(self): if not self.hotkey_start: return self.audio.stop_flag = True self.hotkey_start = False self.screenshot.stop() if self.record_audio: self.screenshot.compose() else: filename = self.fname + '.mp4' target = filename.replace('缓存', '') Path(filename).replace(target) self.set_filename() class MousePaint(QDialog): """移动鼠标获取屏幕捕获范围""" def __init__(self): super().__init__() self.setMouseTracking(True) # 设置窗口布满整个屏幕 self.showFullScreen() # 设置窗体无边框 self.setWindowFlags(Qt.FramelessWindowHint) # 窗口置顶，无边框 # 设置背景透明 self.setWindowOpacity(0.5) self.initUI() self.setFocus() def initUI(self): self.setGeometry(*(QDesktopWidget().screenGeometry()).getRect()) self.pix = QPixmap() self.lastpoint = QPoint() self.endpoint = QPoint() self.pos = None self.bline = 0 def mousePressEvent(self, event): # 监听鼠标按压事件 if event.button() == Qt.LeftButton: self.lastpoint = event.x(), event.y() self.bline = 1 elif event.button() == Qt.RightButton: self.close() event.accept() def mouseReleaseEvent(self, event): # 监听鼠标释放事件 self.endpoint = event.x(), event.y() self.bline = 0 event.accept() self.close() def mouseMoveEvent(self, event): # 监听鼠标移动事件 if self.bline == 1: self.pos = event.x(), event.y() event.accept() self.update() def paintEvent(self, event): # 绘画事件 if self.bline == 1: pp = QPainter(self) pen = QPen() # 定义笔格式对象 pen.setWidth(5) # 设置笔的宽度 pen.setColor(QColor(255, 0, 0)) pp.setPen(pen) lpx, lpy = self.lastpoint pp.drawRect(lpx, lpy, self.pos[0] - lpx, self.pos[1] - lpy) event.accept() # def main(): # """运行函数""" # # 将7zip压缩格式添加到shutil中 # shutil.register_archive_format('7zip', pack_7zarchive, description='7zip archive') # shutil.register_unpack_format('7zip', ['.7z'], unpack_7zarchive, description='7zip archive') # app = QApplication(sys.argv) # app.setAttribute(Qt.AA_UseHighDpiPixmaps) # ui = Ui_MainWindow1() # sys.exit(app.exec_()) #if __name__ == '__main__': # main()