Updated <error> handling.

lexara-prime-ai · Jul 8, 2024 · f731f29 · f731f29
1 parent 1e98fba
commit f731f29
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/hyper/hyper/drive.py b/hyper/hyper/drive.py
@@ -1,4 +1,8 @@
+# Utility modules.
 import constants
+import error_handling
+
+# Google deps.
 from google.oauth2 import service_account
 from googleapiclient.discovery import build
 
@@ -13,7 +17,10 @@ def authenticate():
 
         return credentials
     except Exception as e:
-        print("\n[ERROR] -> Failed to [Authenticate] <wspr_cdk> service user: \n", e)
+        error_handling.ErrorHandling.propagate_error(
+            f"{authenticate.__name__}",
+            f"\n[ERROR] -> Failed to [Authenticate] <wspr_cdk> service user: {e}\n"
+        )
 
 
 def upload_to_drive(file_path):
@@ -31,4 +38,7 @@ def upload_to_drive(file_path):
 
         service.files().create(body=file_metadata, media_body=file_path).execute()
     except Exception as e:
-        print("\n[ERROR] -> Failed to upload to Google Drive: \n", e)
+        error_handling.ErrorHandling.propagate_error(
+            f"{upload_to_drive.__name__}",
+            f"\n[ERROR] -> Failed to upload to Google Drive: {e}\n"
+        )