From 73c5f23ed7cd1d85a3a1cdba932334b7d3e0d072 Mon Sep 17 00:00:00 2001 From: deathbybandaid Date: Tue, 15 Dec 2020 11:43:34 -0500 Subject: [PATCH] Add Tweak for future UDP variants --- fHDHR/device/tuners/__init__.py | 15 ++++++++++----- 1 file changed, 10 insertions(+), 5 deletions(-) diff --git a/fHDHR/device/tuners/__init__.py b/fHDHR/device/tuners/__init__.py index 7ddbb8d..86b651f 100644 --- a/fHDHR/device/tuners/__init__.py +++ b/fHDHR/device/tuners/__init__.py @@ -95,12 +95,17 @@ class Tuners(): if not stream_args["channelUri"]: raise TunerError("806 - Tune Failed") - channelUri_headers = self.fhdhr.web.session.head(stream_args["channelUri"]).headers - stream_args["true_content_type"] = channelUri_headers['Content-Type'] - - if stream_args["true_content_type"].startswith(tuple(["application/", "text/"])): + if stream_args["channelUri"].startswith("udp://"): + stream_args["true_content_type"] = "video/mpeg" stream_args["content_type"] = "video/mpeg" else: - stream_args["content_type"] = stream_args["true_content_type"] + + channelUri_headers = self.fhdhr.web.session.head(stream_args["channelUri"]).headers + stream_args["true_content_type"] = channelUri_headers['Content-Type'] + + if stream_args["true_content_type"].startswith(tuple(["application/", "text/"])): + stream_args["content_type"] = "video/mpeg" + else: + stream_args["content_type"] = stream_args["true_content_type"] return stream_args