k-space/camera-tiler
10
0
Fork 0
Code Issues Pull Requests Projects Activity

Fix formatting errors
All checks were successful
continuous-integration/drone Build is passing
Details

Browse Source
This commit is contained in:
Lauri Võsandi 2022-12-02 13:00:20 +02:00
parent ef7b0f7b7c
commit 53ab49c661
1 changed files with 20 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
camtiler.py
View File

@ -3,17 +3,13 @@
import aiohttp import aiohttp
import asyncio import asyncio
import cv2 import cv2
import io
import numpy as np import numpy as np
import os import os
import socket
import sys import sys
from datetime import datetime
from kubernetes import client, config from kubernetes import client, config
from math import ceil from math import ceil
from prometheus_client import Counter, Gauge from prometheus_client import Counter, Gauge
from sanic import Sanic, response from sanic import Sanic, response
from sanic.response import stream
from sanic_prometheus import monitor from sanic_prometheus import monitor
SERVICE_NAMESPACE = os.getenv("SERVICE_NAMESPACE", "camtiler") SERVICE_NAMESPACE = os.getenv("SERVICE_NAMESPACE", "camtiler")
@ -70,8 +66,7 @@ gauge_build_info.labels(
app = Sanic("camtiler") app = Sanic("camtiler")
STREAM_RESPONSE = \ STREAM_RESPONSE = b"""
b"""
--frame --frame
Content-Type: image/jpeg Content-Type: image/jpeg
@ -83,6 +78,7 @@ MERGED_SUBSAMPLE=3
BLANK = np.full((720, 1280, 3), 0) BLANK = np.full((720, 1280, 3), 0)
GREEN = np.full((720, 1280, 3), (0, 135, 0)) GREEN = np.full((720, 1280, 3), (0, 135, 0))
def tilera(iterable): def tilera(iterable):
hstacking = ceil(len(iterable) ** 0.5) hstacking = ceil(len(iterable) ** 0.5)
vstacking = ceil(len(iterable) / hstacking) vstacking = ceil(len(iterable) / hstacking)
@ -131,7 +127,7 @@ async def client_connect(name, resp):
if buf: if buf:
# If we already have something in buffer, seek end in new data # If we already have something in buffer, seek end in new data
marker = data.find(b'\xff\xd9') marker = data.find(b"\xff\xd9")
if marker < 0: if marker < 0:
# If no end marker was found add it to buffer # If no end marker was found add it to buffer
buf += data buf += data
@ -146,7 +142,7 @@ async def client_connect(name, resp):
counter_rx_frames.inc() counter_rx_frames.inc()
# Seek begin in newly received data # Seek begin in newly received data
marker = data.find(b'\xff\xd8') marker = data.find(b"\xff\xd8")
if marker >= 0: if marker >= 0:
data, buf = data[:marker], data[marker:] data, buf = data[:marker], data[marker:]
@ -156,12 +152,12 @@ async def client_connect(name, resp):
async def client(name, url): async def client(name, url):
print("Opening upstream connection to %s" % url) print("Opening upstream connection to %s" % url)
kwargs = dict( kwargs = {
headers = { "headers": {
"User-Agent": "camtiler/%s" % GIT_COMMIT_TIMESTAMP "User-Agent": "camtiler/%s" % GIT_COMMIT_TIMESTAMP
}, },
skip_auto_headers = True, "timeout": aiohttp.ClientTimeout(connect=5, sock_read=2)
timeout = aiohttp.ClientTimeout(connect=5, sock_read=2)) }
while True: while True:
app.ctx.frames[name] = None app.ctx.frames[name] = None
async with aiohttp.ClientSession(**kwargs) as session: async with aiohttp.ClientSession(**kwargs) as session:
@ -189,10 +185,11 @@ async def stream_wrapper(request):
return response.stream( return response.stream(
stream_tiled, stream_tiled,
content_type='multipart/x-mixed-replace; boundary=frame' content_type="multipart/x-mixed-replace; boundary=frame"
) )
@app.listener('before_server_start')
@app.listener("before_server_start")
async def setup_db(app, loop): async def setup_db(app, loop):
app.ctx.event_frame = asyncio.Event() app.ctx.event_frame = asyncio.Event()
app.ctx.frames = {} app.ctx.frames = {}
@ -202,7 +199,7 @@ async def setup_db(app, loop):
app.ctx.motion_end = None app.ctx.motion_end = None
for name, url in targets: for name, url in targets:
app.ctx.frames[name] = None app.ctx.frames[name] = None
task = asyncio.create_task(client(name, url)) asyncio.create_task(client(name, url))
monitor(app).expose_endpoint() monitor(app).expose_endpoint()