Merge pull request #9 from moan0s/shelter-fixes

Add script to upload data of animal shelters

scripts/upload_animal_shelters.py

@@ -0,0 +1,97 @@
+import argparse
+import json
+import os
+
+import requests
+
+DEFAULT_OSM_DATA_FILE = "osm_data.geojson"
+
+
+def parse_args():
+    """Parse command-line arguments."""
+    parser = argparse.ArgumentParser(description="Upload animal shelter data 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("--data-file", type=str, help="Path to the GeoJSON file containing (only) animal shelters.")
+    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")
+    data_file = args.data_file or os.getenv("NOTFELLCHEN_DATA_FILE", DEFAULT_OSM_DATA_FILE)
+
+    if not api_token or not instance:
+        raise ValueError("API token and instance URL must be provided via environment variables or CLI arguments.")
+
+    return api_token, instance, data_file
+
+
+def load_osm_data(file_path):
+    """Load OSM data from a GeoJSON file."""
+    with open(file_path, "r", encoding="utf-8") as file:
+        data = json.load(file)
+    return data
+
+
+def load_osm_data(file_path):
+    #Load OSM data from a GeoJSON file.
+    with open(file_path, "r", encoding="utf-8") as file:
+        data = json.load(file)
+    return data
+
+
+def transform_osm_data(feature):
+    #Transform a single OSM feature into the API payload format
+    prop = feature.get("properties", {})
+    geometry = feature.get("geometry", {})
+
+    return {
+        "name": prop.get("name", "Unnamed Shelter"),
+        "phone": prop.get("phone"),
+        "website": prop.get("website"),
+        "opening_hours": prop.get("opening_hours"),
+        "email": prop.get("email"),
+        "location_string": f'{prop.get("addr:street", "")} {prop.get("addr:housenumber", "")} {prop.get("addr:postcode", "")} {prop.get("addr:city", "")}',
+        "external_object_id": prop.get("@id"),
+        "external_source_id": "OSM"
+    }
+
+
+def send_to_api(data, endpoint, headers):
+    # Send transformed data to the Notfellchen API.
+    response = requests.post(endpoint, headers=headers, json=data)
+    if response.status_code == 201:
+        print(f"Success: Shelter '{data['name']}' uploaded.")
+    elif response.status_code == 400:
+        print(f"Error: Shelter '{data['name']}' already exists or invalid data. {response.text}")
+    else:
+        print(f"Unexpected Error: {response.status_code} - {response.text}")
+        raise ConnectionError
+
+
+def main():
+    # Get configuration
+    api_token, instance, data_file = get_config()
+
+    # Set headers and endpoint
+    endpoint = f"{instance}/api/organizations/"
+    headers = {
+        "Authorization": f"Token {api_token}",
+        "Content-Type": "application/json"
+    }
+
+    # Step 1: Load OSM data
+    osm_data = load_osm_data(data_file)
+
+    # Step 2: Process each shelter and send it to the API
+    for feature in osm_data.get("features", []):
+        shelter_data = transform_osm_data(feature)
+        send_to_api(shelter_data, endpoint, headers)
+
+
+if __name__ == "__main__":
+    main()

src/fellchensammlung/api/views.py

@@ -16,6 +16,7 @@ from .serializers import (
 from fellchensammlung.models import Animal, RescueOrganization, AdoptionNotice, Species, Image
 from drf_spectacular.utils import extend_schema
 
+
 class AdoptionNoticeApiView(APIView):
     permission_classes = [IsAuthenticated]
 
@@ -84,7 +85,6 @@ class AdoptionNoticeApiView(APIView):
         )
 
 
-
 class AnimalApiView(APIView):
     permission_classes = [IsAuthenticated]
 
@@ -118,6 +118,7 @@ class AnimalApiView(APIView):
             )
         return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
 
+
 class RescueOrganizationApiView(APIView):
     permission_classes = [IsAuthenticated]
 
@@ -159,13 +160,14 @@ class RescueOrganizationApiView(APIView):
         """
         serializer = RescueOrgSerializer(data=request.data, context={"request": request})
         if serializer.is_valid():
-            rescue_org = serializer.save(owner=request.user)
+            rescue_org = serializer.save()
             return Response(
                 {"message": "Rescue organization created/updated successfully!", "id": rescue_org.id},
                 status=status.HTTP_201_CREATED,
             )
         return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
 
+
 class AddImageApiView(APIView):
     permission_classes = [IsAuthenticated]
 

src/fellchensammlung/static/robots.txt

@@ -0,0 +1,7 @@
+User-agent: *
+Disallow: /admin/
+
+User-agent: OpenAI
+Disallow: /
+
+Sitemap: https://notfellchen.org/sitemap.xml