amocrm-api-wrapper 0.0.20__py3-none-any.whl

amocrm_api/errors.py

@@ -0,0 +1,9 @@
+class AmoException(Exception):
+    def __init__(self, error_data: dict, code: int = 500, *args):
+        self.error_data = error_data
+        self.code = code
+        super().__init__(args)
+
+    def __str__(self):
+        error_message = ''.join(f'{k}: {v}' for k, v in self.error_data.items())
+        return f'Code: {self.code}, Detail: {error_message}'

amocrm_api/legacy_client.py

@@ -0,0 +1,58 @@
+from json import JSONDecodeError
+from requests import Session, ConnectionError, ConnectTimeout
+from typing import Optional, Union
+from urllib.parse import urlencode
+
+from .errors import AmoException
+from .base import BaseClient
+
+
+class AmoLegacyClient(BaseClient):
+    def __init__(self, login: str, token: str, crm_url: str) -> None:
+        """Init
+
+        Args:
+            login (str): login
+            token (str): seckter token
+            crm_url (str): your crm url like https://example.amocrm.ru
+        """
+        self.login = login
+        self.token = token
+        self.crm_url = crm_url if not crm_url.endswith('/') else crm_url[:-1]
+        self._session = self._init_session()
+
+    def _init_session(self, headers: Optional[dict] = None) -> Session:
+        """Init session
+
+        Args:
+            headers (Optional[dict], optional): headers params. Defaults to None.
+
+        Raises:
+            AmoException: if invalid auth data
+
+        Returns:
+            Session: session
+        """
+        url = f"{self.crm_url}/private/api/auth.php?type=json"
+        params = {"USER_LOGIN": self.login, "USER_HASH": self.token}
+        session = Session()
+        if isinstance(headers, dict):
+            session.headers.update(**headers)
+        auth_response = session.post(url, json=params).json()
+        if  auth_response.get("response", {}).get("auth", {}):
+            return session
+        raise AmoException(auth_response)
+
+    def _send_api_request(
+        self, method: str, url: str, data: Optional[dict] = None
+    ) -> dict:
+        try:
+            response = self._session.__getattribute__(method)(url, json=data)
+            if response.status_code == 204:
+                return {}
+            json_data = response.json()
+            if 'error' in json_data:
+                raise AmoException(json_data)
+            return json_data
+        except (ConnectTimeout, ConnectionError, JSONDecodeError) as e:
+            raise AmoException({'error': str(e)})

amocrm_api/oauth_client.py

@@ -0,0 +1,92 @@
+from json import JSONDecodeError
+from requests import Session, ConnectionError, ConnectTimeout, post
+from typing import Optional, Union
+from urllib.parse import urlencode
+
+from .errors import AmoException
+from .base import BaseClient
+
+
+class AmoOAuthClient(BaseClient):
+    def __init__(
+        self,
+        access_token: str,
+        refresh_token: str,
+        crm_url: str,
+        client_id: str,
+        client_secret: str,
+        redirect_uri: str,
+    ):
+        self._access_token = access_token
+        self._refresh_token = refresh_token
+        self.crm_url = crm_url if not crm_url.endswith('/') else crm_url[:-1]
+        self.client_id = client_id
+        self.client_secret = client_secret
+        self.redirect_uri = redirect_uri
+        self._session = self._init_session()
+
+    def _init_session(self, params: Optional[dict] = None) -> Session:
+        session = Session()
+        session.headers.update(
+            {
+                'Authorization': f'Bearer {self.access_token}',
+                'User-Agent': 'amocrm-api-client/1.0',
+            }
+        )
+        if params:
+            session.headers.update(params)
+        return session
+
+    def _send_api_request(
+        self,
+        method: str,
+        url: str,
+        data: Optional[dict] = None,
+        update_tokens: bool = False,
+    ) -> dict:
+        try:
+            response = super()._send_api_request(method, url, data)
+            return response
+        except AmoException as e:
+            if 'Jsonstatus: 401' in str(e) and not update_tokens:
+                self.update_tokens()
+                return self._send_api_request(method, url, data, True)
+            raise
+
+    def update_session_auth_headers(self):
+        self.update_session_params({'Authorization': f'Bearer {self.access_token}'})
+
+    def update_tokens(self):
+        url = f'{self.crm_url}/oauth2/access_token'
+        params = {
+            'client_id': self.client_id,
+            'client_secret': self.client_secret,
+            'grant_type': 'refresh_token',
+            'refresh_token': self.refresh_token,
+            'redirect_uri': self.redirect_uri,
+        }
+        r = post(url, json=params)
+        data = r.json()
+        if r.status_code > 204:
+            raise AmoException(data)
+        self._update_token_params(data['access_token'], data['refresh_token'])
+        self.update_session_auth_headers()
+
+    def _update_token_params(self, access_token: str, refresh_token: str):
+        self._access_token = access_token
+        self._refresh_token = refresh_token
+
+    @property
+    def access_token(self) -> str:
+        return self._access_token
+
+    @property
+    def refresh_token(self) -> str:
+        return self._refresh_token
+
+    @property
+    def tokens(self) -> dict:
+        return {
+            'access_token': self.access_token,
+            'refresh_token': self.refresh_token,
+        }