Как сохранять RTSP поток с камеры на Pythone?

Question

[Виталий Питаленко]

Как сохранять RTSP поток с камеры на Pythone? Необходимо чтобы скрипт сохранял поток в mp4 кусками по 5 минут и перезаписывал если память переполнена (как регистратор)

Answer 1

[qlkvg]

Быстро и топорно - взять ffmpeg и каждые 5 минут запускать что-то вроде

ffmpeg -i rtsp://@x.x.x.x:yyyy -acodec copy -vcodec copy c:/file.mp4

У питона есть обвязка для ffmpeg, но можно и без нее обойтись.

Долго и нудно - взять gstreamer и реализовать на нем.

Comments

[Виталий Питаленко]

Это неплохой вариант, но он будет до победы писать пока не заполнит, а потом будет переполнение и стоп. Я находил варианты при помощи cv2 библиотеки но снова же он будет делать до победы

[Александр]

Виталий Питаленко, Вам же написали "быстро и топорно", делаете свою обвязку на питоне
1) функция запуска каждые 5 мин рестартует команду ffmpeg с генарцией выходного файла вида datetime.mp4
2) другая функция в отдельном потоке проверяет место на диске, или кол-во файлов, или дату по файлу, тут смотря какой порог настроите, и удаляет старые записи.

Не уверен что даже питон нужен, тут скрипт на баше строк в 30 примерно.

[Виталий Питаленко]

Александр, пока решил только часть с записью. Может не очень красиво но я думаю для начинающего сойдет так)

import time as t
import subprocess
import os


while True:
    path0 = 'stream/'
    if os.path.isdir(path0):
        print('Done')
    else:
        os.mkdir(path0)
        t.sleep(1)
    path1 = 'stream/' + t.strftime("%Y-%m-%d", t.localtime())
    if os.path.isdir(path1):
        print('Done')
    else:
        os.mkdir(path1)
        t.sleep(1)
    a = t.strftime("%H.%M.%S", t.localtime())
    command = 'cvlc "rtsp://192.168.200.201:554/user=admin&password=&channel=1&stream=0?.sdp" --sout=file/ts:' + str(path1) + '/' + a + 'stream.mp4'
    process = subprocess.Popen(command, stdout=subprocess.PIPE, shell=True)
    pid = process.pid + 1
    print(pid)
    t.sleep(10)
    process.kill()
    command_k = 'kill -9 ' + str(pid)
    process_k = subprocess.Popen(command_k, stdout=subprocess.PIPE, shell=True)
    t.sleep(5)

Answer 2

[Viktor T2]

https://pypi.org/project/rtsp/

Comments

[Виталий Питаленко]

А есть какая-либо документация по данному модулю?

[Viktor T2]

Виталий Питаленко, используйте лучше opencv, захват видео в одну команду
stream = cv2.VideoCapture(rtsp_server_uri)
И доков, примеров сильно больше