当前位置 主页 > 网站技术 > 代码类 >

    python实现实时视频流播放代码实例

    栏目:代码类 时间:2020-01-11 12:07

    这篇文章主要介绍了python实现实时视频流播放代码实例,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下

    @action(methods=['GET'], detail=True)
      def video(self, request, pk=None):
        """
        获取设备实时视频流
        :param request:
        :param pk:
        :return:
        """
        device_obj = self.get_object()
    
        # if device_obj.status == 0:
        #   return Response({'error': '设备离线'})
    
        if not device_obj.rtsp_address:
          return Response({'error': '缺少rtsp地址'})
    
        cache_id = '_video_stream_{}'.format(device_obj.hash)
        cache_status = cache.get(cache_id, None)
        if cache_status is None: # 任务初始化,设置初始时间
          cache.set(cache_id, time.time(), timeout=60)
    
        elif isinstance(cache_status, float) and time.time() - cache_status > 30: # 任务已超时, 返回错误信息, 一段时间内不再入队
          return Response({'error': '连接数目超过限制, 请稍后再试'})
    
        ret = job_queue.enqueue_video(rtsp_address=device_obj.rtsp_address, device_hash=device_obj.hash)
    
        logger.info('fetch device %s video job status: %s', pk, ret._status)
    
        if ret._status == b'started' or 'started': # 视频流正常推送中, 刷新播放时间, 返回视频ID
          cache.set(cache_id, 'continue', timeout=30)
          return Response({'video': ''.join([settings.FFMPEG_VIDEO, device_obj.hash])})
    
        elif ret._status == b'queued' or 'queued': # 视频任务等待中
          return Response({'status': '等待建立视频连接'})
    
        else: # 建立视频任务失败
          return Response({'error': '打开视频失败'})
    class JobQueue:
      """实时视频播放"""
      def __init__(self):
        self.video_queue = django_rq.get_queue('video') # 视频推流消息队列
    
      def enqueue_video(self, rtsp_address, device_hash):
        """视频流队列"""
        job_id = 'video_{}'.format(device_hash)
        job = self.video_queue.fetch_job(job_id)
    
        if not job:
          job = self.video_queue.enqueue_call(
            func='utils.ffmpeg.ffmpeg_play',
            args=(rtsp_address, device_hash),
            timeout=-1,
            ttl=30, # 最多等待30秒
            result_ttl=0,
            job_id=job_id
          )
    
        return job
    # -*- coding: utf-8 -*-
    
    import subprocess
    import threading
    import time
    import logging
    
    from django.core.cache import cache
    
    
    logger = logging.getLogger('server.default')
    
    
    def ffmpeg_play(stream, name):
    
      play = True
      cache_id = '_video_stream_{}'.format(name)
      cache.set(cache_id, 'continue', timeout=30)
      process = None
    
      def upstream():
        cmd = "ffmpeg -i '{}' -c:v h264 -f flv -r 25 -an 'rtmp://127.0.0.1:1935/live/{}'".format(stream, name)
        process = subprocess.Popen(cmd, shell=True, stdin=subprocess.PIPE, stderr=subprocess.DEVNULL)
        try:
          logger.info('device: {} stream thread start: {}'.format(name, stream))
          while play:
            time.sleep(1)
    
        except Exception as e:
          logger.info('device: {} stream thread error {}'.format(name, e))
    
        finally:
          logger.info('device: {} stream thread stop'.format(name))
          process.communicate(b'q')
    
      thr = threading.Thread(target=upstream)
      thr.start()
    
      try:
        while True:
          play = cache.get(cache_id, '')
          if play != 'continue':
            logger.info('stop device {} video stream'.format(name))
            play = False
            break
          time.sleep(1)
    
      except Exception as e:
        logger.info('device: {} play stream error {}'.format(name, e))
        process.communicate(b'q')
    
      logger.info('wait device {} video thread stop'.format(name))
      thr.join()
      logger.info('device {} video job stop'.format(name))