Rename lambda function to adhere with structure required by AWS lambda.

niccolozanotti · Sep 4, 2024 · 2c2878b · 2c2878b
1 parent 495b26c
commit 2c2878b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/.github/workflows/deploy-lambda-to-aws.yml b/.github/workflows/deploy-lambda-to-aws.yml
@@ -28,7 +28,7 @@ jobs:
         cd package
         zip -r9 ../lambda_function.zip .
         cd ..
-        zip -g lambda_function.zip registration-app.py
+        zip -g lambda_function.zip lambda-function.py
 
     - name: Deploy to AWS Lambda
       env:

diff --git a/registration-app.py → lambda_function.py b/registration-app.py → lambda_function.py
@@ -3,7 +3,7 @@
 from datetime import datetime, timezone
 import csv
 import boto3
-import os
+
 
 app = Flask(__name__)
 CORS(app, resources={r"/*": {"origins": "*"}}, supports_credentials=True)
@@ -26,16 +26,16 @@ def register():
     name = data['name']
     occupants.add(name)
     log_action(name, 'register')
-    return jsonify({"status": "registered", "occupants": len(occupants)})
+    return jsonify({"status": "registered", "occupants": list(occupants)})
 
 
 @app.route('/unregister', methods=['POST'])
 def unregister():
     data = request.json
     name = data['name']
-    occupants.remove(name)
+    occupants.discard(name)   # set.discard() avoids KeyError if name not in set
     log_action(name, 'unregister')
-    return jsonify({"status": "unregistered", "occupants": len(occupants)})
+    return jsonify({"status": "unregistered", "occupants": list(occupants)})
 
 
 @app.route('/status', methods=['GET'])