my-aws-helpers 5.8.0__tar.gz → 6.0.0__tar.gz

{my_aws_helpers-5.8.0 → my_aws_helpers-6.0.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: my_aws_helpers
-Version: 5.8.0
+Version: 6.0.0
 Summary: AWS Helpers
 Home-page: https://github.com/JarrodMccarthy/aws_helpers.git
 Author: Jarrod McCarthy

{my_aws_helpers-5.8.0 → my_aws_helpers-6.0.0}/my_aws_helpers/bedrock.py

@@ -111,11 +111,11 @@ class OCRResultQueries(BaseS3Queries):
         results = [r for r in results if r is not None]
         return results
 
-    def get_listing_by_key_prefix(self, prefix: str) -> List[OCRResult]:
+    def get_ocr_results_by_key_prefix(self, prefix: str) -> List[OCRResult]:
         locations = self.s3_client.list_objects_by_prefix(
             bucket_name=self.bucket_name, prefix=prefix
         )
-        objects = self._concurrent_s3_read(listing_locations=locations)
+        objects = self._concurrent_s3_read(locations=locations)
         ocr_results = [OCRResult.from_s3_representation(obj=obj) for obj in objects]
         return ocr_results
 

{my_aws_helpers-5.8.0 → my_aws_helpers-6.0.0}/my_aws_helpers/s3.py

@@ -2,6 +2,7 @@ from __future__ import annotations
 import boto3
 import io
 import json
+from concurrent.futures import Future, ThreadPoolExecutor
 from abc import ABC, abstractmethod
 from typing import Optional, Any, Dict, List
 from datetime import datetime, date
@@ -320,6 +321,23 @@ class BaseS3Queries:
         self.s3_client = s3_client
         self.bucket_name = bucket_name
 
+    def _concurrent_s3_dict_read(
+        self, locations: List[S3Location], max_workers: int = 10
+    ) -> List[BaseS3Object]:
+        results: List[BaseS3Object] = list()
+        futures: List[Future] = list()
+        with ThreadPoolExecutor(max_workers=max_workers) as executor:
+            for location in locations:
+                future = executor.submit(
+                    self.s3_client.read_dict_from_s3,
+                    s3_location=location,
+                )
+                futures.append(future)
+        for f in futures:
+            results.append(f.result())
+        results = [r for r in results if r is not None]
+        return results
+
     def save_s3_object_to_s3(self, object: BaseS3Object) -> Optional[S3Location]:
         try:
             obj = object.to_s3_representation()

{my_aws_helpers-5.8.0 → my_aws_helpers-6.0.0}/my_aws_helpers.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: my-aws-helpers
-Version: 5.8.0
+Version: 6.0.0
 Summary: AWS Helpers
 Home-page: https://github.com/JarrodMccarthy/aws_helpers.git
 Author: Jarrod McCarthy

{my_aws_helpers-5.8.0 → my_aws_helpers-6.0.0}/setup.py

@@ -3,7 +3,7 @@ from setuptools import find_namespace_packages, setup
 
 base_path = os.path.abspath(os.path.dirname(__file__))
 
-version = "5.8.0"
+version = "6.0.0"
 
 core = [
     "boto3==1.34.36",