forked from platypush/platypush
56 lines
1.5 KiB
Python
56 lines
1.5 KiB
Python
import os
|
|
import tempfile
|
|
|
|
from flask import Response, Blueprint, send_from_directory
|
|
|
|
from platypush.backend.http.app import template_folder
|
|
from platypush.backend.http.app.utils import authenticate, send_request
|
|
from platypush.config import Config
|
|
from platypush.plugins.camera.pi import CameraPiPlugin
|
|
|
|
camera_pi = Blueprint('camera-pi', __name__, template_folder=template_folder)
|
|
|
|
# Declare routes list
|
|
__routes__ = [
|
|
camera_pi,
|
|
]
|
|
|
|
|
|
def video_feed():
|
|
camera_conf = Config.get('camera.pi') or {}
|
|
camera = CameraPiPlugin(**camera_conf)
|
|
|
|
with camera:
|
|
while True:
|
|
output = camera.get_stream()
|
|
|
|
with output.ready:
|
|
output.ready.wait()
|
|
frame = output.frame
|
|
|
|
if frame and len(frame):
|
|
yield (b'--frame\r\n'
|
|
b'Content-Type: image/jpeg\r\n\r\n' + frame + b'\r\n')
|
|
|
|
|
|
@camera_pi.route('/camera/pi/frame', methods=['GET'])
|
|
@authenticate()
|
|
def get_frame_img():
|
|
filename = os.path.join(tempfile.gettempdir(), 'camera_pi.jpg')
|
|
response = send_request('camera.pi.take_picture', image_file=filename)
|
|
frame_file = (response.output or {}).get('image_file')
|
|
assert frame_file is not None
|
|
|
|
return send_from_directory(os.path.dirname(frame_file),
|
|
os.path.basename(frame_file))
|
|
|
|
|
|
@camera_pi.route('/camera/pi/stream', methods=['GET'])
|
|
@authenticate()
|
|
def get_stream_feed():
|
|
return Response(video_feed(),
|
|
mimetype='multipart/x-mixed-replace; boundary=frame')
|
|
|
|
|
|
# vim:sw=4:ts=4:et:
|