evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Improve DOODS folder handling and add process time attribute (#40344)

Browse source 
* Updates to DOODS

* Fix import order
This commit is contained in:
lamiskin 2020-09-23 12:43:21 +10:00 • committed by GitHub
parent 72a7f69a08
commit 5ebce075a1
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

8
homeassistant/components/doods/image_processing.py
View file

@ -1,6 +1,7 @@
"""Support for the DOODS service.""" """Support for the DOODS service."""
import io import io
import logging import logging
import os
import time import time
from PIL import Image, ImageDraw, UnidentifiedImageError from PIL import Image, ImageDraw, UnidentifiedImageError
@ -26,6 +27,7 @@ _LOGGER = logging.getLogger(__name__)
ATTR_MATCHES = "matches" ATTR_MATCHES = "matches"
ATTR_SUMMARY = "summary" ATTR_SUMMARY = "summary"
ATTR_TOTAL_MATCHES = "total_matches" ATTR_TOTAL_MATCHES = "total_matches"
ATTR_PROCESS_TIME = "process_time"
CONF_URL = "url" CONF_URL = "url"
CONF_AUTH_KEY = "auth_key" CONF_AUTH_KEY = "auth_key"
@ -203,6 +205,7 @@ class Doods(ImageProcessingEntity):
self._matches = {} self._matches = {}
self._total_matches = 0 self._total_matches = 0
self._last_image = None self._last_image = None
self._process_time = 0
@property @property
def camera_entity(self): def camera_entity(self):
@ -228,6 +231,7 @@ class Doods(ImageProcessingEntity):
label: len(values) for label, values in self._matches.items() label: len(values) for label, values in self._matches.items()
}, },
ATTR_TOTAL_MATCHES: self._total_matches, ATTR_TOTAL_MATCHES: self._total_matches,
ATTR_PROCESS_TIME: self._process_time,
} }
def _save_image(self, image, matches, paths): def _save_image(self, image, matches, paths):
@ -270,6 +274,8 @@ class Doods(ImageProcessingEntity):
for path in paths: for path in paths:
_LOGGER.info("Saving results image to %s", path) _LOGGER.info("Saving results image to %s", path)
if not os.path.exists(os.path.dirname(path)):
os.makedirs(os.path.dirname(path), exist_ok=True)
img.save(path) img.save(path)
def process_image(self, image): def process_image(self, image):
@ -308,6 +314,7 @@ class Doods(ImageProcessingEntity):
_LOGGER.error(response["error"]) _LOGGER.error(response["error"])
self._matches = matches self._matches = matches
self._total_matches = total_matches self._total_matches = total_matches
self._process_time = time.monotonic() - start
return return
for detection in response["detections"]: for detection in response["detections"]:
@ -380,3 +387,4 @@ class Doods(ImageProcessingEntity):
self._matches = matches self._matches = matches
self._total_matches = total_matches self._total_matches = total_matches
self._process_time = time.monotonic() - start