From 2bd1465104d7e458ea0f24980f424873810b2060 Mon Sep 17 00:00:00 2001 From: Brandon Rose Date: Mon, 15 Jul 2024 12:27:37 -0500 Subject: [PATCH] patches for dataset saving --- src/askem_beaker/contexts/dataset/context.py | 5 +++-- .../dataset/procedures/python3/hmi_create_csv_dataset.py | 2 +- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/src/askem_beaker/contexts/dataset/context.py b/src/askem_beaker/contexts/dataset/context.py index 401636c..fe34a62 100644 --- a/src/askem_beaker/contexts/dataset/context.py +++ b/src/askem_beaker/contexts/dataset/context.py @@ -248,11 +248,12 @@ async def save_dataset_request(self, message): new_dataset["id"] = new_dataset_id code = self.get_code( - "df_create_csv_dataset", + "hmi_create_csv_dataset", { "var_name": var_name, "id": new_dataset_id, - "auth": self.get_auth(), + "username": self.auth.username, + "password": self.auth.password, "filename": filename } ) diff --git a/src/askem_beaker/contexts/dataset/procedures/python3/hmi_create_csv_dataset.py b/src/askem_beaker/contexts/dataset/procedures/python3/hmi_create_csv_dataset.py index 5014000..fb6b66e 100644 --- a/src/askem_beaker/contexts/dataset/procedures/python3/hmi_create_csv_dataset.py +++ b/src/askem_beaker/contexts/dataset/procedures/python3/hmi_create_csv_dataset.py @@ -23,7 +23,7 @@ # Make the HTTP PUT request to upload the file bytes url = f'{hmi_server}/datasets/{id}/upload-csv' - response = requests.put(url, data=payload, files=files, auth={{auth}}) + response = requests.put(url, data=payload, files=files, auth=("{{username}}", "{{password}}")) # Check the response status code if response.status_code < 300: