fix: improve img validation to handle numpy arrays safely

Author: codematrixer

.github/workflows/release.yml

@@ -23,7 +23,7 @@ jobs:
  run: |
  python -m pip install --upgrade pip
  pip install setuptools wheel twine poetry
- poetry lock --no-update
+ poetry lock
  poetry install
 
  - name: Get the version from the tag

hmdriver2/_client.py

@@ -17,8 +17,6 @@
 
 UITEST_SERVICE_PORT = 8012
 SOCKET_TIMEOUT = 20
-ASSETS_PATH = os.path.join(os.path.dirname(__file__), "assets")
-AGENT_CLEAR_PATH = ["app", "commons-", "agent", "libagent_antry"]
 
 
 class HmClient:

hmdriver2/_screenrecord.py

@@ -1,9 +1,9 @@
 # -*- coding: utf-8 -*-
-import queue
+
 import typing
 import threading
 import numpy as np
-from queue import Queue
+import queue
 from datetime import datetime
 
 import cv2
@@ -20,7 +20,7 @@ def __init__(self, serial: str, d: Driver):
  self.d = d
 
  self.video_path = None
- self.jpeg_queue = Queue()
+ self.jpeg_queue = queue.Queue()
  self.threads: typing.List[threading.Thread] = []
  self.stop_event = threading.Event()
 
@@ -102,7 +102,7 @@ def _video_writer(self):
  img = cv2.imdecode(np.frombuffer(jpeg_image, np.uint8), cv2.IMREAD_COLOR)
  except queue.Empty:
  pass
- if not img:
+ if img is None or img.size == 0:
  continue
  if cv2_instance is None:
  height, width = img.shape[:2]