amochka 0.1.7__tar.gz → 0.1.8__tar.gz

{amochka-0.1.7 → amochka-0.1.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: amochka
-Version: 0.1.7
+Version: 0.1.8
 Summary: Библиотека для работы с API amoCRM
 Home-page: 
 Author: Timurka

{amochka-0.1.7 → amochka-0.1.8}/amochka/amochka/__init__.py

@@ -2,7 +2,7 @@
 amochka: Библиотека для работы с API amoCRM.
 """
 
-__version__ = "0.1.7"
+__version__ = "0.1.8"
 
 from .client import AmoCRMClient, CacheConfig
 from .etl import (

{amochka-0.1.7 → amochka-0.1.8}/amochka/amochka/client.py

@@ -223,7 +223,11 @@ class AmoCRMClient:
         token_file=None, 
         cache_config=None, 
         log_level=logging.INFO, 
-        disable_logging=False
+        disable_logging=False,
+        *,
+        client_id: Optional[str] = None,
+        client_secret: Optional[str] = None,
+        redirect_uri: Optional[str] = None,
     ):
         """
         Инициализирует клиента, задавая базовый URL, токен авторизации и настройки кэша для кастомных полей.
@@ -238,6 +242,11 @@ class AmoCRMClient:
         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')
+
+        # OAuth2 credentials (используются для авто‑refresh токена)
+        self.client_id = client_id
+        self.client_secret = client_secret
+        self.redirect_uri = redirect_uri
         
         # Создаем логгер для конкретного экземпляра клиента
         self.logger = logging.getLogger(f"{__name__}.{self.domain}")
@@ -266,15 +275,19 @@ class AmoCRMClient:
         
         self.logger.debug(f"AmoCRMClient initialized for domain {self.domain}")
         
-        self.token = self.load_token()
+        self.token = None
+        self.refresh_token = None
+        self.expires_at = None
+        self.load_token()
         self._custom_fields_mapping = None
 
     def load_token(self):
         """
         Загружает токен авторизации из файла или строки, проверяет его срок действия.
+        При наличии refresh_token и учётных данных пробует обновить токен.
 
         :return: Действительный access_token.
-        :raises Exception: Если токен не найден или истёк.
+        :raises Exception: Если токен не найден или истёк и нет возможности обновить.
         """
         data = None
         if os.path.exists(self.token_file):
@@ -288,17 +301,34 @@ class AmoCRMClient:
             except Exception as e:
                 raise Exception("Токен не найден и не удалось распарсить переданное содержимое.") from e
 
+        self.refresh_token = data.get('refresh_token', self.refresh_token)
+        self.client_id = data.get('client_id', self.client_id)
+        self.client_secret = data.get('client_secret', self.client_secret)
+        self.redirect_uri = data.get('redirect_uri', self.redirect_uri)
+
         expires_at_str = data.get('expires_at')
-        try:
-            expires_at = datetime.fromisoformat(expires_at_str).timestamp()
-        except Exception:
-            expires_at = float(expires_at_str)
-        
-        if expires_at and time.time() < expires_at:
+        expires_at = None
+        if expires_at_str:
+            try:
+                expires_at = datetime.fromisoformat(expires_at_str).timestamp()
+            except Exception:
+                try:
+                    expires_at = float(expires_at_str)
+                except Exception:
+                    expires_at = None
+        self.expires_at = expires_at
+
+        access_token = data.get('access_token')
+        if access_token and expires_at and time.time() < expires_at:
             self.logger.debug("Token is valid.")
-            return data.get('access_token')
-        else:
-            raise Exception("Токен найден, но он истёк. Обновите токен.")
+            self.token = access_token
+            return access_token
+
+        if self.refresh_token and self.client_id and self.client_secret and self.redirect_uri:
+            self.logger.info("Access token истёк, пробую обновить через refresh_token…")
+            return self._refresh_access_token()
+
+        raise Exception("Токен истёк или некорректен, и нет данных для refresh_token. Обновите токен.")
 
     @sleep_and_retry
     @limits(calls=RATE_LIMIT, period=1)
@@ -321,6 +351,13 @@ class AmoCRMClient:
         }
         self.logger.debug(f"Making {method} request to {url} with params {params} and data {data}")
         response = requests.request(method, url, headers=headers, params=params, json=data, timeout=timeout)
+
+        if response.status_code == 401 and self.refresh_token:
+            self.logger.info("Получен 401, пробую обновить токен и повторить запрос…")
+            self._refresh_access_token()
+            headers["Authorization"] = f"Bearer {self.token}"
+            response = requests.request(method, url, headers=headers, params=params, json=data, timeout=timeout)
+
         if response.status_code not in (200, 204):
             self.logger.error(f"Request error {response.status_code}: {response.text}")
             raise Exception(f"Ошибка запроса: {response.status_code}, {response.text}")
@@ -328,6 +365,57 @@ class AmoCRMClient:
             return None
         return response.json()
 
+    def _refresh_access_token(self):
+        """Обновляет access_token по refresh_token и сохраняет его в token_file."""
+        if not all([self.refresh_token, self.client_id, self.client_secret, self.redirect_uri]):
+            raise Exception("Нельзя обновить токен: отсутствует refresh_token или client_id/client_secret/redirect_uri")
+
+        payload = {
+            "client_id": self.client_id,
+            "client_secret": self.client_secret,
+            "grant_type": "refresh_token",
+            "refresh_token": self.refresh_token,
+            "redirect_uri": self.redirect_uri,
+        }
+        token_url = f"{self.base_url}/oauth2/access_token"
+        self.logger.debug(f"Refreshing token via {token_url}")
+        resp = requests.post(token_url, json=payload, timeout=10)
+        if resp.status_code != 200:
+            self.logger.error(f"Не удалось обновить токен: {resp.status_code} {resp.text}")
+            raise Exception(f"Не удалось обновить токен: {resp.status_code}")
+
+        data = resp.json() or {}
+        access_token = data.get("access_token")
+        refresh_token = data.get("refresh_token", self.refresh_token)
+        expires_in = data.get("expires_in")
+        if not access_token:
+            raise Exception("Ответ на refresh не содержит access_token")
+
+        expires_at = None
+        if expires_in:
+            expires_at = time.time() + int(expires_in)
+
+        self.token = access_token
+        self.refresh_token = refresh_token
+        self.expires_at = expires_at
+
+        if self.token_file:
+            try:
+                with open(self.token_file, "w") as f:
+                    json.dump({
+                        "access_token": access_token,
+                        "refresh_token": refresh_token,
+                        "expires_at": datetime.fromtimestamp(expires_at).isoformat() if expires_at else None,
+                        "client_id": self.client_id,
+                        "client_secret": self.client_secret,
+                        "redirect_uri": self.redirect_uri,
+                    }, f)
+                self.logger.debug(f"Новый токен сохранён в {self.token_file}")
+            except Exception as exc:
+                self.logger.error(f"Не удалось сохранить обновлённый токен: {exc}")
+
+        return access_token
+
     def _to_timestamp(self, value: Optional[Union[int, float, str, datetime]]) -> Optional[int]:
         """
         Преобразует значение даты/времени в Unix timestamp.

{amochka-0.1.7 → amochka-0.1.8}/amochka/amochka.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: amochka
-Version: 0.1.7
+Version: 0.1.8
 Summary: Библиотека для работы с API amoCRM
 Home-page: 
 Author: Timurka

{amochka-0.1.7 → amochka-0.1.8}/setup.py

@@ -5,7 +5,7 @@ README = (Path(__file__).parent / "README.md").read_text(encoding="utf-8")
 
 setup(
     name='amochka',
-    version='0.1.7',
+    version='0.1.8',
     package_dir={"": "amochka"},
     packages=find_packages(where="amochka"),
     install_requires=[