add param checks to hdx_dataset

EL-BID · Aug 21, 2024 · d461ef5 · d461ef5
1 parent 285f2b8
commit d461ef5
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/urbanpy/download/download.py b/urbanpy/download/download.py
@@ -472,6 +472,14 @@ def hdx_dataset(resource):
         stacklevel=2,
     )
 
-    hdx_url = f"https://data.humdata.org/dataset/{resource}"
+    if not resource.ends_with(".csv"):
+        raise AttributeError("This function only expects a CSV file.")
+
+    if not resource.starts_with("https://data.humdata.org/dataset/"):
+        hdx_url = f"https://data.humdata.org/dataset/{resource}"
+    else:
+        hdx_url = hdx_url
+
     dataset = pd.read_csv(hdx_url)
+
     return dataset