record-camera-and-screen/RecordVideo.py

import datetime,time,sys,os,signal,re,winreg
from datetime import datetime
import subprocess,threading
from subprocess import CalledProcessError
# from multiprocessing import Process
from threading import Thread
import ctypes,inspect
import RecordType
from RecordType import *
import RecordConfig
from RecordConfig import *
import logging
import RunCMD
from RunCMD import get_ffmpeg_path
from winreg import HKEY_CURRENT_USER, OpenKey, QueryInfoKey, EnumValue, SetValueEx, CloseKey, REG_SZ, KEY_READ, KEY_SET_VALUE

class RecordVideo():


    '''
        ffmpeg -f dshow -i video="@device_pnp_\\\\?\\usb#vid_04f2&pid_b354&mi_00#7&30d7ad30&0&0000#{65e8773d-8f56-11d0-a3b9-00a0c9223196}\global":audio="@device_cm_{33D9A762-90C8-11D0-BD43-00A0C911CE86}\wave_{571529B3-7DB3-42A3-ADEF-BBD82925C15D}" -acodec libmp3lame -vcodec libx265 -preset:v ultrafast -tune:v zerolatency -s 1920x1080 -r 7 -y record_camera_20180408_182541.mkv
        
    '''    
    
    def __init__(self, record_video=True, record_voice=True):
        # print('视频录制初始化中...')
        # self.record_video=record_video
        # self.record_voice=record_voice
        #录制状态
        self.recording = False
        self.exception_exit = False
        self.record_type=RecordType.Camera
        #文件名称
        self.file_name='record'
        #文件后缀
        self.file_suffix='.mkv'
        self.process = None
        self.record_thread_name='record'
        self.record_thread=None
        
        self.file_dir = ''
        self.load()
        
    def load(self):
        #日志
        self.logger = logging.getLogger(__name__)
        self.logger.setLevel(level = logging.INFO)
        handler = logging.FileHandler('log.txt')
        handler.setLevel(logging.INFO)
        formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
        handler.setFormatter(formatter)    
        
        self.logger.addHandler(handler)
        
        self.load_config()
        
        
    def load_config(self):
  
        rc = RecordConfig()
        self.config = rc.config
        
        #摄像头名称
        self.camera_name=rc.config.get('devices','camera_device_name')
        #麦克风名称
        self.voice_device_name=rc.config.get('devices','voice_device_name')
        #录制屏幕名称
        self.screen_name=rc.config.get('devices','screen_device_name')
        #系统声音设备名称
        self.system_voice_device_name=rc.config.get('devices','system_voice_device_name')
        #视频编码
        self.video_codec=rc.config.get('record','vcodec')
        #分辨率
        self.resolution=rc.config.get('record','resolution')
        #是否自适应屏幕录制分辨率
        self.adaptive_screen_resolution = rc.config.getboolean('record', 'adaptive_screen_resolution')
        #帧率
        self.brate=rc.config.getfloat('record','frame_rate')
        #文件目录
        self.file_dir= os.path.abspath(rc.config.get('record','file_dir'))
        #线程数
        self.threads = rc.config.getint('record','threads')
        
        self.logger.info('camera device name: %s' % self.camera_name)
        self.logger.info('voice device name: %s' % self.voice_device_name)
        self.logger.info('screen device name: %s' % self.screen_name)
        self.logger.info('system voice device name: %s' % self.system_voice_device_name)
        self.logger.info('vcodec: %s' % self.video_codec)
        self.logger.info('resolution: %s' % self.resolution)
        self.logger.info('frame rate: %s' % self.brate)
        self.logger.info('save dir: %s' % self.file_dir)
        
        
    def start_ffmpeg(self,cmd, shell = True):
    
        try:
            print('录制中...')
            self.logger.info('录制中...')
            # print('cmd:\n%s' % cmd)
            start_time = datetime.now()
            self.process=subprocess.Popen(cmd, shell=shell, universal_newlines = True, stdin = subprocess.PIPE, stderr = subprocess.STDOUT, stdout = subprocess.PIPE)
            line = ''
            while self.recording:
                
                # print(cmd)
                # print(self.recording)
                # tmp_out = self.process.stdout.readline()
                line += str(self.process.stdout.readline())
                # print('test tmp out:%s' % tmp_out)
                #文字输出编码错误记录
                #异常：UnicodeDecodeError: 'gbk' codec can't decode byte 0xb4 in position 2881: illegal multibyte sequence
                #原因：cmd输出包含中文字符
                #解决方案：universal_newlines = False
                #缺陷：需要以字节形式的q来控制退出:write(b'q')
                #最终原因及解决方案：引起gbk编码错误的原因是文件名的中文与数字的连接符号由下划线'_'改成了横杠'-'。
                #为什么这个修改会引起运行时ffmpeg报编码错误，推测终究还是ffmpeg对中文编码的支持问题。
                #最终方案即文件名中的中文后的连接符号改回下划线。
                
                now = datetime.now()
                if (now - start_time).total_seconds() >2:
                    # self.logger.info('recording...')
                    ffmpeg_running = False
                    if self.process:
                        ffmpeg_running = self.process.poll() is None
                        log_text = 'ffmpeg 运行状态:%s' % ('运行中' if ffmpeg_running else '终止')
                        
                        print(log_text)
                        self.logger.info(log_text) 
                        
                    else:
                        txt= 'ffmpeg子进程已终止.'
                        print(txt)
                        self.logger.warning(txt)
                     
                    if not ffmpeg_running:
                        
                        print(self.process.communicate())
                        raise CalledProcessError(self.process.returncode, cmd)
                    
                    # self.logger.info(line)
                    print(line)
                    line = ''
                    start_time = now
                    
                # print(line)
                
                if not self.recording:
                    self.process.stdin.write('q')
                    print(self.process.communicate())
                    break
        
        except CalledProcessError as e:
            log_txt = 'ffmpeg异常终止:\nreturn code: %d\ncmd:\n%s' % (e.returncode, e.cmd)
            print(log_txt)
            self.logger.warning(log_txt)      
            self.recording = False
            self.exception_exit = True
            print('process is None?:%s' % (self.process is None))
            # self.stop_record()
        except Exception as x:
            print('未捕获的异常：')
            print(x)
        # self.logger.info(self.process.communicate())
        # print('over')
        
    def record(self, cmd='ffmpeg -h', target = None):
       
       if target:
            cmd = os.path.join(get_ffmpeg_path(), cmd)
            print('cmd: \n%s' % cmd)
            self.logger.info('record cmd:\n %s' % cmd)
            self.record_thread = Thread(name=self.record_thread_name, target= target, args = (cmd,), daemon=True)
            self.record_thread.start()
            self.recording=True
            self.exception_exit = False
            
            print('record thread,ident:%d' % self.record_thread.ident)
            # th.join()
                
    def stop_record(self):
        
        # print('threading active thread count:%d' % threading.active_count())
        try:
            
            self.recording = False
            self.logger.info('录制将停止...')
            if self.process:
                self.logger.info('ffmpeg进程状态: %s' % (self.process.poll() is not None))
                if self.process.returncode:
                    print('subprocess return code:%d' % self.process.returncode)
            print('record thread status: %s' % self.record_thread.is_alive())
            
            if self.record_thread.is_alive():
                
                self.record_thread.join(1)
            print('record thread status: %s' % self.record_thread.is_alive())
        except (Exception,KeyboardInterrupt) as e:
            print('kill exception:\n %s' % e)
            self.logger.warning('kill exception:\n %s' % e)
        
    def record_camera(self):

        if self.camera_name and self.voice_device_name:
            
            self.record_type=RecordType.Camera
            record_cmd='ffmpeg -f dshow -i video=\"%s\":audio=\"%s\" -acodec libmp3lame -vcodec %s -preset:v ultrafast -tune:v zerolatency -s %s -r %d -threads %d -y %s' %(
            self.deal_with_device_name(self.camera_name),
            self.deal_with_device_name(self.voice_device_name),
            self.video_codec,
            self.resolution,
            self.brate,
            self.threads,
            self.get_file_name()
            )
            # print(record_cmd)
            self.record(record_cmd, self.start_ffmpeg)
    
    def get_screen_device(self):
        pass
    
    def record_screen(self, resolution='1024x768'):
        if self.screen_name and self.system_voice_device_name:
            self.record_type=RecordType.Screen
            if self.adaptive_screen_resolution is not True:
                resolution = self.resolution
            device_cmd_str = ''
            if self.screen_name.lower().find('gdigrab') >=0:
                #使用gdigrab录制屏幕
                device_cmd_str = '-f dshow -i audio="{}" -f gdigrab -i desktop'.format(self.system_voice_device_name)
            else:
                device_cmd_str = '-f dshow -i video="{}":audio="{}"'.format(self.screen_name, self.system_voice_device_name)
            record_cmd='ffmpeg {} -acodec libmp3lame -vcodec {} -preset:v ultrafast -tune:v zerolatency -s {} -r {} -threads {} -y {}'.format(
            device_cmd_str,
            self.video_codec,
            # '1024x768', #屏幕录制分辨率固定
            resolution,
            self.brate,
            self.threads,
            self.get_file_name()
            )
            self.record(record_cmd, self.start_ffmpeg)    
    
    def check_device(self):
        #简单验证摄像头设置是否为空
        ready = True
        l_msg = ''
        if not self.camera_name:
            ready = False
            l_msg += '摄像头设备为空\n'
        
        if not self.voice_device_name:
            ready = False
            l_msg += '麦克风设备为空\n'
           
        if not self.screen_name:
            ready = False
            l_msg += '屏幕录制驱动为空\n'
        
        if not self.system_voice_device_name:
            ready = False
            l_msg += '系统声音录制驱动为空\n'
        
        if ready:
            l_msg = '设备检测正常'
        print(l_msg)
        self.logger.info(l_msg)
        return ready
                    
             
    def check_run_state(self):
        #验证运行有效性逻辑：
        #一、判断是否正常安装
        #判断条件：软件安装时在注册表“HKEY_CURRENT_USER\\SOFTWARE\\Gutin\\Record“记录下安装目录
        #二、非正常安装有效时间为三个月,且只能发生一次
        
        #验证当前运行目录是否存在注册表中
        qualified = False
        run_dir = os.path.abspath('.')
        # print('run_dir:%s' % run_dir)
        reg_path = 'SOFTWARE\\Gutin\\Record'
        feature_name ='InstallDir'
        key = OpenKey(HKEY_CURRENT_USER, reg_path, access = KEY_READ)
        items = QueryInfoKey(key)
        for i in range(items[1]):
            item = EnumValue(key, i)
            name = item[0]
            value = item[1]
            type = item[2]
            if name and name.find(feature_name)>=0:
                if os.path.samefile(value, run_dir):
                    qualified = True
                    break;
        
        if not qualified:
            #查找非正常安装目录记录
            #记录以运行目录的hash值作为键名，值为首次运行的时间
            time_format= '%Y-%m-%d %H:%M:%S'
            run_time = None
            unqualified_key_name  = 'Unqualified'
            has_unqualified = False
            for i in range(items[1]):
                item = EnumValue(key, i)
                name = item[0]
                value = item[1]
                type = item[2]
                if name == unqualified_key_name:
                    has_unqualified = True
                    run_time = value
          
            if not has_unqualified:
                #如果不存在，创建       
                run_time = datetime.now().strftime(time_format)
                key = OpenKey(HKEY_CURRENT_USER, reg_path, access = KEY_SET_VALUE)
                SetValueEx(key, unqualified_key_name, 0, REG_SZ, run_time)            
                      
            #判断时限
            # now = datetime.strptime('2018-06-22 12:11:51', time_format)
            now = datetime.now()
            run_time_obj = datetime.strptime(run_time, time_format)
            print('first run_time:%s' % run_time)
            print('now:%s' % now.strftime(time_format))
            
            qual_days = 91 - (now - run_time_obj).days
            # print('qualified days:%d' % (qual_days))
            # qual_hours = (now - run_time_obj).total_seconds() // 3600
            if qual_days > 0:
            # if 5 - qual_hours > 0:
                qualified = True
         
        CloseKey(key)
        return qualified

    
    def debug_camera(self):
        try:
            play_cmd = ['ffplay','-f','dshow','-i','video={}'.format(self.camera_name),'-window_title','按q退出','-noborder']
            self.record(play_cmd, self.play)
        except Exception as e:
            print(e)
        
    def play(self, cmd):
        try:
            t_process=subprocess.Popen(cmd, shell= False, universal_newlines = True, stderr = subprocess.STDOUT, stdout = subprocess.PIPE)
            while True:
                 line = t_process.stdout.readline()
                 print(line)
                 if line == '':
                    if t_process.poll() is not None:
                        break               
            t_process.communicate()
        except (Exception, KeyboardInterrupt) as e:
            print(e)
            
    def deal_with_device_name(self,device_name):    
        # print(device_name)
        # new_name=device_name.replace('\\','\\\\')
        # print(new_name)
        # return new_name
        return device_name
        
    def get_file_name(self):
        date_dir = datetime.now().strftime('%Y-%m-%d')
        time_str = datetime.now().strftime('%Y-%m-%d-%H%M%S')
        video_type_name = ''
        if self.record_type == RecordType.Camera:
            # video_type_name = 'camera'
            video_type_name = '摄像头'
        if self.record_type == RecordType.Screen:
            video_type_name = '屏幕'
            # video_type_name = 'screen'
        
        today_file_dir = os.path.join(self.file_dir, date_dir)
        if not os.path.exists(today_file_dir):
            os.mkdir(today_file_dir)
        file_name = os.path.join(today_file_dir, '{}_{}{}'.format(video_type_name, time_str, self.file_suffix))
        print('recording file name: %s' % file_name)
        return file_name
        
    def _async_raise(self, tid, exctype):  
        """raises the exception, performs cleanup if needed"""  
        tid = ctypes.c_long(tid)  
        if not inspect.isclass(exctype):  
            exctype = type(exctype)  
        res = ctypes.pythonapi.PyThreadState_SetAsyncExc(tid, ctypes.py_object(exctype))  
        print('async_raise res value:%d' % res)
        if res == 0:  
            raise ValueError("invalid thread id")  
        elif res != 1:  
            # """if it returns a number greater than one, you're in trouble,  
            # and you should call it again with exc=NULL to revert the effect"""  
            ctypes.pythonapi.PyThreadState_SetAsyncExc(tid, None)  
            raise SystemError("PyThreadState_SetAsyncExc failed")  
    
    def kill_process(self, process_name='ffmpeg'):
            cmd='tasklist | findstr {}'.format(process_name)
            output_strs, output_errs=self.run_cmd(cmd)
            pid=[]
            if output_strs:
                print('find "%s" result: \n%s' % (process_name, ''.join(output_strs)))
                for output_str in output_strs:
                    find_re=re.search(r'({}.+?)\s*([0-9]+)'.format(process_name),output_str)
                    if find_re:
                        full_process_name=find_re.group(1).strip()
                        pid=find_re.group(2).strip()
                        print('计划结束任务：{}@pid： {}'.format( full_process_name, pid ))
                        task_kill_cmd = 'taskkill /T /F /pid {}'.format(pid)
                        # status, output = subprocess.getstatusoutput(task_kill_cmd)
                        
                        # if status == 1:
                            # print('任务成功被结束:')
                        # else:
                            # print('任务结束失败:')
                        # print(output)   
                
            else:
                print('not found task about "%s" in tasklist' % process_name )
    
 
        # def stop_thread(self,thread):  
        # self._async_raise(thread.ident, SystemExit)