projects
/
python_utils.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
Persist the zookeeper client and connection. A step towards a config watcher.
[python_utils.git]
/
camera_utils.py
diff --git
a/camera_utils.py
b/camera_utils.py
index 799efd35d19fa39a7879c6c0851f9e697704927b..63ab7f813eeb16f76da2f4589469e1f8b414b189 100644
(file)
--- a/
camera_utils.py
+++ b/
camera_utils.py
@@
-1,12
+1,15
@@
#!/usr/bin/env python3
#!/usr/bin/env python3
+# © Copyright 2021-2022, Scott Gasch
+
"""Utilities for dealing with webcam images."""
import logging
import platform
import subprocess
"""Utilities for dealing with webcam images."""
import logging
import platform
import subprocess
-from typing import NamedTuple, Optional
import warnings
import warnings
+from dataclasses import dataclass
+from typing import Optional
import cv2 # type: ignore
import numpy as np
import cv2 # type: ignore
import numpy as np
@@
-14,23
+17,26
@@
import requests
import decorator_utils
import exceptions
import decorator_utils
import exceptions
+import scott_secrets
logger = logging.getLogger(__name__)
logger = logging.getLogger(__name__)
-class RawJpgHsv(NamedTuple):
+@dataclass
+class RawJpgHsv:
"""Raw image bytes, the jpeg image and the HSV (hue saturation value) image."""
"""Raw image bytes, the jpeg image and the HSV (hue saturation value) image."""
- raw: Optional[bytes]
- jpg: Optional[np.ndarray]
- hsv: Optional[np.ndarray]
+ raw: Optional[bytes]
= None
+ jpg: Optional[np.ndarray]
= None
+ hsv: Optional[np.ndarray]
= None
-class SanityCheckImageMetadata(NamedTuple):
+@dataclass
+class SanityCheckImageMetadata:
"""Is a Blue Iris image bad (big grey borders around it) or infrared?"""
"""Is a Blue Iris image bad (big grey borders around it) or infrared?"""
- is_
bad_image: bool
- is_
infrared_image: bool
+ is_
infrared_image: bool = False
+ is_
bad_image: bool = False
def sanity_check_image(hsv: np.ndarray) -> SanityCheckImageMetadata:
def sanity_check_image(hsv: np.ndarray) -> SanityCheckImageMetadata:
@@
-46,17
+52,14
@@
def sanity_check_image(hsv: np.ndarray) -> SanityCheckImageMetadata:
for r in range(rows):
for c in range(cols):
pixel = hsv[(r, c)]
for r in range(rows):
for c in range(cols):
pixel = hsv[(r, c)]
- if (
- is_near(pixel[0], 16)
- and is_near(pixel[1], 117)
- and is_near(pixel[2], 196)
- ):
+ if is_near(pixel[0], 16) and is_near(pixel[1], 117) and is_near(pixel[2], 196):
weird_orange_count += 1
elif is_near(pixel[0], 0) and is_near(pixel[1], 0):
hs_zero_count += 1
weird_orange_count += 1
elif is_near(pixel[0], 0) and is_near(pixel[1], 0):
hs_zero_count += 1
- logger.debug(
f"hszero#={hs_zero_count}, weird_orange={weird_orange_count}"
)
+ logger.debug(
"hszero#=%d, weird_orange=%d", hs_zero_count, weird_orange_count
)
return SanityCheckImageMetadata(
return SanityCheckImageMetadata(
- hs_zero_count > (num_pixels * 0.75), weird_orange_count > (num_pixels * 0.75)
+ hs_zero_count > (num_pixels * 0.75),
+ weird_orange_count > (num_pixels * 0.75),
)
)
@@
-65,29
+68,39
@@
def fetch_camera_image_from_video_server(
camera_name: str, *, width: int = 256, quality: int = 70
) -> Optional[bytes]:
"""Fetch the raw webcam image from the video server."""
camera_name: str, *, width: int = 256, quality: int = 70
) -> Optional[bytes]:
"""Fetch the raw webcam image from the video server."""
+
camera_name = camera_name.replace(".house", "")
camera_name = camera_name.replace(".cabin", "")
camera_name = camera_name.replace(".house", "")
camera_name = camera_name.replace(".cabin", "")
- url = f"http://10.0.0.226:8080/
Umtxxf1uKMBniFblqeQ9KRbb6DDzN4/jpeg/GKlT2FfiSQ
/{camera_name}/s.jpg"
- logger.debug(
f'Fetching image from {url}'
)
+ url = f"http://10.0.0.226:8080/
{scott_secrets.SHINOBI_KEY1}/jpeg/{scott_secrets.SHINOBI_KEY2}
/{camera_name}/s.jpg"
+ logger.debug(
'Fetching image from %s', url
)
try:
response = requests.get(url, stream=False, timeout=10.0)
if response.ok:
raw = response.content
try:
response = requests.get(url, stream=False, timeout=10.0)
if response.ok:
raw = response.content
- logger.debug(
f'Read {len(response.content)} byte image from HTTP server'
)
+ logger.debug(
'Read %d byte image from HTTP server', len(response.content)
)
tmp = np.frombuffer(raw, dtype="uint8")
logger.debug(
tmp = np.frombuffer(raw, dtype="uint8")
logger.debug(
- f'Translated raw content into {tmp.shape} {type(tmp)} with element type {type(tmp[0])}.'
+ 'Translated raw content into %s %s with element type %s',
+ tmp.shape,
+ type(tmp),
+ type(tmp[0]),
)
jpg = cv2.imdecode(tmp, cv2.IMREAD_COLOR)
logger.debug(
)
jpg = cv2.imdecode(tmp, cv2.IMREAD_COLOR)
logger.debug(
- f'Decoded into {jpg.shape} jpeg {type(jpg)} with element type {type(jpg[0][0])}'
+ 'Decoded into %s jpeg %s with element type %s',
+ jpg.shape,
+ type(jpg),
+ type(jpg[0][0]),
)
hsv = cv2.cvtColor(jpg, cv2.COLOR_BGR2HSV)
logger.debug(
)
hsv = cv2.cvtColor(jpg, cv2.COLOR_BGR2HSV)
logger.debug(
- f'Converted JPG into HSV {hsv.shape} HSV {type(hsv)} with element type {type(hsv[0][0])}'
+ 'Converted JPG into %s HSV HSV %s with element type %s',
+ hsv.shape,
+ type(hsv),
+ type(hsv[0][0]),
)
)
-
(_, is_bad_image)
= sanity_check_image(hsv)
- if not is_bad_image:
+
ret
= sanity_check_image(hsv)
+ if not
ret.
is_bad_image:
return raw
except Exception as e:
logger.exception(e)
return raw
except Exception as e:
logger.exception(e)
@@
-105,8
+118,8
@@
def camera_name_to_hostname(camera_name: str) -> str:
>>> camera_name_to_hostname('cabin_driveway')
'driveway.cabin'
>>> camera_name_to_hostname('cabin_driveway')
'driveway.cabin'
-
"""
"""
+
mapping = {
"driveway": "driveway.house",
"backyard": "backyard.house",
mapping = {
"driveway": "driveway.house",
"backyard": "backyard.house",
@@
-122,13
+135,12
@@
def camera_name_to_hostname(camera_name: str) -> str:
@decorator_utils.retry_if_none(tries=2, delay_sec=1, backoff=1.1)
@decorator_utils.retry_if_none(tries=2, delay_sec=1, backoff=1.1)
-def fetch_camera_image_from_rtsp_stream(
- camera_name: str, *, width: int = 256
-) -> Optional[bytes]:
+def fetch_camera_image_from_rtsp_stream(camera_name: str, *, width: int = 256) -> Optional[bytes]:
"""Fetch the raw webcam image straight from the webcam's RTSP stream."""
"""Fetch the raw webcam image straight from the webcam's RTSP stream."""
+
hostname = camera_name_to_hostname(camera_name)
hostname = camera_name_to_hostname(camera_name)
- stream = f"rtsp://camera:
IaLaIok
@{hostname}:554/live"
- logger.debug(
f'Fetching image from RTSP stream {stream}'
)
+ stream = f"rtsp://camera:
{scott_secrets.CAMERA_PASSWORD}
@{hostname}:554/live"
+ logger.debug(
'Fetching image from RTSP stream %s', stream
)
try:
cmd = [
"/usr/bin/timeout",
try:
cmd = [
"/usr/bin/timeout",
@@
-146,9
+158,7
@@
def fetch_camera_image_from_rtsp_stream(
f"scale={width}:-1",
"-",
]
f"scale={width}:-1",
"-",
]
- with subprocess.Popen(
- cmd, stdout=subprocess.PIPE, stderr=subprocess.DEVNULL
- ) as proc:
+ with subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.DEVNULL) as proc:
out, _ = proc.communicate(timeout=10)
return out
except Exception as e:
out, _ = proc.communicate(timeout=10)
return out
except Exception as e:
@@
-160,14
+170,13
@@
def fetch_camera_image_from_rtsp_stream(
@decorator_utils.timeout(seconds=30, use_signals=False)
@decorator_utils.timeout(seconds=30, use_signals=False)
-def _fetch_camera_image(
- camera_name: str, *, width: int = 256, quality: int = 70
-) -> RawJpgHsv:
+def _fetch_camera_image(camera_name: str, *, width: int = 256, quality: int = 70) -> RawJpgHsv:
"""Fetch a webcam image given the camera name."""
"""Fetch a webcam image given the camera name."""
+
logger.debug("Trying to fetch camera image from video server")
logger.debug("Trying to fetch camera image from video server")
- raw = fetch_camera_image_from_video_server(
- camera_name
, width=width, quality=quality
- )
+ if camera_name == 'frontdoor':
+ camera_name
= 'front_door'
+
raw = fetch_camera_image_from_video_server(camera_name, width=width, quality=quality
)
if raw is None:
logger.debug("Reading from video server failed; trying direct RTSP stream")
raw = fetch_camera_image_from_rtsp_stream(camera_name, width=width)
if raw is None:
logger.debug("Reading from video server failed; trying direct RTSP stream")
raw = fetch_camera_image_from_rtsp_stream(camera_name, width=width)
@@
-186,9
+195,9
@@
def _fetch_camera_image(
return RawJpgHsv(None, None, None)
return RawJpgHsv(None, None, None)
-def fetch_camera_image(
- camera_name: str, *, width: int = 256, quality: int = 70
-) -> RawJpgHsv:
+def fetch_camera_image(
camera_name: str, *, width: int = 256, quality: int = 70) -> RawJpgHsv:
+ """Fetch an image given the camera_name."""
+
try:
return _fetch_camera_image(camera_name, width=width, quality=quality)
except exceptions.TimeoutError:
try:
return _fetch_camera_image(camera_name, width=width, quality=quality)
except exceptions.TimeoutError: