AareDB: added comments/messages for upload_image method
This commit is contained in:
1 parent
6bda961231
commit
989624b319
1 file changed
+11
-4
+11
-4
@@ -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):
|
||||
|
||||
Reference in new issue
Block a user