Created
January 7, 2022 22:56
-
-
Save baseplate-admin/f82ae740ae5b33477d817df8c1663466 to your computer and use it in GitHub Desktop.
Stream Videos with django
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import os | |
| import re | |
| import mimetypes | |
| from wsgiref.util import FileWrapper | |
| from django.http.response import StreamingHttpResponse | |
| range_re = re.compile(r'bytes\s*=\s*(\d+)\s*-\s*(\d*)', re.I) | |
| class RangeFileWrapper(object): | |
| def __init__(self, filelike, blksize=8192, offset=0, length=None): | |
| self.filelike = filelike | |
| self.filelike.seek(offset, os.SEEK_SET) | |
| self.remaining = length | |
| self.blksize = blksize | |
| def close(self): | |
| if hasattr(self.filelike, 'close'): | |
| self.filelike.close() | |
| def __iter__(self): | |
| return self | |
| def __next__(self): | |
| if self.remaining is None: | |
| # If remaining is None, we're reading the entire file. | |
| data = self.filelike.read(self.blksize) | |
| if data: | |
| return data | |
| raise StopIteration() | |
| else: | |
| if self.remaining <= 0: | |
| raise StopIteration() | |
| data = self.filelike.read(min(self.remaining, self.blksize)) | |
| if not data: | |
| raise StopIteration() | |
| self.remaining -= len(data) | |
| return data | |
| def stream_video(request, path): | |
| range_header = request.META.get('HTTP_RANGE', '').strip() | |
| range_match = range_re.match(range_header) | |
| size = os.path.getsize(path) | |
| content_type, encoding = mimetypes.guess_type(path) | |
| content_type = content_type or 'application/octet-stream' | |
| if range_match: | |
| first_byte, last_byte = range_match.groups() | |
| first_byte = int(first_byte) if first_byte else 0 | |
| last_byte = int(last_byte) if last_byte else size - 1 | |
| if last_byte >= size: | |
| last_byte = size - 1 | |
| length = last_byte - first_byte + 1 | |
| resp = StreamingHttpResponse(RangeFileWrapper(open(path, 'rb'), offset=first_byte, length=length), status=206, content_type=content_type) | |
| resp['Content-Length'] = str(length) | |
| resp['Content-Range'] = 'bytes %s-%s/%s' % (first_byte, last_byte, size) | |
| else: | |
| resp = StreamingHttpResponse(FileWrapper(open(path, 'rb')), content_type=content_type) | |
| resp['Content-Length'] = str(size) | |
| resp['Accept-Ranges'] = 'bytes' | |
| return resp |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment