reach_commons 0.18.22__py3-none-any.whl → 0.18.24__py3-none-any.whl

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/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/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,

{reach_commons-0.18.22.dist-info → reach_commons-0.18.24.dist-info}/METADATA

@@ -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.dist-info → reach_commons-0.18.24.dist-info}/RECORD

@@ -16,17 +16,17 @@ reach_commons/mongo/customer_persistence_async.py,sha256=BmcP8TXyyQah-GYM3wcKi1b
 reach_commons/mongo/validation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 reach_commons/reach_aws/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 reach_commons/reach_aws/commons.py,sha256=qQba0li75BIpmyVc0sDVrrxbtYvDCedF6RmFD-V4MYQ,259
-reach_commons/reach_aws/dynamo_db.py,sha256=ndwLlaDwb96IlsnpAeeKPg-2DOjxe0JSM0y-zdlOc6E,7082
+reach_commons/reach_aws/dynamo_db.py,sha256=BL3QcKzx4uZic-Ui12tln_GMSKe297FdfyIzFPE7veE,7140
 reach_commons/reach_aws/exceptions.py,sha256=x0RL5ktNtzxg0KykhEVWReBq_dEtciK6B2vMs_s4C9k,915
 reach_commons/reach_aws/firehose.py,sha256=1xFKLWMv3bNo3PPW5gtaL6NqzUDyVil6B768slj2wbY,5674
-reach_commons/reach_aws/kms.py,sha256=-X4EJOh9IoncNn9C46X55mpSHPO5mR0B36lYLA2hhic,4670
+reach_commons/reach_aws/kms.py,sha256=ZOfyJMQUgxJEojRoB7-aCxtATpNx1Ig522IUYH11NZ4,4678
 reach_commons/reach_aws/s3.py,sha256=2MLlDNFx0SROJBpE_KjJefyrB7lMqTlrYuRhSZx4iKs,3945
-reach_commons/reach_aws/sqs.py,sha256=qH8wAg0T4FYg-zZWXjfCk06_w3UHbz7N-U7CGOwqDoE,17000
+reach_commons/reach_aws/sqs.py,sha256=9_sVM7bqhswBSksJ2bhPvtDmW1N9jHh1OUC25vXYLdU,18563
 reach_commons/reach_base_model.py,sha256=vgdGDcZr3iXMmyRhmBhOf_LKWB_6QzT3r_Yiyo6OmEk,3009
 reach_commons/redis_manager.py,sha256=SgUtdtt0eV4bUwsWDankIa9Bjfgcm2DKcmVMQT6ptF0,2946
 reach_commons/sms_smart_encoding.py,sha256=92y0RmZ0l4ONHpC9qeO5KfViSNq64yE2rc7lhNDSZqE,1241
 reach_commons/utils.py,sha256=dMgKIGqTgoSItuBI8oz81gKtW3qi21Jkljv9leS_V88,8475
 reach_commons/validations.py,sha256=CJdWg8qc9ILe3rYHYTLdr2upRspsM_8ghhm4IXX4Pg8,1028
-reach_commons-0.18.22.dist-info/METADATA,sha256=1OUoKoNdTE0sLAh8aIUA5Y_eCRzOgdKRdnrWejqQg6E,1861
-reach_commons-0.18.22.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-reach_commons-0.18.22.dist-info/RECORD,,
+reach_commons-0.18.24.dist-info/METADATA,sha256=udH-_sPBf_Cvuyx34M5z16tOxlnu_eLdiMpdONMTmag,1965
+reach_commons-0.18.24.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+reach_commons-0.18.24.dist-info/RECORD,,

{reach_commons-0.18.22.dist-info → reach_commons-0.18.24.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 1.8.1
+Generator: poetry-core 2.2.1
 Root-Is-Purelib: true
 Tag: py3-none-any