boto3-assist 0.35.0__py3-none-any.whl → 0.36.0__py3-none-any.whl

boto3_assist/dynamodb/dynamodb_connection.py

@@ -35,6 +35,7 @@ class DynamoDBConnection(Connection):
         assume_role_arn: Optional[str] = None,
         assume_role_chain: Optional[List[str]] = None,
         assume_role_duration_seconds: Optional[int] = 3600,
+        use_connection_pool: bool = False,
     ) -> None:
         super().__init__(
             service_name="dynamodb",
@@ -46,6 +47,7 @@ class DynamoDBConnection(Connection):
             assume_role_arn=assume_role_arn,
             assume_role_chain=assume_role_chain,
             assume_role_duration_seconds=assume_role_duration_seconds,
+            use_connection_pool=use_connection_pool,
         )
 
         self.__dynamodb_client: DynamoDBClient | None = None

boto3_assist/s3/__init__.py

@@ -0,0 +1,9 @@
+"""
+Geek Cafe, LLC
+Maintainers: Eric Wilson
+MIT License. See Project Root for the license information.
+"""
+
+from .s3 import S3
+
+__all__ = ["S3"]

boto3_assist/s3/s3.py

@@ -26,6 +26,7 @@ class S3(S3Connection):
         aws_end_point_url: Optional[str] = None,
         aws_access_key_id: Optional[str] = None,
         aws_secret_access_key: Optional[str] = None,
+        use_connection_pool: bool = False,
     ) -> None:
         """_summary_
 
@@ -35,6 +36,7 @@ class S3(S3Connection):
             aws_end_point_url (Optional[str], optional): _description_. Defaults to None.
             aws_access_key_id (Optional[str], optional): _description_. Defaults to None.
             aws_secret_access_key (Optional[str], optional): _description_. Defaults to None.
+            use_connection_pool (bool, optional): Use connection pooling. Defaults to False.
         """
         super().__init__(
             aws_profile=aws_profile,
@@ -42,11 +44,52 @@ class S3(S3Connection):
             aws_end_point_url=aws_end_point_url,
             aws_access_key_id=aws_access_key_id,
             aws_secret_access_key=aws_secret_access_key,
+            use_connection_pool=use_connection_pool,
         )
 
         self.__s3_object: S3Object | None = None
         self.__s3_bucket: S3Bucket | None = None
 
+    @classmethod
+    def from_pool(
+        cls,
+        aws_profile: Optional[str] = None,
+        aws_region: Optional[str] = None,
+        aws_end_point_url: Optional[str] = None,
+        **kwargs,
+    ) -> "S3":
+        """
+        Create S3 connection using connection pool (recommended for Lambda).
+
+        This is the recommended pattern for Lambda functions as it reuses
+        boto3 sessions across invocations in warm containers.
+
+        Args:
+            aws_profile: AWS profile name (optional)
+            aws_region: AWS region (optional)
+            aws_end_point_url: Custom endpoint URL (optional, for moto testing)
+            **kwargs: Additional S3 parameters
+
+        Returns:
+            S3 instance configured to use connection pool
+
+        Example:
+            >>> # Recommended pattern for Lambda
+            >>> s3 = S3.from_pool()
+            >>> s3.object.upload_file(file_path="/tmp/file.txt", bucket="my-bucket", key="file.txt")
+            >>>
+            >>> # Subsequent calls reuse the same connection
+            >>> s3_2 = S3.from_pool()
+            >>> assert s3.session is s3_2.session
+        """
+        return cls(
+            aws_profile=aws_profile,
+            aws_region=aws_region,
+            aws_end_point_url=aws_end_point_url,
+            use_connection_pool=True,
+            **kwargs,
+        )
+
     @property
     def object(self) -> S3Object:
         """s3 object"""

boto3_assist/s3/s3_connection.py

@@ -35,6 +35,7 @@ class S3Connection(Connection):
         aws_access_key_id: Optional[str] = None,
         aws_secret_access_key: Optional[str] = None,
         signature_version: Optional[str] = None,
+        use_connection_pool: bool = False,
     ) -> None:
         # Build S3-specific config if signature_version is specified
         config: Optional[Config] = None
@@ -50,6 +51,7 @@ class S3Connection(Connection):
             aws_secret_access_key=aws_secret_access_key,
             aws_end_point_url=aws_end_point_url,
             config=config,
+            use_connection_pool=use_connection_pool,
         )
 
         self.__client: S3Client | None = None

boto3_assist/sqs/sqs_connection.py

@@ -33,6 +33,7 @@ class SQSConnection(Connection):
         aws_end_point_url: Optional[str] = None,
         aws_access_key_id: Optional[str] = None,
         aws_secret_access_key: Optional[str] = None,
+        use_connection_pool: bool = False,
     ) -> None:
         """
         Initialize SQS connection.
@@ -43,6 +44,7 @@ class SQSConnection(Connection):
             aws_end_point_url: Custom endpoint URL (for LocalStack, etc.)
             aws_access_key_id: AWS access key ID
             aws_secret_access_key: AWS secret access key
+            use_connection_pool: Use connection pooling (recommended for Lambda)
         """
         super().__init__(
             service_name="sqs",
@@ -51,11 +53,52 @@ class SQSConnection(Connection):
             aws_access_key_id=aws_access_key_id,
             aws_secret_access_key=aws_secret_access_key,
             aws_end_point_url=aws_end_point_url,
+            use_connection_pool=use_connection_pool,
         )
 
         self.__client: SQSClient | None = None
         self.__resource: SQSServiceResource | None = None
 
+    @classmethod
+    def from_pool(
+        cls,
+        aws_profile: Optional[str] = None,
+        aws_region: Optional[str] = None,
+        aws_end_point_url: Optional[str] = None,
+        **kwargs,
+    ) -> "SQSConnection":
+        """
+        Create SQS connection using connection pool (recommended for Lambda).
+
+        This is the recommended pattern for Lambda functions as it reuses
+        boto3 sessions across invocations in warm containers.
+
+        Args:
+            aws_profile: AWS profile name (optional)
+            aws_region: AWS region (optional)
+            aws_end_point_url: Custom endpoint URL (optional, for LocalStack/moto)
+            **kwargs: Additional SQS parameters
+
+        Returns:
+            SQSConnection instance configured to use connection pool
+
+        Example:
+            >>> # Recommended pattern for Lambda
+            >>> sqs = SQSConnection.from_pool()
+            >>> queue = SQSQueue(sqs, queue_url="https://...")
+            >>>
+            >>> # Subsequent calls reuse the same connection
+            >>> sqs2 = SQSConnection.from_pool()
+            >>> assert sqs.session is sqs2.session
+        """
+        return cls(
+            aws_profile=aws_profile,
+            aws_region=aws_region,
+            aws_end_point_url=aws_end_point_url,
+            use_connection_pool=True,
+            **kwargs,
+        )
+
     @property
     def client(self) -> SQSClient:
         """Get SQS client."""

boto3_assist/version.py

@@ -1 +1 @@
-__version__ = "0.35.0"
+__version__ = "0.36.0"

{boto3_assist-0.35.0.dist-info → boto3_assist-0.36.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: boto3_assist
-Version: 0.35.0
+Version: 0.36.0
 Summary: Additional boto3 wrappers to make your life a little easier
 Author-email: Eric Wilson <boto3-assist@geekcafe.com>
 License-File: LICENSE-EXPLAINED.txt

{boto3_assist-0.35.0.dist-info → boto3_assist-0.36.0.dist-info}/RECORD

@@ -1,12 +1,13 @@
-boto3_assist/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+boto3_assist/__init__.py,sha256=lS-oHTASzgSd4UyYByxydFm_EQjTCYvIelzkG6pPgY4,231
 boto3_assist/aws_config.py,sha256=evmk_blj498ugptJa8lxIDJOIursxY6mT4joLfEbrl0,6558
 boto3_assist/boto3session.py,sha256=p4FKVSX5A-xNHdpRan8pgMoY4iIywNfwriyTfjQ-zTQ,2967
-boto3_assist/connection.py,sha256=dfVq2nAwElmo0ek0PLBEf9HdULOuzKZG0fu7tDDewhA,5046
+boto3_assist/connection.py,sha256=veKhcZJew3hDukSkUgQ84Ow3XbxwVeEPlGj5IBORMtI,8288
+boto3_assist/connection_pool.py,sha256=I4pfZu7PKW7nkz2lWhlUDK6OB8G7a--GjTyBjerD6FE,6053
 boto3_assist/connection_tracker.py,sha256=UgfR9RlvXf3A4ssMr3gDMpw89ka8mSRvJn4M34SzhbU,4378
 boto3_assist/http_status_codes.py,sha256=G0zRSWenwavYKETvDF9tNVUXQz3Ae2gXdBETYbjvJe8,3284
 boto3_assist/role_assumption_mixin.py,sha256=PMUU5yC2FUBjFD1UokVkRY3CPB5zTw85AhIB5BMtbc8,1031
 boto3_assist/session_setup_mixin.py,sha256=X-JQKyyaWNA8Z8kKgf2V2I5vsiLAH8udLTX_xepnsdQ,3140
-boto3_assist/version.py,sha256=w3TvNFSYTcLj3nBuRCGicelGZOzvgYOfnmgrBAuIKqY,23
+boto3_assist/version.py,sha256=ITH8IVCEQOekRajZpPo6sKNlMBNDRQRnzipIN0dqXoc,23
 boto3_assist/aws_lambda/event_info.py,sha256=OkZ4WzuGaHEu_T8sB188KBgShAJhZpWASALKRGBOhMg,14648
 boto3_assist/aws_lambda/mock_context.py,sha256=LPjHP-3YSoY6iPl1kPqJDwSVf1zLNTcukUunDtYcbK0,116
 boto3_assist/cloudwatch/cloudwatch_connection.py,sha256=mnGWaLSQpHh5EeY7Ek_2o9JKHJxOELIYtQVMX1IaHn4,2480
@@ -19,8 +20,9 @@ boto3_assist/cognito/cognito_connection.py,sha256=deuXR3cNHz0mCYff2k0LfAvK--9Okq
 boto3_assist/cognito/cognito_utility.py,sha256=IVZAg58nHG1U7uxe7FsTYpqwwZiwwdIBGiVTZuLCFqg,18417
 boto3_assist/cognito/jwks_cache.py,sha256=1Y9r-YfQ8qrgZN5xYPvjUEEV0vthbdcPdAIaPbZP7kU,373
 boto3_assist/cognito/user.py,sha256=qc44qLx3gwq6q2zMxcPQze1EjeZwy5Kuav93vbe-4WU,820
-boto3_assist/dynamodb/dynamodb.py,sha256=MElfzO0kDKHdMye5vtIOC4sGH32P2xxvPlxj76NEDe4,48632
-boto3_assist/dynamodb/dynamodb_connection.py,sha256=D4KmVpMpE0OuVOwW5g4JBWllUNkwy0hMXEGUiToAMBc,3608
+boto3_assist/dynamodb/__init__.py,sha256=uuRqUY9FHTwequ3oMq_VcMcmzhpAKG08GSnZPMNge2k,163
+boto3_assist/dynamodb/dynamodb.py,sha256=VHt5keB-N38nbbytT_pyY6P6PmveI5t4xrYVqVM6HR8,49163
+boto3_assist/dynamodb/dynamodb_connection.py,sha256=JtdS7ciJDuB1pLOUM2UKWxqNh2gXJtVIU9KYzV4VN2U,3704
 boto3_assist/dynamodb/dynamodb_helpers.py,sha256=BYJEuXaQVCPbDfbtPswWA_OvV_yC3fVoTtKvIoZeIBc,12092
 boto3_assist/dynamodb/dynamodb_importer.py,sha256=nCKsyRQeMqDSf0Q5mQ_X_oVIg4PRnu0hcUzZnBli610,3471
 boto3_assist/dynamodb/dynamodb_index.py,sha256=2AKxHo8HrRbaxL0ePj7S6ek36_sy5cHkDp5I9wIp8Kw,19797
@@ -42,15 +44,16 @@ boto3_assist/erc/__init__.py,sha256=ZVpE1TayNer4ZFb3t3wlo5LkWD9G-HbYE2DkoQoMI9w,
 boto3_assist/erc/ecr_connection.py,sha256=5fbJiouHe2uta4OiN-NKOo3fS2608Zcc01fWBOyPbI4,1370
 boto3_assist/errors/custom_exceptions.py,sha256=QAMW49NbClELVnRd00u4NHfzVtRS3Tc1TrsIMUP9wLw,1041
 boto3_assist/models/serializable_model.py,sha256=ZMrRJRvJWLY8PBSKK_nPCgYKv1qUxDPEVdcADKbIHsI,266
-boto3_assist/s3/s3.py,sha256=ESTPXtyDi8mrwHaYNWjQLNGTuTUV4CxKDqw-O_KGzKs,2052
+boto3_assist/s3/__init__.py,sha256=trlSMj2kjpY27qGkuSVXKwSbbvwTaosPznLkD9xYUVo,145
+boto3_assist/s3/s3.py,sha256=rw27qX7vQuZZ5KCenHun9p8K0vHjDNdZVSCdUrrsaLk,3593
 boto3_assist/s3/s3_bucket.py,sha256=GfyBbuI5BWz_ybwU_nDqUZiC0wt24PNt49GKZmb05OY,2018
-boto3_assist/s3/s3_connection.py,sha256=iIWtOgTXaj4FQfkyWLI9jnD340XKkKT2UqPAMvItBM4,2421
+boto3_assist/s3/s3_connection.py,sha256=jfoqeEYq4LF7Y9hVutzkPHuLR1oYiPA2k0O4PP1EPnc,2517
 boto3_assist/s3/s3_event_data.py,sha256=Q7QUI1pwkc7g6yZ3IZWMXBIAfsMlPRC7wac2RvrQoA4,4112
 boto3_assist/s3/s3_object.py,sha256=77jZeIUFpQX3cFYGGwRFBvL-peCe54iILnthm-GFjMc,22518
 boto3_assist/securityhub/securityhub.py,sha256=ne-J_v4DaCVZm5YgJa_-LKVomLJQo5Gpw6wleAKSsws,5467
 boto3_assist/securityhub/securityhub_connection.py,sha256=hWfcj9gjS2lNXUObyw4cShtveoqJPIp8kKFuz-fz1J4,1449
 boto3_assist/sqs/__init__.py,sha256=VRuPPit-hSSbumKL2pN5AUtZKr_c9MpLeuqQ5a6wBLk,285
-boto3_assist/sqs/sqs_connection.py,sha256=Gacka0Aqco3qNSTVmPiJf8LeaLgNzhL1-npCOn85GmQ,2469
+boto3_assist/sqs/sqs_connection.py,sha256=MEDA2KKlpoRiDcCVh_i37CLhA0a5dlNkfuA9voG-Lxs,4011
 boto3_assist/sqs/sqs_queue.py,sha256=sjz2Q4BbJMCO-cxSNM02UxMmgnRfPLFAC3Pt1kiqc14,9542
 boto3_assist/ssm/connection.py,sha256=gYpKn5HsUR3hcRUqJzF5QcTITCk0DReq9KhoE_8-Htg,1370
 boto3_assist/ssm/parameter_store/parameter_store.py,sha256=2ISi-SmR29mESHFH-onJkxPX1aThIgBRojA3ZoNcP9s,3949
@@ -63,8 +66,8 @@ boto3_assist/utilities/logging_utility.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5N
 boto3_assist/utilities/numbers_utility.py,sha256=wzv9d0uXT_2_ZHHio7LBzibwxPqhGpvbq9HinrVn_4A,10160
 boto3_assist/utilities/serialization_utility.py,sha256=m5wRZNeWW9VltQPVNziR27OGKO3MDJm6mFmcDHwN-n4,24479
 boto3_assist/utilities/string_utility.py,sha256=XxUIz19L2LFFTRDAAmdPa8Qhn40u9yO7g4nULFuvg0M,11033
-boto3_assist-0.35.0.dist-info/METADATA,sha256=xa5U1elxKIetkG0QKmbCEt_3ge6uO5ZyhdGNsNUW1o8,2879
-boto3_assist-0.35.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-boto3_assist-0.35.0.dist-info/licenses/LICENSE-EXPLAINED.txt,sha256=WFREvTpfTjPjDHpOLADxJpCKpIla3Ht87RUUGii4ODU,606
-boto3_assist-0.35.0.dist-info/licenses/LICENSE.txt,sha256=PXDhFWS5L5aOTkVhNvoitHKbAkgxqMI2uUPQyrnXGiI,1105
-boto3_assist-0.35.0.dist-info/RECORD,,
+boto3_assist-0.36.0.dist-info/METADATA,sha256=UJcDEqfuDx-499Z5YP6YwuhfkTWPfxjBUNNAOHyIxRQ,2879
+boto3_assist-0.36.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+boto3_assist-0.36.0.dist-info/licenses/LICENSE-EXPLAINED.txt,sha256=WFREvTpfTjPjDHpOLADxJpCKpIla3Ht87RUUGii4ODU,606
+boto3_assist-0.36.0.dist-info/licenses/LICENSE.txt,sha256=PXDhFWS5L5aOTkVhNvoitHKbAkgxqMI2uUPQyrnXGiI,1105
+boto3_assist-0.36.0.dist-info/RECORD,,