introduce AccessToken dataclass (#128)

Author: kacperf531

changelog.md

@@ -10,6 +10,7 @@ All notable changes to this project will be documented in this file.
 - Updated outdated packages.
 - Enhanced error logging for improved troubleshooting: Automatically includes response headers in the log for server errors, providing detailed information (such as x-debug-id) for more effective issue diagnosis.
 - Enhanced timeouts for the RTM and WEB clients.
+- Introduced `AccessToken` structure which allows keeping token type in separate field. Previous way of passing tokens as a string of format `type: token` remains supported for backwards compatibility.
 
 ### Bugfixes
 - Enabled instantiation for `CustomerRtmV36` within the 3.6 version of the Customer RTM API.

examples/agent_rtm_example.py

@@ -1,10 +1,14 @@
 ''' Agent RTM client example usage. '''
 
 from livechat.agent.rtm.base import AgentRTM
+from livechat.utils.structures import AccessToken, TokenType
 
 agent_rtm = AgentRTM.get_client()
 agent_rtm.open_connection()
-agent_rtm.login(token='<your access token>')
+
+# token can be also passed as a raw string like `Bearer dal:A420qcNvdVS4cRMJP269GfgT1LA`
+agent_rtm.login(token=AccessToken(scheme=TokenType.BEARER,
+                                  token='dal:A420qcNvdVS4cRMJP269GfgT1LA'))
 response = agent_rtm.start_chat(continuous=True)
 chat_id = response.payload.get('chat_id')
 thread_id = response.payload.get('thread_id')

examples/agent_web_example.py

@@ -1,8 +1,11 @@
 ''' Agent WEB client example usage. '''
 
 from livechat.agent.web.base import AgentWeb
+from livechat.utils.structures import AccessToken, TokenType
 
-agent_web = AgentWeb.get_client(access_token='<your access token>')
+# token can be also passed as a raw string like `Bearer dal:A420qcNvdVS4cRMJP269GfgT1LA`
+agent_web = AgentWeb.get_client(access_token=AccessToken(
+    scheme=TokenType.BEARER, token='dal:A420qcNvdVS4cRMJP269GfgT1LA'))
 results = agent_web.start_chat(continuous=True)
 chat_id = results.json().get('chat_id')
 thread_id = results.json().get('thread_id')

examples/configuration_api_example.py

@@ -1,9 +1,11 @@
 ''' Configuration API client example usage. '''
 
 from livechat.configuration.base import ConfigurationApi
+from livechat.utils.structures import AccessToken, TokenType
 
 # Get list of existing groups.
-configuration_api = ConfigurationApi.get_client(token='<your access token>')
+configuration_api = ConfigurationApi.get_client(token=AccessToken(
+    scheme=TokenType.BEARER, token='dal:A420qcNvdVS4cRMJP269GfgT1LA'))
 groups = configuration_api.list_groups()
 print(groups.json())
 

examples/customer_rtm_example.py

@@ -1,11 +1,13 @@
 ''' Customer RTM client example usage. '''
 
 from livechat.customer.rtm.base import CustomerRTM
+from livechat.utils.structures import AccessToken, TokenType
 
 customer_rtm = CustomerRTM.get_client(
     organization_id='142cf3ad-5d54-4cf6-8ce1-3773d14d7f3f')
 customer_rtm.open_connection()
-customer_rtm.login(token='Bearer <your bearer token>')
+customer_rtm.login(token=AccessToken(scheme=TokenType.BEARER,
+                                     token='dal:A6420cNvdVS4cRMJP269GfgT1LA'))
 response = customer_rtm.start_chat(continuous=True)
 chat_id = response.payload.get('chat_id')
 thread_id = response.payload.get('thread_id')

examples/customer_web_example.py

@@ -1,10 +1,12 @@
 ''' Customer WEB client example usage. '''
 
 from livechat.customer.web.base import CustomerWeb
+from livechat.utils.structures import AccessToken, TokenType
 
 customer_web = CustomerWeb.get_client(
     organization_id='142cf3ad-5d54-4cf6-8ce1-3773d14d7f3f',
-    access_token='<your access token>')
+    access_token=AccessToken(scheme=TokenType.BEARER,
+                             token='dal:A6420cNvdVS4cRMJP269GfgT1LA'))
 results = customer_web.start_chat(continuous=True)
 chat_id = results.json().get('chat_id')
 thread_id = results.json().get('thread_id')

examples/reports_api_example.py

@@ -1,9 +1,11 @@
 ''' Reports API client example usage. '''
 
 from livechat.reports.base import ReportsApi
+from livechat.utils.structures import AccessToken, TokenType
 
 # Get number of chats occured during specified period.
-reports_api = ReportsApi.get_client(token='<your access token>')
+reports_api = ReportsApi.get_client(token=AccessToken(
+    scheme=TokenType.BEARER, token='dal:A420qcNvdVS4cRMJP269GfgT1LA'))
 chats_occured = reports_api.total_chats(filters={
     'to': '2020-09-14T23:59:59+02:00',
     'from': '2020-09-01T00:00:00+02:00'

livechat/agent/rtm/api/v33.py

@@ -1,9 +1,9 @@
 ''' Module containing Agent RTM API client implementation for v3.3. '''
 
-from typing import Any, Optional
+from typing import Any, Optional, Union
 
 from livechat.utils.helpers import prepare_payload
-from livechat.utils.structures import RtmResponse
+from livechat.utils.structures import AccessToken, RtmResponse
 from livechat.utils.ws_client import WebsocketClient
 
 # pylint: disable=unused-argument, too-many-arguments, invalid-name, redefined-builtin
@@ -789,7 +789,7 @@ def unfollow_customer(self,
 # Status
 
     def login(self,
-              token: str = None,
+              token: Union[AccessToken, str] = None,
               timezone: str = None,
               reconnect: bool = None,
               push_notifications: dict = None,
@@ -822,6 +822,8 @@ def login(self,
                 RtmResponse: RTM response structure (`request_id`, `action`,
                              `type`, `success` and `payload` properties)
         '''
+        if token:
+            token = str(token)
         if payload is None:
             payload = prepare_payload(locals())
         return self.ws.send({'action': 'login', 'payload': payload})

livechat/agent/rtm/api/v34.py

@@ -1,9 +1,9 @@
 ''' Module containing Agent RTM API client implementation for v3.4. '''
 
-from typing import Any, Optional
+from typing import Any, Optional, Union
 
 from livechat.utils.helpers import prepare_payload
-from livechat.utils.structures import RtmResponse
+from livechat.utils.structures import AccessToken, RtmResponse
 from livechat.utils.ws_client import WebsocketClient
 
 # pylint: disable=unused-argument, too-many-arguments, invalid-name, redefined-builtin
@@ -755,7 +755,7 @@ def unfollow_customer(self,
 # Status
 
     def login(self,
-              token: str = None,
+              token: Union[AccessToken, str] = None,
               timezone: str = None,
               reconnect: bool = None,
               push_notifications: dict = None,
@@ -788,6 +788,8 @@ def login(self,
                 RtmResponse: RTM response structure (`request_id`, `action`,
                              `type`, `success` and `payload` properties)
         '''
+        if token:
+            token = str(token)
         if payload is None:
             payload = prepare_payload(locals())
         return self.ws.send({'action': 'login', 'payload': payload})

livechat/agent/rtm/api/v35.py

@@ -1,9 +1,9 @@
 ''' Module containing Agent RTM API client implementation for v3.5. '''
 
-from typing import Any, Optional
+from typing import Any, Optional, Union
 
 from livechat.utils.helpers import prepare_payload
-from livechat.utils.structures import RtmResponse
+from livechat.utils.structures import AccessToken, RtmResponse
 from livechat.utils.ws_client import WebsocketClient
 
 # pylint: disable=unused-argument, too-many-arguments, invalid-name, redefined-builtin
@@ -755,7 +755,7 @@ def unfollow_customer(self,
 # Status
 
     def login(self,
-              token: str = None,
+              token: Union[AccessToken, str] = None,
               timezone: str = None,
               reconnect: bool = None,
               push_notifications: dict = None,
@@ -788,6 +788,8 @@ def login(self,
                 RtmResponse: RTM response structure (`request_id`, `action`,
                              `type`, `success` and `payload` properties)
         '''
+        if token:
+            token = str(token)
         if payload is None:
             payload = prepare_payload(locals())
         return self.ws.send({'action': 'login', 'payload': payload})