From 60d0b0194d97053b773d9ab8cbcdad47a0494a91 Mon Sep 17 00:00:00 2001
From: Andrew <andreasohaodha@proton.me>
Date: Mon, 24 Mar 2025 23:13:35 +0000
Subject: [PATCH] [server]: Add update_average_punctuality

---
 .../update_average_punctuality/__init__.py    |  0
 .../lambda_function.py                        | 54 +++++++++++++++++++
 .../requirements.txt                          |  2 +
 3 files changed, 56 insertions(+)
 create mode 100644 server/src/functions/update_average_punctuality/__init__.py
 create mode 100644 server/src/functions/update_average_punctuality/lambda_function.py
 create mode 100644 server/src/functions/update_average_punctuality/requirements.txt

diff --git a/server/src/functions/update_average_punctuality/__init__.py b/server/src/functions/update_average_punctuality/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/server/src/functions/update_average_punctuality/lambda_function.py b/server/src/functions/update_average_punctuality/lambda_function.py
new file mode 100644
index 0000000..8c0f3f2
--- /dev/null
+++ b/server/src/functions/update_average_punctuality/lambda_function.py
@@ -0,0 +1,54 @@
+import json
+import boto3
+import requests
+
+dynamodb = boto3.resource("dynamodb")
+table = dynamodb.Table("punctuality_by_objectID")
+API_URL = "https://281bc6mcm5.execute-api.us-east-1.amazonaws.com/transient_data?objectType=IrishRailTrain"
+
+def fetch_train_data():
+    response = requests.get(API_URL)
+    if response.status_code == 200:
+        return response.json()
+    else:
+        print(f"Failed to fetch data: {response.status_code}")
+        return []
+
+
+def update_punctuality(objectID, new_punctuality):
+    response = table.get_item(Key={"objectID": objectID})
+    if "Item" in response:
+        item = response["Item"]
+        old_avg = float(item["average_punctuality"])
+        count = int(item["count"])
+
+        # Calculate new average
+        new_avg = ((old_avg * count) + new_punctuality) / (count + 1)
+        count += 1
+
+        # Update the DynamoDB table
+        table.update_item(
+            Key={"objectID": objectID},
+            UpdateExpression="SET average_punctuality = :avg, count = :cnt",
+            ExpressionAttributeValues={":avg": new_avg, ":cnt": count}
+        )
+    else:
+        # Insert new train punctuality record
+        table.put_item(
+            Item={"objectID": objectID, "average_punctuality": new_punctuality, "count": 1}
+        )
+
+
+def lambda_handler(event, context):
+    train_data = fetch_train_data()
+    for train in train_data:
+        objectID = train.get("objectID")
+        punctuality = int(train.get("trainPunctuality", 0))
+
+        if objectID:
+            update_punctuality(objectID, punctuality)
+
+    return {
+        "statusCode": 200,
+        "body": json.dumps("Punctuality data updated successfully")
+    }
diff --git a/server/src/functions/update_average_punctuality/requirements.txt b/server/src/functions/update_average_punctuality/requirements.txt
new file mode 100644
index 0000000..ea948fa
--- /dev/null
+++ b/server/src/functions/update_average_punctuality/requirements.txt
@@ -0,0 +1,2 @@
+boto3
+requests