record-camera-and-screen/RecordWindow.py

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 *

class RecordWindow(QtWidgets.QWidget):

    def __init__(self, parent = None):
        super(RecordWindow,self).__init__(parent) 
        self.setupUi()
        self.load_modules()
        #更新设置
        self.need_update_config = False
        #初始化状态
        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.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:
        
            # 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.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.sw.showSettingWindow)  
        #帮助
        self.aboutAction=self.contextMenu.addAction('帮助')        
        #退出
        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):
    
        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 rtype == RecordType.Camera:
            print('开始录制摄像头...')
            self.rv.record_camera()
        elif rtype == RecordType.Screen:
            print('开始录制屏幕...')
            self.rv.record_screen()
              
        self.start_timer()
        self.update_state()

    ''''
        鼠标拖动窗体

    '''
        
    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 monitor_shortcut(self):
        
        sc = Shortcut()
        
        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:
            sc.add(1, camera_shortcut, lambda: self.record(RecordType.Camera))
        if screen_key_group:
            sc.add(2, screen_shortcut, lambda: self.record(RecordType.Screen))
        if stop_record_key_group:
            sc.add(3, stop_shortcut, self.stop_record)
        sc.monitor()
        
    '''
        窗体事件
    '''
    def eventFilter(self, obj, event):
        # print('e type: %s' % event.type())
        # print('deactivate type id :%d' % QEvent.WindowDeactivate)
        if event.type() == QEvent.WindowDeactivate:
            # print('丢失焦点')
            self.setVisible(False)
            
        return False
        
if __name__ == '__main__':  
    app = QtWidgets.QApplication(sys.argv)
    rw = RecordWindow()    
    rw.monitor_shortcut()
    rw.rti.show()
    rw.show()
    sys.exit(app.exec_())