add bad url query

code/mapper.py

@@ -43,7 +43,9 @@ def getResumeUrl(context, queryParams):
     if not 'uid' in queryParams:
         return badRequest("Query parameter 'uid' is missing.")
     try:
-        url: str = get_resume_url(queryParams['uid'])
+        url: str | None = get_resume_url(queryParams['uid'])
+        if not url:
+            return badRequest("This UID has no resume.")
         rval = {
             "statusCode": 200,
             "body": json.dumps({

code/recruiter/get.py

@@ -1,11 +1,21 @@
 import boto3, os
 s3 = boto3.client('s3', region_name=os.environ.get("AWS_REGION", "us-east-1"))
-def get_resume_url(uid: str) -> str:
+def check_key_exists(bucket, key):
+    try:
+        s3.head_object(Bucket='bucket_name', Key='file_path')
+        return True
+    except:
+        return False
+
+def get_resume_url(uid: str) -> str | None:
+    bucket = 'infra-resume-book-pdfs'
     filename = f'resume_{uid}.pdf'
+    if not check_key_exists(bucket, filename):
+        return None
     return s3.generate_presigned_url(
         ClientMethod="get_object",
         Params={
-            "Bucket": 'infra-resume-book-pdfs',
+            "Bucket": bucket,
             "Key": filename
         },
         ExpiresIn=3600