feat: add script to export shelters with websites for #29

scripts/extract_rescue_org_websites.py

@@ -0,0 +1,66 @@
+import argparse
+import csv
+import os
+
+import requests
+
+OUTPUT_FILE = "export.csv"
+
+
+def parse_args():
+    """Parse command-line arguments."""
+    parser = argparse.ArgumentParser(
+        description="Download animal shelter data from the Overpass API to the Notfellchen API.")
+    parser.add_argument("--api-token", type=str, help="API token for authentication.")
+    parser.add_argument("--instance", type=str, help="API instance URL.")
+    parser.add_argument("--output-file", type=str, help="Path output file.")
+    return parser.parse_args()
+
+
+def get_config():
+    """Get configuration from environment variables or command-line arguments."""
+    args = parse_args()
+
+    api_token = args.api_token or os.getenv("NOTFELLCHEN_API_TOKEN")
+    instance = args.instance or os.getenv("NOTFELLCHEN_INSTANCE")
+    output_file = args.output_file or OUTPUT_FILE
+
+    return api_token, instance, output_file
+
+
+def rat_specific_url_or_none(org):
+    try:
+        urls = org["species_specific_urls"]
+        for url in urls:
+            # 1 is the key for rats
+            if url["species"] == 1:
+                return url["url"]
+        # Return none if no url for this species is found
+        return None
+    except KeyError:
+        return None
+
+
+def main():
+    api_token, instance, output_file = get_config()
+
+    # Set headers and endpoint
+    endpoint = f"{instance}/api/organizations/"
+    h = {'Authorization': f'Token {api_token}', "content-type": "application/json"}
+
+    rescue_orgs_result = requests.get(endpoint, headers=h)
+
+    with open(output_file, 'w') as csvfile:
+        fieldnames = ['id', 'name', 'website', 'rat_specific_website']
+        writer = csv.DictWriter(csvfile, fieldnames=fieldnames)
+
+        writer.writeheader()
+        for org in rescue_orgs_result.json():
+            writer.writerow({'id': org["id"],
+                             'name': org["name"],
+                             'website': org["website"],
+                             "rat_specific_website": rat_specific_url_or_none(org)})
+
+
+if __name__ == "__main__":
+    main()