reach_commons 0.18.35__py3-none-any.whl → 0.18.36__py3-none-any.whl

reach_commons/app_logging/logging_utils.py

@@ -1,17 +1,17 @@
-import logging
-import os
-
-
-def init_logger(name: str):
-    logging.basicConfig(
-        level=getattr(logging, os.getenv("LOG_LEVEL", "INFO").upper(), logging.INFO),
-        format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
-        force=True,
-    )
-    for noisy in ("botocore", "boto3", "urllib3"):
-        logging.getLogger(noisy).setLevel(logging.WARNING)
-    return logging.getLogger(name)
-
-
-def log_with_event(level_fn, msg, event):
-    level_fn(f"{msg} | event={event}")
+import logging
+import os
+
+
+def init_logger(name: str):
+    logging.basicConfig(
+        level=getattr(logging, os.getenv("LOG_LEVEL", "INFO").upper(), logging.INFO),
+        format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
+        force=True,
+    )
+    for noisy in ("botocore", "boto3", "urllib3"):
+        logging.getLogger(noisy).setLevel(logging.WARNING)
+    return logging.getLogger(name)
+
+
+def log_with_event(level_fn, msg, event):
+    level_fn(f"{msg} | event={event}")

reach_commons/mongo/mongo_token_bucket.py

@@ -0,0 +1,108 @@
+# mongo_token_bucket.py
+import random
+import time
+from dataclasses import dataclass
+from typing import Optional, Tuple
+
+_LUA_WINDOW_LIMITER = """
+-- KEYS[1] = window key
+-- ARGV[1] = tokens_to_consume
+-- ARGV[2] = ttl_seconds
+-- ARGV[3] = limit
+
+local tokens = tonumber(ARGV[1])
+local ttl = tonumber(ARGV[2])
+local limit = tonumber(ARGV[3])
+
+local current = redis.call('INCRBY', KEYS[1], tokens)
+
+-- If this was the first increment (key was 0 / nonexistent), set TTL
+if current == tokens then
+  redis.call('EXPIRE', KEYS[1], ttl)
+end
+
+if current <= limit then
+  return 1
+else
+  return 0
+end
+"""
+
+
+@dataclass(frozen=True)
+class AcquireResult:
+    allowed: bool
+    retry_after_seconds: int  # for visibility timeout / delay
+
+
+class MongoTokenBucketManager:
+    """
+    Window-based limiter:
+      - limit tokens per interval_seconds
+      - atomic via Redis Lua
+    """
+
+    def __init__(
+        self,
+        redis_manager,
+        limit_per_window: int,
+        interval_seconds: int = 2,
+        bucket_key: str = "global",
+        jitter_seconds: Optional[int] = None,
+        key_prefix: str = "mongo_write_budget",
+    ):
+        if interval_seconds <= 0:
+            raise ValueError("interval_seconds must be > 0")
+        if limit_per_window <= 0:
+            raise ValueError("limit_per_window must be > 0")
+
+        self.redis = redis_manager
+        self.limit = int(limit_per_window)
+        self.interval = int(interval_seconds)
+        self.bucket_key = bucket_key
+        self.key_prefix = key_prefix
+        self.jitter = (
+            int(jitter_seconds) if jitter_seconds is not None else self.interval
+        )  # default: 0..interval
+
+        # Cache the script SHA if you want; eval is fine for now (2-day fix).
+        self._lua = _LUA_WINDOW_LIMITER
+
+    def _now(self) -> float:
+        return time.time()
+
+    def _window_start(self, now: float) -> int:
+        return int(now // self.interval) * self.interval
+
+    def _redis_key(self, window_start: int) -> str:
+        return f"{self.key_prefix}:{self.bucket_key}:{window_start}"
+
+    def acquire(self, tokens: int = 1) -> AcquireResult:
+        """
+        Try to consume tokens. If denied, returns retry_after_seconds to push visibility timeout.
+        """
+        now = self._now()
+        window_start = self._window_start(now)
+        window_end = window_start + self.interval
+
+        key = self._redis_key(window_start)
+
+        # TTL a bit bigger than the window so old keys go away safely
+        ttl_seconds = max(self.interval * 2, 5)
+
+        allowed = self.redis.eval(
+            self._lua,
+            numkeys=1,
+            keys=[key],
+            args=[str(int(tokens)), str(int(ttl_seconds)), str(int(self.limit))],
+        )
+
+        if allowed == 1:
+            return AcquireResult(allowed=True, retry_after_seconds=0)
+
+        # Denied: retry after next window, plus jitter to avoid waves
+        base = max(0.0, window_end - now)  # seconds until next window
+        jitter = random.uniform(0.0, float(self.jitter))
+        retry_after = int(max(1.0, base + jitter))  # at least 1s
+
+        return AcquireResult(allowed=False, retry_after_seconds=retry_after)

reach_commons/redis_manager.py

@@ -80,3 +80,14 @@ class RedisManager:
             name=key, value="1", ex=expire_seconds, nx=True
         )
         return result is True
+
+    def incrby(self, key: str, amount: int = 1) -> int:
+        return int(self.redis_connection.incrby(key, amount))
+
+    def expire(self, key: str, seconds: int) -> bool:
+        return bool(self.redis_connection.expire(key, seconds))
+
+    def eval(self, script: str, numkeys: int, keys=None, args=None):
+        keys = keys or []
+        args = args or []
+        return self.redis_connection.eval(script, numkeys, *(keys + args))

{reach_commons-0.18.35.dist-info → reach_commons-0.18.36.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.4
+Metadata-Version: 2.1
 Name: reach_commons
-Version: 0.18.35
+Version: 0.18.36
 Summary: Reach Commons is a versatile utility library designed to streamline and enhance development workflows within the Reach ecosystem.
 License: MIT
 Author: Engineering
@@ -14,8 +14,6 @@ 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.35.dist-info → reach_commons-0.18.36.dist-info}/RECORD

@@ -1,12 +1,10 @@
-reach_commons/.DS_Store,sha256=M7WbC5LcTvuTEuj3k7z3tqzMSVGgu87auq6UOLoSv2M,6148
 reach_commons/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 reach_commons/app_logging/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 reach_commons/app_logging/http_logger.py,sha256=mljQCdmsmtD2HsC_gsFwZAxPlAiLPYVirVrCjXFxitY,2541
 reach_commons/app_logging/log_deprecated_endpoints.py,sha256=yXs9Jh7V0_0cMnzwXV9WRgCdFXe_tybcFE1eQl2KNC4,2020
 reach_commons/app_logging/logger.py,sha256=Iq2XTl1zLgHDmVsTMdlFadcYJOqQNhBcFSscacKs_Xs,2295
 reach_commons/app_logging/logging_config.py,sha256=Y1JaZOoQBWgQjkOqYmeDRIm0p2eCOl3yTzgsgqyqm8I,1539
-reach_commons/app_logging/logging_utils.py,sha256=zeDRm3DugbuqMIDstTsVyH5xU_ZV-WAcCINxb6qYdJY,491
-reach_commons/clients/.DS_Store,sha256=1lFlJ5EFymdzGAUAaI30vcaaLHt3F1LwpG7xILf9jsM,6148
+reach_commons/app_logging/logging_utils.py,sha256=-QQ4l9CTYs-lCL-VNxGdWvPquRQtrnXWqzNfxOLO9ys,508
 reach_commons/clients/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 reach_commons/clients/event_processor.py,sha256=KmYF0kuZxLmHQjJASXMr5jz2D_D3WNHB0c4QOlZo1_E,2024
 reach_commons/clients/hubspot.py,sha256=ntAzvwoaq78MkKaVoZ7geND-AafAzccNnJogfJDahVA,5497
@@ -16,6 +14,7 @@ reach_commons/clients/reach_ops_api.py,sha256=1kTY4cF2eBCl_kqs1r5lbzxt4_7EFRmltb
 reach_commons/mongo/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 reach_commons/mongo/customer_persistence.py,sha256=acrtpyCWr9vLVq61saJ3_Vp4DYHFBTM9XqoYC72J84w,3735
 reach_commons/mongo/customer_persistence_async.py,sha256=BmcP8TXyyQah-GYM3wcKi1baqSCycjw7UadlxGywyQM,3892
+reach_commons/mongo/mongo_token_bucket.py,sha256=Q98EGoetrNhtAqlfvGk2tXex4EEl0d-npGXygld0gSQ,3272
 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
@@ -26,10 +25,10 @@ reach_commons/reach_aws/kms.py,sha256=ZOfyJMQUgxJEojRoB7-aCxtATpNx1Ig522IUYH11NZ
 reach_commons/reach_aws/s3.py,sha256=2MLlDNFx0SROJBpE_KjJefyrB7lMqTlrYuRhSZx4iKs,3945
 reach_commons/reach_aws/sqs.py,sha256=IKKWrd-qbhMMVYUvGbaq1ouVRdx-0u-SqwYaTcp0tWY,21645
 reach_commons/reach_base_model.py,sha256=vgdGDcZr3iXMmyRhmBhOf_LKWB_6QzT3r_Yiyo6OmEk,3009
-reach_commons/redis_manager.py,sha256=SgUtdtt0eV4bUwsWDankIa9Bjfgcm2DKcmVMQT6ptF0,2946
+reach_commons/redis_manager.py,sha256=RXtEksz13_uXwJAxfDvT8sCNwNIPCdqhsLUlewRq3Jg,3392
 reach_commons/sms_smart_encoding.py,sha256=92y0RmZ0l4ONHpC9qeO5KfViSNq64yE2rc7lhNDSZqE,1241
 reach_commons/utils.py,sha256=dMgKIGqTgoSItuBI8oz81gKtW3qi21Jkljv9leS_V88,8475
 reach_commons/validations.py,sha256=x_lkrtlrCAJC_f7mZb19JjfKFbYlPFv-P84K_lbZyYs,1056
-reach_commons-0.18.35.dist-info/METADATA,sha256=gD0kZHrRs6H0VzzuUYrxs4EFOju-ubt9hdnB4AdLMLM,1965
-reach_commons-0.18.35.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
-reach_commons-0.18.35.dist-info/RECORD,,
+reach_commons-0.18.36.dist-info/METADATA,sha256=cCN7jv1Iyn2GMKoYAkt15QalHcoN4pVWVwmXqKBESXg,1863
+reach_commons-0.18.36.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+reach_commons-0.18.36.dist-info/RECORD,,

{reach_commons-0.18.35.dist-info → reach_commons-0.18.36.dist-info}/WHEEL

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