110 lines
3.2 KiB
Python
110 lines
3.2 KiB
Python
import os
|
|
import shutil
|
|
import tempfile
|
|
import time
|
|
|
|
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
|
|
|
|
camera = Blueprint('camera', __name__, template_folder=template_folder)
|
|
|
|
# Declare routes list
|
|
__routes__ = [
|
|
camera,
|
|
]
|
|
|
|
|
|
def get_device_id(device_id=None):
|
|
if device_id is None:
|
|
device_id = str(send_request(action='camera.get_default_device_id').output)
|
|
return device_id
|
|
|
|
|
|
def get_frame_file(device_id=None):
|
|
device_id = get_device_id(device_id)
|
|
was_recording = True
|
|
frame_file = None
|
|
status = send_request(action='camera.status', device_id=device_id).output
|
|
|
|
if device_id not in status:
|
|
was_recording = False
|
|
send_request(action='camera.start_recording',
|
|
device_id=device_id)
|
|
|
|
while not frame_file:
|
|
frame_file = send_request(action='camera.status', device_id=device_id). \
|
|
output.get(device_id, {}).get('image_file')
|
|
|
|
if not frame_file:
|
|
time.sleep(0.1)
|
|
|
|
if not was_recording:
|
|
with tempfile.NamedTemporaryFile(prefix='camera_capture_', suffix='.jpg',
|
|
delete=False) as f:
|
|
# stop_recording will delete the temporary frames. Copy the image file
|
|
# to a temporary file before stopping recording
|
|
tmp_file = f.name
|
|
|
|
shutil.copyfile(frame_file, tmp_file)
|
|
frame_file = tmp_file
|
|
send_request(action='camera.stop_recording', device_id=device_id)
|
|
|
|
return frame_file
|
|
|
|
|
|
def video_feed(device_id=None):
|
|
device_id = get_device_id(device_id)
|
|
send_request(action='camera.start_recording', device_id=device_id)
|
|
last_frame_file = None
|
|
|
|
try:
|
|
while True:
|
|
frame_file = get_frame_file(device_id)
|
|
if frame_file == last_frame_file:
|
|
continue
|
|
|
|
with open(frame_file, 'rb') as f:
|
|
frame = f.read()
|
|
|
|
yield (b'--frame\r\n'
|
|
b'Content-Type: image/jpeg\r\n\r\n' + frame + b'\r\n')
|
|
|
|
last_frame_file = frame_file
|
|
finally:
|
|
send_request(action='camera.stop_recording', device_id=device_id)
|
|
|
|
|
|
@camera.route('/camera/<device_id>/frame', methods=['GET'])
|
|
@authenticate()
|
|
def get_camera_frame(device_id):
|
|
frame_file = get_frame_file(device_id)
|
|
return send_from_directory(os.path.dirname(frame_file),
|
|
os.path.basename(frame_file))
|
|
|
|
|
|
@camera.route('/camera/frame', methods=['GET'])
|
|
@authenticate()
|
|
def get_default_camera_frame():
|
|
frame_file = get_frame_file()
|
|
return send_from_directory(os.path.dirname(frame_file),
|
|
os.path.basename(frame_file))
|
|
|
|
|
|
@camera.route('/camera/stream', methods=['GET'])
|
|
@authenticate()
|
|
def get_default_stream_feed():
|
|
return Response(video_feed(),
|
|
mimetype='multipart/x-mixed-replace; boundary=frame')
|
|
|
|
|
|
@camera.route('/camera/<device_id>/stream', methods=['GET'])
|
|
@authenticate()
|
|
def get_stream_feed(device_id):
|
|
return Response(video_feed(device_id),
|
|
mimetype='multipart/x-mixed-replace; boundary=frame')
|
|
|
|
|
|
# vim:sw=4:ts=4:et:
|