AareDB: added comments/messages for upload_image method

This commit is contained in:
appleb_m committed 2026-03-09 17:14:39 +01:00
1 parent 6bda961231
commit 989624b319
1 file changed
+11 -4
+11 -4
View File
@@ -185,7 +185,7 @@ class AareWrapper:
except Exception as e:
logger.error(f"Error sending message to db: {e}")
def upload_image(self, sample_id: int, filename: str, bgr_image: np.ndarray):
def upload_image(self, sample_id: int, filename: str, bgr_image: np.ndarray, message: str | None = None):
_, buffer = cv2.imencode('.jpg', bgr_image)
jpeg_bytes = io.BytesIO(buffer)
url = f"{self.__host}/protected_router/sample_runner/{sample_id}/upload-images"
@@ -193,9 +193,16 @@ class AareWrapper:
"accept": "application/json",
"X-Shared-Password": os.getenv("AAREDB_SHARED_PASSWORD")
}
response = requests.post(url,
files={'uploaded_file': (filename + ".jpg", jpeg_bytes, "image/jpeg")},
verify=False, headers=headers)
request_kwargs = {
"files": {'uploaded_file': (filename + ".jpg", jpeg_bytes, "image/jpeg")},
"verify": False,
"headers": headers,
}
if message is not None:
request_kwargs["data"] = {"comment": message}
response = requests.post(url, **request_kwargs)
logger.debug(f"Response status code: {response.status_code}")
def upload_jpg(self, sample_id: int, filename: str, jpg_image):