57 lines
1.5 KiB
Python
57 lines
1.5 KiB
Python
import logging
|
|
import socket
|
|
import time
|
|
import picamera
|
|
|
|
from platypush.backend import Backend
|
|
|
|
class CameraPiBackend(Backend):
|
|
def __init__(self, listen_port, x_resolution=640, y_resolution=480,
|
|
framerate=24, hflip=False, vflip=False, **kwargs):
|
|
super().__init__(**kwargs)
|
|
|
|
self.listen_port = listen_port
|
|
self.x_resolution = x_resolution
|
|
self.y_resolution = y_resolution
|
|
self.framerate = framerate
|
|
self.hflip = hflip
|
|
self.vflip = vflip
|
|
|
|
self.server_socket = socket.socket()
|
|
self.server_socket.bind(('0.0.0.0', self.listen_port))
|
|
self.server_socket.listen(0)
|
|
|
|
self.camera = picamera.PiCamera()
|
|
self.camera.resolution = (self.x_resolution, self.y_resolution)
|
|
self.camera.framerate = framerate
|
|
self.camera.hflip = self.hflip
|
|
self.camera.vflip = self.vflip
|
|
|
|
logging.info('Initialized Pi camera backend')
|
|
|
|
def send_message(self, msg):
|
|
pass
|
|
|
|
def run(self):
|
|
super().run()
|
|
|
|
while True:
|
|
connection = self.server_socket.accept()[0].makefile('wb')
|
|
|
|
try:
|
|
self.camera.start_recording(connection, format='h264')
|
|
while True:
|
|
self.camera.wait_recording(60)
|
|
except ConnectionError as e:
|
|
pass
|
|
finally:
|
|
try:
|
|
self.camera.stop_recording()
|
|
connection.close()
|
|
except:
|
|
pass
|
|
|
|
|
|
# vim:sw=4:ts=4:et:
|
|
|