Update run.py

dvr/ffmpeg/run.py

-import json, os, socket, time, datetime, pytz, subprocess as SP
+import json, os, socket, time, datetime, pytz, cv2 as cv, subprocess as SP
 from paho.mqtt.client import Client as MQTT
 from colorama import Fore, Back, Style
 
@@ -75,7 +75,7 @@ while True:
                 
                 rtsp_in = "{0}/{1}".format(rtsp_in, key)
                 
-                p = SP.Popen([
+                command = [
                     'ffmpeg',                 # DVR to EasyDarwin
                     '-rtsp_transport', 'tcp', # 指定來源協定
                     '-i', rtsp_in,            # 指定來源路徑
@@ -83,9 +83,14 @@ while True:
                     '-an',                    # 指定輸出聲音編碼 (an: 無聲音)
                     '-rtsp_transport', 'tcp', # 指定輸出協定
                     '-f','rtsp', rtsp_out,    # 指定輸出格式與路徑
-                ])
+                ]
+                
+                p = SP.Popen(command)
                 
                 try:
+                    stream = cv.VideoCapture(' '.join(command))
+                    
+                    if stream.isOpened():
                         payload['status'] = 1
                         
                         payload['msg'] = "Ready to Transfer."
@@ -93,6 +98,8 @@ while True:
                         client.publish(topic, json.dumps(payload), 1)
                     
                         p.communicate()
+                    else:
+                        stream.release()
 
                         payload['status'] = 0