From 1a313d4da5aeadc8186cad6be94aead7bca531af Mon Sep 17 00:00:00 2001
From: appleb_m <martin.appleby@psi.ch>
Date: Fri, 28 Nov 2025 16:50:37 +0100
Subject: [PATCH] Aaredb - add function to uplaod diffraction image

---
 daq/src/aaredaq/aaredb.py | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/daq/src/aaredaq/aaredb.py b/daq/src/aaredaq/aaredb.py
index 9ff01df2..c507301e 100644
--- a/daq/src/aaredaq/aaredb.py
+++ b/daq/src/aaredaq/aaredb.py
@@ -199,7 +199,19 @@ class AareWrapper:
         response = requests.post(url,
                                  files={'uploaded_file': (filename + ".jpg", jpeg_bytes, "image/jpeg")},
                                  verify=False, headers=headers)
-        print(f"Response status code: {response.status_code}")
+        logger.debug(f"Response status code: {response.status_code}")
+
+    def upload_jpg(self, sample_id: int, filename: str, jpg_image):
+        logger.debug(f"jppg_image of type: {type(jpg_image)}")
+        url = f"{self.__host}/protected_router/sample_runner/{sample_id}/upload-images"
+        headers = {
+            "accept": "application/json",
+            "X-Shared-Password": os.getenv("AAREDB_SHARED_PASSWORD")
+        }
+        response = requests.post(url,
+                                 files={'uploaded_file': (filename + ".jpg", jpg_image, "image/jpeg")},
+                                 verify=False, headers=headers)
+        logger.debug(f"Response status code: {response.status_code}")
 
     def create_rotation_run(self, s: Optional[SampleShortInfo], r:RotationScanRequest, d:DAQStatusModel):
         if s is None: