amochka 0.3.1__tar.gz → 0.3.2__tar.gz

{amochka-0.3.1 → amochka-0.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: amochka
-Version: 0.3.1
+Version: 0.3.2
 Summary: Python library for working with amoCRM API with ETL capabilities
 Author-email: Timur <timurdt@gmail.com>
 License: MIT
@@ -26,7 +26,6 @@ Classifier: Topic :: Internet :: WWW/HTTP
 Requires-Python: >=3.6
 Description-Content-Type: text/markdown
 Requires-Dist: requests>=2.25.0
-Requires-Dist: ratelimit>=2.2.0
 Requires-Dist: psycopg2-binary>=2.9.0
 Requires-Dist: python-dotenv>=1.0.0
 

{amochka-0.3.1 → amochka-0.3.2}/amochka/client.py

@@ -5,7 +5,7 @@ import requests
 import logging
 from datetime import datetime
 from typing import Iterator, List, Optional, Sequence, Union
-from ratelimit import limits, sleep_and_retry
+# ratelimit больше не используется - rate limiting реализован вручную через self.rate_limit
 
 # Создаём базовый логгер
 logger = logging.getLogger(__name__)
@@ -15,7 +15,7 @@ if not logger.handlers:
     ch.setFormatter(formatter)
     logger.addHandler(ch)
 
-RATE_LIMIT = 7  # Максимум 7 запросов в секунду
+DEFAULT_RATE_LIMIT = 7  # Максимум запросов в секунду по умолчанию
 
 class Deal(dict):
     """
@@ -262,12 +262,13 @@ class AmoCRMClient:
     либо полностью отключить логирование, установив disable_logging=True.
     """
     def __init__(
-        self, 
-        base_url, 
-        token_file=None, 
-        cache_config=None, 
-        log_level=logging.INFO, 
+        self,
+        base_url,
+        token_file=None,
+        cache_config=None,
+        log_level=logging.INFO,
         disable_logging=False,
+        rate_limit: Optional[int] = None,
         *,
         client_id: Optional[str] = None,
         client_secret: Optional[str] = None,
@@ -275,14 +276,17 @@ class AmoCRMClient:
     ):
         """
         Инициализирует клиента, задавая базовый URL, токен авторизации и настройки кэша для кастомных полей.
-        
+
         :param base_url: Базовый URL API amoCRM.
         :param token_file: Файл, содержащий токен авторизации.
         :param cache_config: Конфигурация кэширования (объект CacheConfig или None для значений по умолчанию)
         :param log_level: Уровень логирования (например, logging.DEBUG, logging.INFO).
         :param disable_logging: Если True, логирование будет отключено.
+        :param rate_limit: Максимальное количество запросов в секунду (по умолчанию 7).
         """
         self.base_url = base_url.rstrip('/')
+        self.rate_limit = rate_limit if rate_limit is not None else DEFAULT_RATE_LIMIT
+        self._request_times: List[float] = []  # Для отслеживания времени запросов
         domain = self.base_url.split("//")[-1].split(".")[0]
         self.domain = domain
         self.token_file = token_file or os.path.join(os.path.expanduser('~'), '.amocrm_token.json')
@@ -378,8 +382,24 @@ class AmoCRMClient:
 
         raise Exception("Токен истёк или некорректен, и нет данных для refresh_token. Обновите токен.")
 
-    @sleep_and_retry
-    @limits(calls=RATE_LIMIT, period=1)
+    def _wait_for_rate_limit(self):
+        """Ожидает, если превышен лимит запросов в секунду."""
+        now = time.time()
+        # Удаляем запросы старше 1 секунды
+        self._request_times = [t for t in self._request_times if now - t < 1.0]
+
+        if len(self._request_times) >= self.rate_limit:
+            # Ждём до освобождения слота
+            sleep_time = 1.0 - (now - self._request_times[0])
+            if sleep_time > 0:
+                self.logger.debug(f"Rate limit: ожидание {sleep_time:.3f}с (лимит {self.rate_limit} req/s)")
+                time.sleep(sleep_time)
+            # Очищаем старые записи после ожидания
+            now = time.time()
+            self._request_times = [t for t in self._request_times if now - t < 1.0]
+
+        self._request_times.append(time.time())
+
     def _make_request(self, method, endpoint, params=None, data=None, timeout=10):
         """
         Выполняет HTTP-запрос к API amoCRM с учетом ограничения по скорости (rate limit).
@@ -392,6 +412,9 @@ class AmoCRMClient:
         :return: Ответ в формате JSON или None (если статус 204).
         :raises Exception: При получении кода ошибки, отличного от 200/204.
         """
+        # Ручной rate limiting
+        self._wait_for_rate_limit()
+
         url = f"{self.base_url}{endpoint}"
         headers = {
             "Authorization": f"Bearer {self.token}",

{amochka-0.3.1 → amochka-0.3.2}/amochka.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: amochka
-Version: 0.3.1
+Version: 0.3.2
 Summary: Python library for working with amoCRM API with ETL capabilities
 Author-email: Timur <timurdt@gmail.com>
 License: MIT
@@ -26,7 +26,6 @@ Classifier: Topic :: Internet :: WWW/HTTP
 Requires-Python: >=3.6
 Description-Content-Type: text/markdown
 Requires-Dist: requests>=2.25.0
-Requires-Dist: ratelimit>=2.2.0
 Requires-Dist: psycopg2-binary>=2.9.0
 Requires-Dist: python-dotenv>=1.0.0
 

{amochka-0.3.1 → amochka-0.3.2}/amochka.egg-info/requires.txt

@@ -1,4 +1,3 @@
 requests>=2.25.0
-ratelimit>=2.2.0
 psycopg2-binary>=2.9.0
 python-dotenv>=1.0.0

{amochka-0.3.1 → amochka-0.3.2}/etl/config.py

@@ -107,6 +107,7 @@ class AmoCRMAccount:
     mybi_account_id: int  # Внутренний account_id как в mybi.ru (для совместимости)
     pipeline_ids: Optional[List[int]] = None  # None = все воронки
     cache_dir: Optional[Path] = None
+    rate_limit: int = 7  # Максимум запросов в секунду (по умолчанию 7)
 
     def __post_init__(self):
         if isinstance(self.token_path, str):
@@ -125,6 +126,7 @@ class AmoCRMAccount:
             mybi_account_id=int(d.get("mybi_account_id", 0)),
             pipeline_ids=d.get("pipeline_ids"),
             cache_dir=Path(d.get("cache_dir", ".cache")) if d.get("cache_dir") else None,
+            rate_limit=int(d.get("rate_limit", 7)),
         )
 
 

{amochka-0.3.1 → amochka-0.3.2}/etl/extractors.py

@@ -48,6 +48,7 @@ class AmoCRMExtractor:
                 base_url=self.account.base_url,
                 token_file=str(self.account.token_path),
                 cache_config=cache_config,
+                rate_limit=self.account.rate_limit,
             )
 
     @property

{amochka-0.3.1 → amochka-0.3.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "amochka"
-version = "0.3.1"
+version = "0.3.2"
 description = "Python library for working with amoCRM API with ETL capabilities"
 readme = "README.md"
 authors = [
@@ -31,7 +31,6 @@ keywords = ["amocrm", "crm", "api", "client", "automation", "etl"]
 requires-python = ">=3.6"
 dependencies = [
     "requests>=2.25.0",
-    "ratelimit>=2.2.0",
     "psycopg2-binary>=2.9.0",
     "python-dotenv>=1.0.0"
 ]