From 0e0504581775f9660c23f8056f9c12fde93118a0 Mon Sep 17 00:00:00 2001
From: brmnjsh <jburman@yas.getyardstick.com>
Date: Sun, 3 Sep 2023 21:40:35 +0000
Subject: [PATCH] add return types

---
 app/helpers/aws_helper.py | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/app/helpers/aws_helper.py b/app/helpers/aws_helper.py
index ff7d757..3c5e029 100644
--- a/app/helpers/aws_helper.py
+++ b/app/helpers/aws_helper.py
@@ -16,21 +16,21 @@ else:
     s3 = session.client('s3', region_name=ApplicationConfigs.region_name)
     sqs = session.client('sqs', region_name=ApplicationConfigs.region_name)
 
-def get_key_from_message(body: dict):
+def get_key_from_message(body: dict) -> str:
     return body['Records'][0]['s3']['object']['key']
 
 
-def get_bucket_from_message(body: dict):
+def get_bucket_from_message(body: dict) -> str:
     return body['Records'][0]['s3']['bucket']['name']
 
 
-def get_object(key: str, bucket: str):
+def get_object(key: str, bucket: str) -> bytes:
     return s3.get_object(
         Bucket=bucket,
         Key=key,
     )['Body'].read()
 
-def get_object_tag(key: str, bucket: str, tag_key: str):
+def get_object_tag(key: str, bucket: str, tag_key: str) -> str | None:
     tags = get_object_tags(key, bucket)
     tag_index = 0
 
@@ -41,7 +41,7 @@ def get_object_tag(key: str, bucket: str, tag_key: str):
     else:
         return None
 
-def get_object_tags(key: str, bucket: str):
+def get_object_tags(key: str, bucket: str) -> list:
     tags = s3.get_object_tagging(Bucket=bucket, Key=key)['TagSet']
     return tags