Add example for movie streaming range request support

timothycrosley · timothycrosley · commit bc2212c8fd66 · 2015-11-10T16:56:24.000-08:00
diff --git a/examples/streaming_movie_server/movie.mp4 b/examples/streaming_movie_server/movie.mp4
diff --git a/examples/streaming_movie_server/movie_server.py b/examples/streaming_movie_server/movie_server.py
@@ -0,0 +1,8 @@
+'''A simple streaming movie server example'''
+import hug
+
+
+@hug.get(output=hug.output_format.mp4_video)
+def watch():
+    '''Watch an example movie, streamed directly to you from hug'''
+    return 'movie.mp4'
diff --git a/hug/decorators.py b/hug/decorators.py
@@ -365,12 +365,26 @@ def interface(request, response, api_version=None, **kwargs):
             else:
                 to_return = transform(to_return)
 
-        to_return = (function_output(to_return, **function_output_kwargs) if
-                     not hasattr(to_return, 'read') else to_return)
+        to_return = function_output(to_return, **function_output_kwargs)
         if hasattr(to_return, 'read'):
-            response.stream = to_return
+            size = None
             if hasattr(to_return, 'name') and os.path.isfile(to_return.name):
-                response.stream_len = os.path.getsize(to_return.name)
+                size = os.path.getsize(to_return.name)
+            if request.range and size:
+                start, end = request.range
+                if end < 0:
+                    end = size + end
+                end = min(end, size)
+                length = end - start + 1
+                to_return.seek(start)
+                response.data = to_return.read(length)
+                response.status = falcon.HTTP_206
+                response.content_range = (start, end, size)
+                to_return.close()
+            else:
+                response.stream = to_return
+                if size:
+                    response.stream_len = size
         else:
             response.data = to_return
 
