From 0a58d9eaf0f96e797db1c69dfe00ebd3aed0b8bd Mon Sep 17 00:00:00 2001
From: SeaCelo <SeaCelo@users.noreply.github.com>
Date: Thu, 29 Aug 2024 17:18:07 -0400
Subject: [PATCH] Update input_output.py

Added a backup source for the SAM in a GH repository
---
 ogzaf/input_output.py | 37 ++++++++++++++++++++++++++++++-------
 1 file changed, 30 insertions(+), 7 deletions(-)

diff --git a/ogzaf/input_output.py b/ogzaf/input_output.py
index 6db9288..ecf454a 100644
--- a/ogzaf/input_output.py
+++ b/ogzaf/input_output.py
@@ -9,16 +9,39 @@
 # Read in SAM file
 storage_options = {"User-Agent": "Mozilla/5.0"}
 SAM_path = "https://www.wider.unu.edu/sites/default/files/Data/SASAM-2015-Data-Resource.xlsx"
+SAM_path_alt = "https://raw.githubusercontent.com/EAPD-DRB/SAM-files/main/Data/ZAF/SASAM-2015-Data-Resource.xlsx"
+
 if is_connected():
-    SAM = pd.read_excel(
-        SAM_path,
-        sheet_name="Micro SAM 2015",
-        skiprows=6,
-        index_col=0,
-        storage_options=storage_options,
-    )
+    try:
+        SAM = pd.read_excel(
+            SAM_path,
+            sheet_name="Micro SAM 2015",
+            skiprows=6,
+            index_col=0,
+            storage_options=storage_options,
+        )
+        print("Successfully read SAM from WIDER.")
+    except Exception as e:
+        print(f"Failed to read from WIDER: {e}")
+        try:
+            # Attempt to read from the GitHub repository
+            SAM = pd.read_excel(
+                SAM_path_alt,
+                sheet_name="Micro SAM 2015",
+                skiprows=6,
+                index_col=0,
+                storage_options=storage_options,
+            )
+            print("Successfully read SAM from GitHub repository.")
+        except Exception as e:
+            print(f"Failed to read from the GitHub repository: {e}")
+            SAM = None
+    # If both attempts fail, SAM will be None
+    if SAM is None:
+        print("Failed to read SAM from both sources.")
 else:
     SAM = None
+    print("No internet connection. SAM cannot be read.")
 
 
 def get_alpha_c(sam=SAM, cons_dict=CONS_DICT):