From 4788012ce6a5d411c04e33df0f8c6eefd3417cc7 Mon Sep 17 00:00:00 2001
From: brmnjsh <jburman@yas.getyardstick.com>
Date: Mon, 11 Sep 2023 13:23:04 +0000
Subject: [PATCH] make more pythonic

---
 app/helpers/aws_helper.py | 8 +-------
 1 file changed, 1 insertion(+), 7 deletions(-)

diff --git a/app/helpers/aws_helper.py b/app/helpers/aws_helper.py
index cccda41..38eb4fe 100644
--- a/app/helpers/aws_helper.py
+++ b/app/helpers/aws_helper.py
@@ -31,13 +31,7 @@ def get_object(key: str, bucket: str) -> bytes:
 
 def get_object_tag(key: str, bucket: str, tag_key: str) -> Union[str, None]:
     tags = get_object_tags(key, bucket)
-
-    for i, tag in enumerate(tags):
-        tag = tags[i]
-        if tag['Key'] == tag_key:
-            return tag
-    else:
-        return None
+    return next((tag for tag in tags if tag['Key'] == tag_key), None)
 
 def get_object_tags(key: str, bucket: str) -> list:
     tags = s3.get_object_tagging(Bucket=bucket, Key=key)['TagSet']