reach_commons 0.18.22__tar.gz → 0.18.24__tar.gz

{reach_commons-0.18.22 → reach_commons-0.18.24}/PKG-INFO

@@ -1,10 +1,10 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: reach_commons
-Version: 0.18.22
+Version: 0.18.24
 Summary: Reach Commons is a versatile utility library designed to streamline and enhance development workflows within the Reach ecosystem.
 License: MIT
-Author: Wilson Moraes
-Author-email: wmoraes@getreach.ai
+Author: Engineering
+Author-email: engineering@getreach.ai
 Requires-Python: >=3.8,<4.0
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
@@ -14,6 +14,8 @@ Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
 Requires-Dist: curlify (==3.0.0)
 Requires-Dist: fastapi (>=0.115.5)
 Requires-Dist: pydantic (>=2.9.2)

{reach_commons-0.18.22 → reach_commons-0.18.24}/README.md

@@ -1,26 +1,26 @@
-# reach_commons
-
-## Description
-
-`reach_commons` is a Python library that provides various classes and utilities to assist developers in their projects. This library aims to simplify and streamline processes by offering reusable components that can be easily integrated into various Python applications.
-
-## Installation
-
-Install `reach_commons` using pip:
-
-```bash
-pip install reach_commons
-
-from reach_commons.api_client_v2 import ReachApiGatewayV2
-
-api_gateway = ReachApiGatewayV2(
-                        base_url=os.environ["api_gateway_v2_endpoint"],
-                        access_token=os.environ["event_processor_token"],
-                    )
-response = api_gateway.stripe_create_booking_guarantee(
-                        business_id=business_id, booking_price=booking_price
-                    )
-{
-            "statusCode": stripe_customer.status_code,
-            "body": json.dumps(response.json()),
+# reach_commons
+
+## Description
+
+`reach_commons` is a Python library that provides various classes and utilities to assist developers in their projects. This library aims to simplify and streamline processes by offering reusable components that can be easily integrated into various Python applications.
+
+## Installation
+
+Install `reach_commons` using pip:
+
+```bash
+pip install reach_commons
+
+from reach_commons.api_client_v2 import ReachApiGatewayV2
+
+api_gateway = ReachApiGatewayV2(
+                        base_url=os.environ["api_gateway_v2_endpoint"],
+                        access_token=os.environ["event_processor_token"],
+                    )
+response = api_gateway.stripe_create_booking_guarantee(
+                        business_id=business_id, booking_price=booking_price
+                    )
+{
+            "statusCode": stripe_customer.status_code,
+            "body": json.dumps(response.json()),
         }

{reach_commons-0.18.22 → reach_commons-0.18.24}/pyproject.toml

@@ -1,41 +1,41 @@
-# isort .; black .; poetry build; poetry publish
-[tool.poetry]
-name = "reach_commons"
-version = "0.18.22"
-description = "Reach Commons is a versatile utility library designed to streamline and enhance development workflows within the Reach ecosystem."
-authors = ["Wilson Moraes <wmoraes@getreach.ai>"]
-license = "MIT"
-readme = "README.md"
-
-classifiers = [
-    "Programming Language :: Python :: 3",
-    "License :: OSI Approved :: MIT License",
-    "Operating System :: OS Independent",
-]
-
-[tool.poetry.dependencies]
-python = "^3.8"
-#pip = "23.3.1"
-setuptools = "70.0.0"
-wheel = "0.41.2"
-twine = "4.0.2"
-curlify = "3.0.0"
-pydantic = ">=2.9.2"
-fastapi = ">=0.115.5"
-
-
-[tool.poetry.dev-dependencies]
-black = "23.10.0"
-isort = "5.12.0"
-requests = "2.31.0"
-boto3 = "1.28.68"
-botocore = "^1.31.68"
-redis = "^4.1.3"
-pydantic-settings = "2.3.1"
-jsonschema = "4.23.0"
-motor = "3.6.0"
-pymongo = "4.9.2"
-
-[build-system]
-requires = ["poetry-core>=1.0.0"]
+# isort .; black .; poetry build; poetry publish
+[tool.poetry]
+name = "reach_commons"
+version = "0.18.24"
+description = "Reach Commons is a versatile utility library designed to streamline and enhance development workflows within the Reach ecosystem."
+authors = ["Engineering <engineering@getreach.ai>"]
+license = "MIT"
+readme = "README.md"
+
+classifiers = [
+    "Programming Language :: Python :: 3",
+    "License :: OSI Approved :: MIT License",
+    "Operating System :: OS Independent",
+]
+
+[tool.poetry.dependencies]
+python = "^3.8"
+#pip = "23.3.1"
+setuptools = "70.0.0"
+wheel = "0.41.2"
+twine = "4.0.2"
+curlify = "3.0.0"
+pydantic = ">=2.9.2"
+fastapi = ">=0.115.5"
+
+
+[tool.poetry.dev-dependencies]
+black = "23.10.0"
+isort = "5.12.0"
+requests = "2.31.0"
+boto3 = "1.28.68"
+botocore = "^1.31.68"
+redis = "^4.1.3"
+pydantic-settings = "2.3.1"
+jsonschema = "4.23.0"
+motor = "3.6.0"
+pymongo = "4.9.2"
+
+[build-system]
+requires = ["poetry-core>=1.0.0"]
 build-backend = "poetry.core.masonry.api"

{reach_commons-0.18.22 → reach_commons-0.18.24}/reach_commons/reach_aws/dynamo_db.py

@@ -174,9 +174,11 @@ class BusinessReviewsCommonQueries:
                 Item={
                     "PK": f"business#{business_id}",
                     "SK": "info",
-                    "created_at": created_at_isoformat
-                    if created_at_isoformat
-                    else datetime.utcnow().isoformat(),
+                    "created_at": (
+                        created_at_isoformat
+                        if created_at_isoformat
+                        else datetime.utcnow().isoformat()
+                    ),
                     **filtered_info,
                 },
                 ConditionExpression="attribute_not_exists(PK) AND attribute_not_exists(SK)",

{reach_commons-0.18.22 → reach_commons-0.18.24}/reach_commons/reach_aws/kms.py

@@ -97,9 +97,9 @@ class KMSClient(BaseKMSClient):
             ciphertext = self.client.encrypt(
                 KeyId=key_alias,
                 Plaintext=bytes(data_to_encrypt, encoding="utf8"),
-                EncryptionContext={"CustomContext": f"{custom_context}"}
-                if custom_context
-                else None,
+                EncryptionContext=(
+                    {"CustomContext": f"{custom_context}"} if custom_context else None
+                ),
             )
             encrypted = base64.b64encode(ciphertext["CiphertextBlob"])
         except ClientError as exc:

{reach_commons-0.18.22 → reach_commons-0.18.24}/reach_commons/reach_aws/sqs.py

@@ -4,7 +4,6 @@ from decimal import Decimal
 from functools import cached_property
 
 import boto3
-import requests
 from botocore.exceptions import ClientError
 
 from reach_commons.app_logging.logger import get_reach_logger
@@ -147,6 +146,81 @@ class SQSClient(BaseSQSClient):
 
         return True
 
+    def _get_visible_messages_count(self) -> int:
+        try:
+            resp = self.client.get_queue_attributes(
+                QueueUrl=self.topic_name,
+                AttributeNames=["ApproximateNumberOfMessages"],
+            )
+            return int(resp["Attributes"].get("ApproximateNumberOfMessages", 0))
+        except ClientError as exc:
+            self.logger.error(
+                "error_fetching_queue_attributes, topic_name=%s, error=%s",
+                self.topic_name,
+                str(exc),
+            )
+            # If there's an error, return 0 so as not to block the caller.
+            return 0
+
+    def publish_with_capacity_guard(
+        self,
+        message_data: dict,
+        max_capacity: int,
+        delay_seconds: int = 60,
+        message_attributes: dict = None,
+    ) -> dict:
+        """
+        Attempts to publish a message while enforcing an approximate queue capacity limit.
+
+        Behavior:
+            - If the queue is below `max_capacity`, the message is published normally.
+            - If the queue is at or above `max_capacity`, the message is NOT published.
+
+        Returns a dict with:
+            - posted: bool        -> whether the message was actually published
+            - visible_count: int  -> ApproximateNumberOfMessages at publication time
+            - max_capacity: int   -> the capacity threshold used
+
+        Example:
+            sqs = SQSClient(topic_name="<enviroment>-messages-queue")
+
+            Result = sqs.publish_with_capacity_guard(
+                message_data={"foo": "bar"},
+                max_capacity=100_000,
+                delay_seconds=30,
+                message_attributes={"service_name": "TestClient"},
+            )
+
+            if not Result["posted"]:
+                print("Queue is over capacity:", Result)
+        """
+        visible_count = self._get_visible_messages_count()
+
+        if visible_count >= max_capacity:
+            self.logger.warning(
+                "queue_over_capacity, topic_name=%s, visible=%s, max_capacity=%s",
+                self.topic_name,
+                visible_count,
+                max_capacity,
+            )
+            return {
+                "posted": False,
+                "visible_count": visible_count,
+                "max_capacity": max_capacity,
+            }
+
+        self.publish(
+            message_data=message_data,
+            delay_seconds=delay_seconds,
+            message_attributes=message_attributes,
+        )
+
+        return {
+            "posted": True,
+            "visible_count": visible_count,
+            "max_capacity": max_capacity,
+        }
+
 
 class MessageFilter:
     """
@@ -234,15 +308,6 @@ class HubspotSQSNotifier(SQSClient):
             logger=logger,
             topic_name=f"{env}-{self.TOPIC_NAME}",
         )
-        self.env = env
-
-    @cached_property
-    def lambda_client(self):
-        session = boto3.Session(
-            region_name=self.region_name,
-            profile_name=self.profile_name,
-        )
-        return session.client("lambda")
 
     def send_message(
         self, object_name, object_id, secondary_object_id=None, delay_seconds=60
@@ -283,30 +348,6 @@ class HubspotSQSNotifier(SQSClient):
         return delay_seconds
 
     def notify_business_change(self, business_id, delay_seconds=60):
-        try:
-            resp = requests.get(
-                "https://asoekaw.free.beeceptor.com/processing/eligibility",
-                timeout=10,
-            )
-
-            data = resp.json()
-
-            if data.get("allowed", False):
-                repeat = int(data.get("repeat", 1))
-                for _ in range(repeat):
-                    self.lambda_client.invoke(
-                        FunctionName=f"{self.env}-nightly-lambda",
-                        InvocationType="Event",
-                        Payload=json.dumps(
-                            {
-                                "event": "hourly_15",
-                                "override_status": True,
-                            }
-                        ).encode("utf-8"),
-                    )
-        except Exception:
-            pass
-
         return self.send_message(
             object_name="business_change",
             object_id=business_id,
@@ -538,7 +579,6 @@ class POSApiPullerSQSNotifier(SQSClient):
             logger=logger,
             topic_name=f"{env}-{self.TOPIC_NAME}",
         )
-        self.env = env
 
     def notify_for_send_message(
         self,