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

Merge pull request #6 from deathbybandaid/dev

Browse Source 
Check If Chunk Method can close better
This commit is contained in:
Deathbybandaid 2020-09-30 15:49:07 -04:00 committed by GitHub
commit 2b1c14bebc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
fakehdhr/__init__.py
View File

@ -263,12 +263,16 @@ class HDHR_HTTP_Server():
req = requests.get(channelUri, stream=True) req = requests.get(channelUri, stream=True)
def generate(): def generate():
yield '' try:
for chunk in req.iter_content(chunk_size=hdhr.config["direct_stream"]['chunksize']): yield ''
if not duration == 0 and not time.time() < duration: for chunk in req.iter_content(chunk_size=hdhr.config["direct_stream"]['chunksize']):
req.close() if not duration == 0 and not time.time() < duration:
break req.close()
yield chunk break
yield chunk
except GeneratorExit:
req.close()
print("Connection Closed.")
return Response(generate(), content_type=req.headers['content-type'], direct_passthrough=True) return Response(generate(), content_type=req.headers['content-type'], direct_passthrough=True)