fHDHR/fHDHR_NextPVR
1
0
Fork 0
mirror of https://github.com/fHDHR/fHDHR_NextPVR.git synced 2025-12-06 09:06:59 -05:00
Code Issues Releases Wiki Activity

Merge pull request #67 from deathbybandaid/dev

Browse Source 
Dev
This commit is contained in:
Deathbybandaid 2020-11-30 14:18:47 -05:00 committed by GitHub
commit 48db9dcbd1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
5 changed files with 156 additions and 122 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
fHDHR/device/tuners/stream/__init__.py
View File

@ -1,6 +1,7 @@
from .direct_stream import Direct_Stream from .direct_stream import Direct_Stream
from .direct_m3u8_stream import Direct_M3U8_Stream
from .ffmpeg_stream import FFMPEG_Stream from .ffmpeg_stream import FFMPEG_Stream
from .vlc_stream import VLC_Stream from .vlc_stream import VLC_Stream
@ -15,8 +16,12 @@ class Stream():
self.method = FFMPEG_Stream(fhdhr, stream_args, tuner) self.method = FFMPEG_Stream(fhdhr, stream_args, tuner)
if stream_args["method"] == "vlc": if stream_args["method"] == "vlc":
self.method = VLC_Stream(fhdhr, stream_args, tuner) self.method = VLC_Stream(fhdhr, stream_args, tuner)
elif stream_args["method"] == "direct": elif (stream_args["method"] == "direct" and
not self.stream_args["true_content_type"].startswith(tuple(["application/", "text/"]))):
self.method = Direct_Stream(fhdhr, stream_args, tuner) self.method = Direct_Stream(fhdhr, stream_args, tuner)
elif (stream_args["method"] == "direct" and
self.stream_args["true_content_type"].startswith(tuple(["application/", "text/"]))):
self.method = Direct_M3U8_Stream(fhdhr, stream_args, tuner)
def get(self): def get(self):
return self.method.get() return self.method.get()

102
fHDHR/device/tuners/stream/direct_m3u8_stream.py Normal file
View File

@ -0,0 +1,102 @@
import sys
import time
import m3u8
from Crypto.Cipher import AES
# from fHDHR.exceptions import TunerError
class Direct_M3U8_Stream():
def __init__(self, fhdhr, stream_args, tuner):
self.fhdhr = fhdhr
self.stream_args = stream_args
self.tuner = tuner
self.chunksize = int(self.fhdhr.config.dict["direct_stream"]['chunksize'])
def get(self):
if not self.stream_args["duration"] == 0:
self.stream_args["time_end"] = self.stream_args["duration"] + time.time()
self.fhdhr.logger.info("Detected stream URL is m3u8: %s" % self.stream_args["true_content_type"])
channelUri = self.stream_args["channelUri"]
while True:
self.fhdhr.logger.info("Opening m3u8 for reading %s" % channelUri)
videoUrlM3u = m3u8.load(channelUri)
if len(videoUrlM3u.playlists):
self.fhdhr.logger.info("%s m3u8 varients found" % len(videoUrlM3u.playlists))
channelUri = videoUrlM3u.playlists[0].absolute_uri
else:
break
def generate():
try:
played_chunk_urls = []
while self.tuner.tuner_lock.locked():
playlist = m3u8.load(channelUri)
segments = playlist.segments
if len(played_chunk_urls):
newsegments = 0
for segment in segments:
if segment.absolute_uri not in played_chunk_urls:
newsegments += 1
self.fhdhr.logger.info("Refreshing m3u8, Loaded %s new segments." % str(newsegments))
else:
self.fhdhr.logger.info("Loaded %s segments." % str(len(segments)))
if playlist.keys != [None]:
keys = [{"url": key.absolute_uri, "method": key.method, "iv": key.iv} for key in playlist.keys if key]
else:
keys = [None for i in range(0, len(segments))]
for segment, key in zip(segments, keys):
chunkurl = segment.absolute_uri
if chunkurl and chunkurl not in played_chunk_urls:
played_chunk_urls.append(chunkurl)
if (not self.stream_args["duration"] == 0 and
not time.time() < self.stream_args["time_end"]):
self.fhdhr.logger.info("Requested Duration Expired.")
self.tuner.close()
chunk = self.fhdhr.web.session.get(chunkurl).content
if not chunk:
break
# raise TunerError("807 - No Video Data")
if key:
if key["url"]:
keyfile = self.fhdhr.web.session.get(key["url"]).content
cryptor = AES.new(keyfile, AES.MODE_CBC, keyfile)
self.fhdhr.logger.info("Decrypting Chunk #%s with key: %s" % (len(played_chunk_urls), key["url"]))
chunk = cryptor.decrypt(chunk)
chunk_size = int(sys.getsizeof(chunk))
self.fhdhr.logger.info("Passing Through Chunk #%s with size %s: %s" % (len(played_chunk_urls), chunk_size, chunkurl))
yield chunk
self.tuner.add_downloaded_size(chunk_size)
if playlist.target_duration:
time.sleep(int(playlist.target_duration))
self.fhdhr.logger.info("Connection Closed: Tuner Lock Removed")
except GeneratorExit:
self.fhdhr.logger.info("Connection Closed.")
except Exception as e:
self.fhdhr.logger.info("Connection Closed: " + str(e))
finally:
self.tuner.close()
# raise TunerError("806 - Tune Failed")
return generate()

140
fHDHR/device/tuners/stream/direct_stream.py
View File

@ -1,8 +1,5 @@
import sys import sys
import time import time
import m3u8
from Crypto.Cipher import AES
# from fHDHR.exceptions import TunerError # from fHDHR.exceptions import TunerError
@ -21,125 +18,46 @@ class Direct_Stream():
if not self.stream_args["duration"] == 0: if not self.stream_args["duration"] == 0:
self.stream_args["time_end"] = self.stream_args["duration"] + time.time() self.stream_args["time_end"] = self.stream_args["duration"] + time.time()
if not self.stream_args["true_content_type"].startswith(tuple(["application/", "text/"])): self.fhdhr.logger.info("Direct Stream of %s URL: %s" % (self.stream_args["true_content_type"], self.stream_args["channelUri"]))
self.fhdhr.logger.info("Direct Stream of %s URL: %s" % (self.stream_args["true_content_type"], self.stream_args["channelUri"])) req = self.fhdhr.web.session.get(self.stream_args["channelUri"], stream=True)
req = self.fhdhr.web.session.get(self.stream_args["channelUri"], stream=True) def generate():
def generate(): try:
try: chunk_counter = 1
chunk_counter = 1 while self.tuner.tuner_lock.locked():
while self.tuner.tuner_lock.locked(): for chunk in req.iter_content(chunk_size=self.chunksize):
for chunk in req.iter_content(chunk_size=self.chunksize): if (not self.stream_args["duration"] == 0 and
not time.time() < self.stream_args["time_end"]):
req.close()
self.fhdhr.logger.info("Requested Duration Expired.")
self.tuner.close()
if (not self.stream_args["duration"] == 0 and if not chunk:
not time.time() < self.stream_args["time_end"]): break
req.close() # raise TunerError("807 - No Video Data")
self.fhdhr.logger.info("Requested Duration Expired.")
self.tuner.close()
if not chunk: chunk_size = int(sys.getsizeof(chunk))
break self.fhdhr.logger.info("Passing Through Chunk #%s with size %s" % (chunk_counter, chunk_size))
# raise TunerError("807 - No Video Data") yield chunk
self.tuner.add_downloaded_size(chunk_size)
self.fhdhr.logger.info("Passing Through Chunk #%s with size %s" % (chunk_counter, self.chunksize)) chunk_counter += 1
yield chunk
chunk_size = int(sys.getsizeof(chunk))
self.tuner.add_downloaded_size(chunk_size)
chunk_counter += 1 self.fhdhr.logger.info("Connection Closed: Tuner Lock Removed")
self.fhdhr.logger.info("Connection Closed: Tuner Lock Removed") except GeneratorExit:
self.fhdhr.logger.info("Connection Closed.")
except GeneratorExit: except Exception as e:
self.fhdhr.logger.info("Connection Closed.") self.fhdhr.logger.info("Connection Closed: " + str(e))
except Exception as e: finally:
self.fhdhr.logger.info("Connection Closed: " + str(e)) req.close()
finally: self.tuner.close()
req.close() # raise TunerError("806 - Tune Failed")
self.tuner.close()
# raise TunerError("806 - Tune Failed")
else:
self.fhdhr.logger.info("Detected stream URL is m3u8: %s" % self.stream_args["true_content_type"])
channelUri = self.stream_args["channelUri"]
while True:
videoUrlM3u = m3u8.load(channelUri)
if len(videoUrlM3u.playlists):
channelUri = videoUrlM3u.playlists[0].absolute_uri
else:
break
def generate():
try:
played_chunk_urls = []
while self.tuner.tuner_lock.locked():
playlist = m3u8.load(channelUri)
segments = playlist.segments
if len(played_chunk_urls):
newsegments = 0
for segment in segments:
if segment.absolute_uri not in played_chunk_urls:
newsegments += 1
self.fhdhr.logger.info("Refreshing m3u8, Loaded %s new segments." % str(newsegments))
else:
self.fhdhr.logger.info("Loaded %s segments." % str(len(segments)))
if playlist.keys != [None]:
keys = [{"url": key.absolute_uri, "method": key.method, "iv": key.iv} for key in playlist.keys if key]
else:
keys = [None for i in range(0, len(segments))]
for segment, key in zip(segments, keys):
chunkurl = segment.absolute_uri
if chunkurl and chunkurl not in played_chunk_urls:
played_chunk_urls.append(chunkurl)
if (not self.stream_args["duration"] == 0 and
not time.time() < self.stream_args["time_end"]):
self.fhdhr.logger.info("Requested Duration Expired.")
self.tuner.close()
chunk = self.fhdhr.web.session.get(chunkurl).content
if not chunk:
break
# raise TunerError("807 - No Video Data")
if key:
if key["url"]:
keyfile = self.fhdhr.web.session.get(key["url"]).content
cryptor = AES.new(keyfile, AES.MODE_CBC, keyfile)
chunk = cryptor.decrypt(chunk)
self.fhdhr.logger.info("Passing Through Chunk: %s" % chunkurl)
yield chunk
chunk_size = int(sys.getsizeof(chunk))
self.tuner.add_downloaded_size(chunk_size)
if playlist.target_duration:
time.sleep(int(playlist.target_duration))
self.fhdhr.logger.info("Connection Closed: Tuner Lock Removed")
except GeneratorExit:
self.fhdhr.logger.info("Connection Closed.")
except Exception as e:
self.fhdhr.logger.info("Connection Closed: " + str(e))
finally:
self.tuner.close()
# raise TunerError("806 - Tune Failed")
return generate() return generate()

25
fHDHR/device/tuners/tuner.py
View File

@ -53,12 +53,19 @@ class Tuner():
return stream.get() return stream.get()
def set_status(self, stream_args): def set_status(self, stream_args):
self.status = { if self.status["status"] != "Active":
"status": "Active", self.status = {
"method": stream_args["method"], "status": "Active",
"accessed": stream_args["accessed"], "clients": [],
"channel": stream_args["channel"], "clients_id": [],
"proxied_url": stream_args["channelUri"], "method": stream_args["method"],
"time_start": datetime.datetime.utcnow(), "accessed": [stream_args["accessed"]],
"downloaded": 0 "channel": stream_args["channel"],
} "proxied_url": stream_args["channelUri"],
"time_start": datetime.datetime.utcnow(),
"downloaded": 0
}
if stream_args["client"] not in self.status["clients"]:
self.status["clients"].append(stream_args["client"])
if stream_args["client_id"] not in self.status["clients_id"]:
self.status["clients_id"].append(stream_args["client_id"])

4
fHDHR/http/api/watch.py
View File

@ -1,5 +1,6 @@
from flask import Response, request, redirect, abort, stream_with_context from flask import Response, request, redirect, abort, stream_with_context
import urllib.parse import urllib.parse
import uuid
from fHDHR.exceptions import TunerError from fHDHR.exceptions import TunerError
@ -63,7 +64,8 @@ class Watch():
"duration": duration, "duration": duration,
"transcode": transcode, "transcode": transcode,
"accessed": accessed_url, "accessed": accessed_url,
"client": client_address "client": client_address,
"client_id": str(client_address) + "_" + str(uuid.uuid4())
} }
try: try: