Add if checks for status code in put function

bcostaaa01 · Oct 29, 2023 · 7e88256 · 7e88256
1 parent 9aee413
commit 7e88256
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/classes/migration.py b/classes/migration.py
@@ -41,7 +41,10 @@ def post_data(self, data):
     def put_data(self, data):
         # put data to path
         response = requests.put(self.path, json=data, headers=self.headers)
-        return response
+        if response.status_code in self.status_codes:
+            print(f"Successfully put data to {self.path} 🎉")
+        else:
+            print(f"Error putting data to {self.path}: {response.status_code} - {response.text} ❌")
 
     def delete_data(self):
         # delete data from path