reach_commons 0.18.35__tar.gz → 0.18.36__tar.gz

{reach_commons-0.18.35 → reach_commons-0.18.36}/PKG-INFO

@@ -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 → reach_commons-0.18.36}/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.35 → reach_commons-0.18.36}/pyproject.toml

@@ -1,41 +1,41 @@
-# isort .; black .; poetry build; poetry publish
-[tool.poetry]
-name = "reach_commons"
-version = "0.18.35"
-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"]
+# isort .; black .; poetry build; poetry publish
+[tool.poetry]
+name = "reach_commons"
+version = "0.18.36"
+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.35 → reach_commons-0.18.36}/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-0.18.36/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-0.18.35 → reach_commons-0.18.36}/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))