rtsp_client.py

"""Audio-enabled MJPEG RTSP client convenience wrapper.

This preserves the original `rtsp_client.py` workflow while using the generic
multitrack client so the default camera / display path validates both video and
audio delivery.
"""

import sys

from rtsp_client_multitrack import main as multitrack_main


def has_option(argv, option):
    return any(arg == option or arg.startswith(f"{option}=") for arg in argv)


def build_argv(argv):
    forwarded = list(argv)
    expect_audio = True
    if "--no-audio-check" in forwarded:
        forwarded.remove("--no-audio-check")
        expect_audio = False

    if not has_option(forwarded, "--path"):
        forwarded.extend(["--path", "/mjpeg/1"])
    if not has_option(forwarded, "--service-name"):
        forwarded.extend(["--service-name", "python rtsp server"])
    if not has_option(forwarded, "--require-decoded-video"):
        forwarded.append("--require-decoded-video")
    if expect_audio and not has_option(forwarded, "--expect-audio"):
        forwarded.append("--expect-audio")
    return forwarded


if __name__ == "__main__":
    sys.exit(multitrack_main(build_argv(sys.argv[1:])))