k-space/camera-motion-detect
10
0
Fork 0
Code Issues 2 Pull Requests Projects Activity

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

Browse Source
This commit is contained in:
Lauri Võsandi 2022-08-03 08:02:53 +03:00
parent 05b5bc30f2
commit eebd2f3d96
2 changed files with 22 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
.flake8
View File

@ -1,3 +0,0 @@
[flake8]
inline-quotes = "
indent-size = 4

47
camdetect.py
View File

@ -116,7 +116,6 @@ gauge_queue_frames.labels("hold").set(0)
gauge_queue_frames.labels("upload").set(0) gauge_queue_frames.labels("upload").set(0)
counter_frames.labels("motion").inc(0) counter_frames.labels("motion").inc(0)
assert SLIDE_WINDOW <= 8 # This is 256 frames which should be enough assert SLIDE_WINDOW <= 8 # This is 256 frames which should be enough
@ -260,22 +259,20 @@ async def motion_detector(reference_frame, download_queue, upload_queue):
# Handle event start # Handle event start
if motion_detected and not event_id: if motion_detected and not event_id:
result = await app.ctx.coll.find_one_and_update({ result = await app.ctx.coll.find_one_and_update({
"timestamp": dt, # TODO: Account for clock skew "timestamp": dt, # TODO: Account for clock skew
"source": SOURCE_NAME,
}, {
"$setOnInsert": {
"timestamp": dt,
"source": SOURCE_NAME, "source": SOURCE_NAME,
}, { "event": "motion-detected",
"$setOnInsert": { "started": dt,
"timestamp": dt, "finished": dt + timedelta(minutes=2),
"source": SOURCE_NAME, "component": "camdetect",
"event": "motion-detected", "screenshots": [],
"started": dt, "action": "event",
"finished": dt + timedelta(minutes=2), }
"component": "camdetect", }, upsert=True, return_document=ReturnDocument.AFTER)
"screenshots": [],
"action": "event",
}
},
upsert = True,
return_document=ReturnDocument.AFTER)
app.ctx.event_id = event_id = result["_id"] app.ctx.event_id = event_id = result["_id"]
# Handle buffering frames prior event start # Handle buffering frames prior event start
@ -375,7 +372,7 @@ async def download(resp, queue):
continue continue
else: else:
# Assemble JPEG blob # Assemble JPEG blob
blob = buf + data[:marker+2] blob = buf + data[:marker + 2]
# Parse DCT coefficients # Parse DCT coefficients
try: try:
@ -394,7 +391,7 @@ async def download(resp, queue):
except asyncio.QueueFull: except asyncio.QueueFull:
counter_dropped_frames.labels("download").inc() counter_dropped_frames.labels("download").inc()
gauge_queue_frames.labels("download").set(queue.qsize()) gauge_queue_frames.labels("download").set(queue.qsize())
data = data[marker+2:] data = data[marker + 2:]
buf = b"" buf = b""
counter_receive_frames.inc() counter_receive_frames.inc()
@ -446,8 +443,8 @@ async def bypass_stream_wrapper(request):
counter_transmit_bytes.inc(len(data)) counter_transmit_bytes.inc(len(data))
counter_transmit_frames.inc() counter_transmit_frames.inc()
return response.stream( return response.stream(
stream_camera, stream_camera,
content_type="multipart/x-mixed-replace; boundary=frame") content_type="multipart/x-mixed-replace; boundary=frame")
@app.route("/debug") @app.route("/debug")
@ -467,8 +464,8 @@ async def stream_wrapper(request):
for y in range(0, app.ctx.mask.shape[0]): for y in range(0, app.ctx.mask.shape[0]):
for x in range(0, app.ctx.mask.shape[1]): for x in range(0, app.ctx.mask.shape[1]):
if app.ctx.mask[y][x] > 0: if app.ctx.mask[y][x] > 0:
img[y*DCT_BLOCK_SIZE:(y+1)*DCT_BLOCK_SIZE, img[y * DCT_BLOCK_SIZE:(y + 1) * DCT_BLOCK_SIZE,
x*DCT_BLOCK_SIZE:(x+1)*DCT_BLOCK_SIZE, x * DCT_BLOCK_SIZE:(x + 1) * DCT_BLOCK_SIZE,
channel] = 255 channel] = 255
# Compress modified frame as JPEG frame # Compress modified frame as JPEG frame
@ -490,9 +487,9 @@ async def ready_check(request):
# Check if S3 is accessible # Check if S3 is accessible
session = aioboto3.Session() session = aioboto3.Session()
async with session.resource("s3", async with session.resource("s3",
aws_access_key_id=AWS_ACCESS_KEY_ID, aws_access_key_id=AWS_ACCESS_KEY_ID,
aws_secret_access_key=AWS_SECRET_ACCESS_KEY, aws_secret_access_key=AWS_SECRET_ACCESS_KEY,
endpoint_url=S3_ENDPOINT_URL) as s3: endpoint_url=S3_ENDPOINT_URL) as s3:
bucket = await s3.Bucket(S3_BUCKET_NAME) bucket = await s3.Bucket(S3_BUCKET_NAME)
await bucket.upload_fileobj(io.BytesIO(b"test"), "test") await bucket.upload_fileobj(io.BytesIO(b"test"), "test")