import sys, datetime
import PyQt5
from PyQt5 import QtCore, QtGui, QtWidgets
from PyQt5.QtCore import Qt, QCoreApplication
from PyQt5.QtWidgets import QMessageBox
import RecordVideo,RecordType
from RecordVideo import *
from RecordType import *
import SettingWindow
from SettingWindow import *
import Shortcut
from Shortcut import *
import RecordTrayIcon
from RecordTrayIcon import *
import RecordConfig
from RecordConfig import *
import RecordHelp
from RecordHelp import *
class RecordWindow(QtWidgets.QWidget):
def __init__(self, parent = None, screen_resolution=None):
super(RecordWindow,self).__init__(parent)
self.setupUi()
self.load_modules()
#更新设置
self.need_update_config = False
self.screen_resolution = screen_resolution
# self.need_hide = True
#初始化状态
print('初始化状态...')
self.update_state()
def load_modules(self):
#录制
self.rv=RecordVideo()
#鼠标拖动
self.m_drag = False
#托盘图标
self.rti = RecordTrayIcon(self)
self.rti.update_state(self.recording, self.record_type)
self.sc = Shortcut()
self.update_setting(True)
# self.rc = RecordConfig()
# self.file_dir = self.rc.config.get('record','file_dir')
# self.debugCameraAction.triggered.connect(self.rv.debug_camera)
def closeEvent(self, event):
# print('close window.')
# self.close_signal.emit()
# self.close()
if self.recording:
self.stop_record(force = True)
# question = QMessageBox(self)
# question.setText('系统正在录制中，确定要退出吗？')
# question.setWindowTitle('提示')
# question.setIcon(QMessageBox.Question)
# question.addButton(QMessageBox.Yes)
# question.addButton(QMessageBox.No)
# question.setDefaultButton(QMessageBox.No)
# ret = question.exec()
# if ret == QMessageBox.Yes:
# self.stop_record()
# QCoreApplication.instance().quit()
# else:
# question = QMessageBox()
# question.setText('确定要退出吗？')
# question.setWindowTitle('提示')
# question.setIcon(QMessageBox.Question)
# question.addButton(QMessageBox.Yes)
# question.addButton(QMessageBox.No)
# question.setDefaultButton(QMessageBox.No)
# ret = question.exec()
# if ret == QMessageBox.Yes:
print('软件将退出.')
QCoreApplication.instance().quit()
# event.ignore()
def setupUi(self):
self.setObjectName("RecordWindow")
self.resize(94, 81)
self.move(1100,600)
self.pushButton = QtWidgets.QPushButton(self)
self.pushButton.setGeometry(QtCore.QRect(0, 30, 91, 51))
self.pushButton.setObjectName("pushButton")
# self.register_slot(self.pushButton.clicked, self.record)
# self.pushButton.mousePressEvent.connect(self.mousePressEvent)
# print(dir(self.pushButton))
#添加右键菜单
self.createContextMenu()
#添加计时器
self.lcd = QLCDNumber(self)
self.lcd.setDigitCount(10)
self.lcd.setMode(QLCDNumber.Dec)
self.lcd.setGeometry(QtCore.QRect(0, 0, 91, 31))
self.lcd.setSegmentStyle(QLCDNumber.Flat)
self.init_lcd()
#新建一个QTimer对象
self.timer = QTimer()
self.timer.setInterval(1000)
self.timer.timeout.connect(self.onTimerOut)
self.retranslateUi()
#调整窗体属性
self.setWindowFlags(QtCore.Qt.FramelessWindowHint | QtCore.Qt.Tool)
# self.setFocusPolicy(QtCore.Qt.StrongFocus)
self.installEventFilter(self)
def retranslateUi(self):
_translate = QtCore.QCoreApplication.translate
self.setWindowTitle(_translate("RecordWindow", "RecordWindow"))
self.pushButton.setText(_translate("RecordWindow", "开始"))
def init_lcd(self):
self.lcd.display('0:00:00')
def start_timer(self):
self.start_time = datetime.now()
self.init_lcd()
self.timer.start()
def stop_timer(self):
self.timer.stop()
self.init_lcd()
# 刷新录制时间
def onTimerOut(self):
print('on timer out monitor record status :%s' % self.recording)
if self.recording:
self.lcd.display(self.get_display_time(self.start_time))
else:
# self.stop_timer()
self.stop_record(force = self.exception_exit)
if self.exception_exit:
time.sleep(2)
self.record(self.record_type)
def get_display_time(self,old_time):
delta_time = datetime.now() - old_time
delta_time_str = str(delta_time)
pos = delta_time_str.find('.')
time_text = delta_time_str[0:pos]
return time_text
#打开文件目录
def open_file_dir(self):
dirpath = self.file_dir
if os.path.isdir(dirpath):
os.startfile(dirpath)
else:
print('错误的文件目录：%s' % dirpath)
#显示菜单
def showContextMenu(self):
self.contextMenu.exec_(QtGui.QCursor.pos())
#添加右键菜单
def createContextMenu(self):
#更改右键菜单为自定义
print('初始化右键菜单...')
self.setContextMenuPolicy(Qt.CustomContextMenu)
self.customContextMenuRequested.connect(self.showContextMenu)
self.contextMenu = QtWidgets.QMenu(self)
#停止/开始录制
self.recordSwitchAction = self.contextMenu.addAction('开始/停止录制')
#开始录制摄像头
self.recordCameraAction = self.contextMenu.addAction('开始录制摄像头')
#录制屏幕
self.recordScreenAction = self.contextMenu.addAction('开始录制屏幕')
#分隔栏
self.separatorAction = self.contextMenu.addAction('分隔栏')
self.separatorAction.setSeparator(True)
self.openFileDirAction = self.contextMenu.addAction('打开文件目录')
self.openFileDirAction.triggered.connect(self.open_file_dir)
#调试摄像头
# self.debugCameraAction = self.contextMenu.addAction('调试摄像头')
# self.debugCameraAction.setVisible(False)
#设置
self.settingAction=self.contextMenu.addAction('设置')
self.sw = SettingWindow()
self.settingAction.triggered.connect(self.show_setting)
#帮助
self.aboutAction=self.contextMenu.addAction('帮助')
self.rh = RecordHelp()
self.aboutAction.triggered.connect(self.rh.showWindow)
#退出
self.exitAction = self.contextMenu.addAction('退出')
self.exitAction.triggered.connect(self.close)
'''
功能事件
'''
@property
def recording(self):
return self.rv.recording
@property
def exception_exit(self):
return self.rv.exception_exit
@property
def record_type(self):
return self.rv.record_type
def register_slot(self, event_obj, new_action, disconnect = True):
if disconnect:
try:
event_obj.disconnect()
except Exception as e:
pass
event_obj.connect(new_action)
def update_state(self):
if self.recording:
print('录制状态：录制中.')
if self.record_type == RecordType.Camera:
print('正在录制摄像头.')
else:
print('正在录制屏幕.')
print('recording:%s' % self.recording)
print('record_type:%s' % self.record_type)
if self.recording:
self.recordSwitchAction.setText('停止录制')
self.register_slot(self.pushButton.clicked, self.stop_record)
self.register_slot(self.recordSwitchAction.triggered, self.stop_record)
#正在录制摄像头
if self.record_type == RecordType.Camera:
self.pushButton.setText('正在录制摄像头\n点击停止')
self.recordCameraAction.setText('停止录制摄像头')
self.register_slot(self.recordCameraAction.triggered, self.stop_record)
self.recordScreenAction.setText('开始录制屏幕')
self.register_slot(self.recordScreenAction.triggered, lambda: self.record(RecordType.Screen))
#正在录制屏幕
if self.record_type == RecordType.Screen:
self.pushButton.setText('正在录制屏幕\n点击停止')
self.recordScreenAction.setText('停止录制屏幕')
self.register_slot(self.recordScreenAction.triggered, self.stop_record)
self.recordCameraAction.setText('开始录制摄像头')
self.register_slot(self.recordCameraAction.triggered, lambda: self.record(RecordType.Camera))
else:
self.pushButton.setText('开始')
self.recordSwitchAction.setText('开始录制')
self.register_slot(self.pushButton.clicked, lambda: self.record(self.record_type))
self.register_slot(self.recordSwitchAction.triggered, lambda: self.record(self.record_type))
self.recordScreenAction.setText('开始录制屏幕')
self.register_slot(self.recordScreenAction.triggered, lambda: self.record(RecordType.Screen))
self.recordCameraAction.setText('开始录制摄像头')
self.register_slot(self.recordCameraAction.triggered, lambda: self.record(RecordType.Camera))
self.rti.update_state(self.recording, self.record_type)
def stop_record(self, force = False):
#force应用在两种情况：
#1.timer实时刷新ffmpeg进程状态，当出现异常退时force=true,即force=exception_exit。
#2.应用退出时
if self.recording or force:
self.stop_timer()
if self.record_type == RecordType.Camera:
print('停止录制摄像头.')
if self.record_type == RecordType.Screen:
print('停止录制屏幕.')
try:
self.rv.stop_record()
except KeyboardInterrupt as e:
print(e)
finally:
self.update_state()
# else:
# exit_tip = '系统即将退出'
# print(exit_tip)
# 退出系统
# self.close()
def record(self, rtype):
if self.recording:
if rtype == self.record_type:
return True
print('检测到正在录制，录制类型将切换...')
self.stop_record()
if self.rv.check_device():
#开始录制
if rtype == RecordType.Camera:
print('开始录制摄像头...')
self.rv.record_camera()
elif rtype == RecordType.Screen:
print('开始录制屏幕...')
self.rv.record_screen(resolution=self.screen_resolution)
self.start_timer()
self.update_state()
else:
# self.need_hide = False
# question = QMessageBox.information(self, '提示', '检测到录制设备缺失，无法进行录制，请先完善设备设置。', QMessageBox.Yes)
# self.need_hide = True
question = QMessageBox()
question.setText('检测到录制设备缺失，无法进行录制，请先完善设备设置。')
question.setWindowTitle('提示')
question.setIcon(QMessageBox.Question)
question.addButton(QMessageBox.Yes)
tmp_btn = question.button(QMessageBox.Yes)
tmp_btn.setText('确定')
question.adjustSize()
screen_center = QApplication.desktop().screenGeometry()
question.move(((screen_center.width() - question.width()) /2), ((screen_center.height() - question.height())/2))
# question.addButton(QMessageBox.No)
# question.setDefaultButton(QMessageBox.No)
ret = question.exec()
# if ret == QMessageBox.Yes:
# print('软件将退出.')
def show_setting(self):
# QWidget.connect(self.sw, update_setting(bool), self, self.update_setting(bool))
self.sw.update_setting.connect(self.update_setting)
self.sw.showSettingWindow()
def update_setting(self, changed):
if changed:
print('update setting..')
self.rv.load_config()
self.rc = RecordConfig()
self.load_shortcut()
self.file_dir = self.rc.config.get('record','file_dir')
''''
鼠标拖动窗体
'''
def mousePressEvent(self, e):
if not isinstance(self,RecordWindow):
e.ignore()
else:
if e.button() == Qt.LeftButton:
self.m_drag = True
self.m_DragPosition = e.globalPos() - self.pos()
e.accept()
self.setCursor(QtGui.QCursor(Qt.OpenHandCursor))
def mouseReleaseEvent(self, e):
if not isinstance(self,RecordWindow):
e.ignore()
else:
if e.button() == Qt.LeftButton:
self.m_drag = False
self.setCursor(QtGui.QCursor(Qt.ArrowCursor))
def mouseMoveEvent(self, e):
if not isinstance(self,RecordWindow):
e.ignore()
else:
if Qt.LeftButton and self.m_drag:
self.move(e.globalPos() - self.m_DragPosition)
e.accept()
# print('(x:%d/y:%d)' % (self.x(),self.y()))
'''
快捷键监听
'''
def load_shortcut(self):
self.sc.clear()
camera_key_group = self.rc.config.get('shortcut','camera')
screen_key_group = self.rc.config.get('shortcut','screen')
stop_record_key_group = self.rc.config.get('shortcut','stop')
camera_shortcut = [int(key) for key in camera_key_group.split(',')]
screen_shortcut = [int(key) for key in screen_key_group.split(',')]
stop_shortcut = [int(key) for key in stop_record_key_group.split(',')]
print('camera shortcut: %s' % camera_shortcut)
print('screen shortcut: %s' % screen_shortcut)
print('stop shortcut: %s' % stop_shortcut)
if camera_key_group:
self.sc.add(1, camera_shortcut, lambda: self.record(RecordType.Camera))
if screen_key_group:
self.sc.add(2, screen_shortcut, lambda: self.record(RecordType.Screen))
if stop_record_key_group:
self.sc.add(3, stop_shortcut, self.stop_record)
def monitor_shortcut(self):
if self.rv.check_run_state():
self.load_shortcut()
self.sc.monitor()
else:
question = QMessageBox(self)
question.setText('检测到软件非正常运行，以下功能将被禁用（如有疑问，请联系开发商）：
')
question.setWindowTitle('限制使用模式')
question.setIcon(QMessageBox.Warning)
question.addButton(QMessageBox.Yes)
# question.setButtonText(QMessageBox.Yes, QString('确定'))
btn = question.button(QMessageBox.Yes)
# print(btn.text())
btn.setText('确定')
# question.addButton('确定', QMessageBox.ButtonRole.AcceptRole)
# print('screen x:%d,y:%d; question x:%d,y:%d.' % (screen_center.x(), screen_center.y(), question.x(), question.y()))
# print(question.frameGeometry())
question.adjustSize()
screen_center = QApplication.desktop().screenGeometry()
#居中x和y的值计算与我想的不一样
#原因：question在exec之前无法准确的获取其size。(485,170)是exec之后的实际size。
question.move(((screen_center.width() - 485)/2), ((screen_center.height() - 170)/2))
# question.move((screen_center.width() / 2 - question.width()), (screen_center.height() /2 - question.height()))
# print('screen x:%d,y:%d; question x:%d,y:%d; question width:%d,height:%d.' % (screen_center.width(), screen_center.height(), question.geometry().width(), question.geometry().height(), question.width(), question.height()))
# question.move(QApplication.desktop().screenGeometry().center()- self.rect().center())
ret = question.exec()
'''
窗体事件
'''
def eventFilter(self, obj, event):
etype=event.type()
# print('e type: %s' % etype)
# print('obj is window? %s' % (isinstance(obj,RecordWindow)))
# print('deactivate type id :%d' % QEvent.WindowDeactivate)
if etype == QEvent.WindowDeactivate:
print('丢失焦点')
#代码功能：当窗口失去焦点（不是键盘事件的焦点），窗体隐藏。
#引起的问题：当关闭从此窗口弹出的提示框和子窗口时，此窗口异常退出。
#原因分析：提示框和子窗口在弹出时，引发WindowDeactivate事件主窗口同时隐藏，
#当提示框和子窗口关闭时，因为父窗口已隐藏子窗口似乎是找不到父窗口，或者猜测是误以为父窗口不存在，然后整个应用随即退出。
#解决方案1（先用）：用变量need_hide(bool)来控制是否需要隐藏窗体，need_hide=True才隐藏窗体。在打开子窗口和弹出提示窗之前将need_hide=False。
#缺陷：快捷键操作仍然会有同样的问题，因为窗体大多时候是隐藏的。（考虑：把快捷键监控放在托盘图标上？）
#更好的解决方案：1.为什么当父窗口是隐藏状态，子窗口关闭，整个应用也随之退出？(关键)
#2.对窗体事件有更准确的了解，换一种更好的方式来实现窗体失去焦点隐藏主窗体。
#最终解决方案：1.设置QApplication.setQuitOnLastWindowClosed(False)
#2显示逻辑：由此处和RecordTrayIcon.actived事件分别控制，此处负责隐藏；RecordTrayIcon根据此窗体隐藏状态做判断：如果隐藏则显示（即单击或双击显示），
#如果非隐藏但不是activeWindow，则设置activeWindow=True，否则隐藏。
self.setVisible(False)
# if self.need_hide:
# self.setVisible(False)
# pass
# else:
# self.need_hide = True
return False
if __name__ == '__main__':
app = QtWidgets.QApplication(sys.argv)
#当这个属性为True，应用程序会在最后一个可见窗口关闭时退出。
app.setQuitOnLastWindowClosed(False)
screen = app.desktop().screenGeometry()
resolution = '{}x{}'.format(screen.width(), screen.height())
rw = RecordWindow(screen_resolution=resolution)
rw.monitor_shortcut()
rw.rti.show()
rw.show()
sys.exit(app.exec_())