Fix formatting errors
continuous-integration/drone Build is passing Details

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

View File

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