symphony_bdk_python 2.10.dev0__py3-none-any.whl → 2.11.0__py3-none-any.whl

symphony/bdk/core/service/user/user_util.py

@@ -4,6 +4,7 @@ The tenant ID is stored in the 27 highest bits (minus the sign bit which is unus
 positive value) which allows for 134 million pods.
 This leaves 36 lowest bits for the user ID, which allows 68.7 billion users per tenant.
 """
+
 TENANT_ID_BIT_LENGTH = 27
 SUBTENANT_ID_BIT_LENGTH = 36
 TENANT_ID_INDEX = 1
@@ -20,8 +21,7 @@ def extract_tenant_id(user_id: int):
 
 
 class NumberUtil:
-    """Used to compute the segments bases in the input sizes
-    """
+    """Used to compute the segments bases in the input sizes"""
 
     def __init__(self, sizes: []):
         self._segments = []
@@ -38,15 +38,13 @@ class NumberUtil:
             raise ValueError("total size is larger than the bit-count of 64")
 
     def extract(self, value: int, index: int):
-        """Extract the tenant_id given the user_id and the index
-        """
+        """Extract the tenant_id given the user_id and the index"""
         segment = self._segments[index]
         return value >> segment._shift & segment._mask
 
 
 class Segment:
-    """Helper class to initialize a Segment
-    """
+    """Helper class to initialize a Segment"""
 
     def __init__(self, size: int, shift: int):
         mask = 0

symphony/bdk/core/service/version/agent_version_service.py

@@ -0,0 +1,62 @@
+import re
+from datetime import datetime, timezone
+
+from symphony.bdk.core.auth.jwt_helper import generate_expiration_time
+from symphony.bdk.core.config.model.bdk_retry_config import BdkRetryConfig
+from symphony.bdk.core.retry import retry
+from symphony.bdk.gen.agent_api.signals_api import SignalsApi
+from symphony.bdk.gen.exceptions import ApiException
+
+MIN_MAJOR_VERSION = 24
+MIN_MINOR_VERSION = 12
+VERSION_REGEXP = r"Agent-(\d+)\.(\d+)\..*"
+
+
+class AgentVersionService:
+    """Service class has one purpose only. It checks if version of agents supports simplified key delivery mechanism"""
+
+    def __init__(self, signals_api: SignalsApi, retry_config: BdkRetryConfig):
+        self._signals_api = signals_api
+        self._retry_config = retry_config
+        self._is_skd_supported = None
+        self._expire_at = -1
+
+    async def is_skd_supported(self) -> bool:
+        """AgentVersionService stores cached version  flag.
+        Caching interval is the same as in to session token caching.
+        Once cache is expired it calls agent info api to update version.
+
+        :return: boolean flag if skd supported for agent
+        """
+        if (
+            self._is_skd_supported is not None
+            and self._expire_at > datetime.now(timezone.utc).timestamp()
+        ):
+            return self._is_skd_supported
+        self._expire_at = generate_expiration_time()
+        self._is_skd_supported = await self._get_agent_skd_support()
+        return self._is_skd_supported
+
+    @retry
+    async def _get_agent_skd_support(self) -> bool:
+        try:
+            agent_info = await self._signals_api.v1_info_get()
+            if not agent_info or not agent_info.version:
+                return False
+        except ApiException:
+            return False
+        agent_major_version, agent_minor_version = self._parse_version(agent_info.version)
+        if not agent_major_version:
+            return False
+        if agent_major_version == MIN_MAJOR_VERSION:
+            return agent_minor_version >= MIN_MINOR_VERSION
+        return agent_major_version > MIN_MAJOR_VERSION
+
+    @staticmethod
+    def _parse_version(version_string):
+        if not version_string:
+            return None, None
+        match = re.match(VERSION_REGEXP, version_string)
+        if match:
+            return int(match.group(1)), int(match.group(2))
+        return None, None

symphony/bdk/core/service_factory.py

@@ -1,23 +1,32 @@
 """Service factory module"""
+
 from symphony.bdk.core.auth.auth_session import AuthSession
 from symphony.bdk.core.client.api_client_factory import ApiClientFactory
 from symphony.bdk.core.config.model.bdk_config import BdkConfig
 from symphony.bdk.core.service.application.application_service import ApplicationService
-from symphony.bdk.core.service.connection.connection_service import ConnectionService, OboConnectionService
-from symphony.bdk.core.service.datafeed.abstract_datafeed_loop import DatafeedVersion, AbstractDatafeedLoop
-from symphony.bdk.core.service.datafeed.abstract_ackId_event_loop import AbstractAckIdEventLoop
+from symphony.bdk.core.service.connection.connection_service import (
+    ConnectionService,
+    OboConnectionService,
+)
+from symphony.bdk.core.service.datafeed.abstract_datafeed_loop import (
+    AbstractDatafeedLoop,
+    DatafeedVersion,
+)
 from symphony.bdk.core.service.datafeed.abstract_datahose_loop import AbstractDatahoseLoop
 from symphony.bdk.core.service.datafeed.datafeed_loop_v1 import DatafeedLoopV1
 from symphony.bdk.core.service.datafeed.datafeed_loop_v2 import DatafeedLoopV2
 from symphony.bdk.core.service.datafeed.datahose_loop import DatahoseLoop
 from symphony.bdk.core.service.health.health_service import HealthService
 from symphony.bdk.core.service.message.message_service import MessageService, OboMessageService
-from symphony.bdk.core.service.message.multi_attachments_messages_api import MultiAttachmentsMessagesApi
-from symphony.bdk.core.service.presence.presence_service import PresenceService, OboPresenceService
+from symphony.bdk.core.service.message.multi_attachments_messages_api import (
+    MultiAttachmentsMessagesApi,
+)
+from symphony.bdk.core.service.presence.presence_service import OboPresenceService, PresenceService
 from symphony.bdk.core.service.session.session_service import SessionService
-from symphony.bdk.core.service.signal.signal_service import SignalService, OboSignalService
-from symphony.bdk.core.service.stream.stream_service import StreamService, OboStreamService
-from symphony.bdk.core.service.user.user_service import UserService, OboUserService
+from symphony.bdk.core.service.signal.signal_service import OboSignalService, SignalService
+from symphony.bdk.core.service.stream.stream_service import OboStreamService, StreamService
+from symphony.bdk.core.service.user.user_service import OboUserService, UserService
+from symphony.bdk.core.service.version.agent_version_service import AgentVersionService
 from symphony.bdk.gen.agent_api.attachments_api import AttachmentsApi
 from symphony.bdk.gen.agent_api.audit_trail_api import AuditTrailApi
 from symphony.bdk.gen.agent_api.datafeed_api import DatafeedApi
@@ -52,16 +61,15 @@ class ServiceFactory:
     """
 
     def __init__(
-            self,
-            api_client_factory: ApiClientFactory,
-            auth_session: AuthSession,
-            config: BdkConfig
+        self, api_client_factory: ApiClientFactory, auth_session: AuthSession, config: BdkConfig
     ):
         self._pod_client = api_client_factory.get_pod_client()
         self._agent_client = api_client_factory.get_agent_client()
         self._auth_session = auth_session
         self._config = config
-        self._session_service = SessionService(SessionApi(self._pod_client), self._auth_session, self._config.retry)
+        self._session_service = SessionService(
+            SessionApi(self._pod_client), self._auth_session, self._config.retry
+        )
 
     def get_user_service(self) -> UserService:
         """Returns a fully initialized UserService
@@ -74,7 +82,8 @@ class ServiceFactory:
             AuditTrailApi(self._agent_client),
             PodSystemApi(self._pod_client),
             self._auth_session,
-            self._config.retry
+            self._config.retry,
+            self._config.manifest,
         )
 
     def get_message_service(self) -> MessageService:
@@ -91,7 +100,7 @@ class ServiceFactory:
             AttachmentsApi(self._agent_client),
             DefaultApi(self._pod_client),
             self._auth_session,
-            self._config.retry
+            self._config.retry,
         )
 
     def get_connection_service(self) -> ConnectionService:
@@ -100,9 +109,7 @@ class ServiceFactory:
         :return: a new ConnectionService instance.
         """
         return ConnectionService(
-            ConnectionApi(self._pod_client),
-            self._auth_session,
-            self._config.retry
+            ConnectionApi(self._pod_client), self._auth_session, self._config.retry
         )
 
     def get_stream_service(self) -> StreamService:
@@ -115,7 +122,8 @@ class ServiceFactory:
             RoomMembershipApi(self._pod_client),
             ShareApi(self._agent_client),
             self._auth_session,
-            self._config.retry)
+            self._config.retry,
+        )
 
     def get_application_service(self) -> ApplicationService:
         """Returns a fully initialized ApplicationService
@@ -126,7 +134,7 @@ class ServiceFactory:
             ApplicationApi(self._pod_client),
             AppEntitlementApi(self._pod_client),
             self._auth_session,
-            self._config.retry
+            self._config.retry,
         )
 
     def get_signal_service(self) -> SignalService:
@@ -134,11 +142,7 @@ class ServiceFactory:
 
         :return: a new SignalService instance
         """
-        return SignalService(
-            SignalsApi(self._agent_client),
-            self._auth_session,
-            self._config.retry
-        )
+        return SignalService(SignalsApi(self._agent_client), self._auth_session, self._config.retry)
 
     def get_session_service(self) -> SessionService:
         """Returns a fully initialized SessionService
@@ -158,13 +162,10 @@ class ServiceFactory:
                 DatafeedApi(self._agent_client),
                 self._session_service,
                 self._auth_session,
-                self._config
+                self._config,
             )
         return DatafeedLoopV2(
-            DatafeedApi(self._agent_client),
-            self._session_service,
-            self._auth_session,
-            self._config
+            DatafeedApi(self._agent_client), self._session_service, self._auth_session, self._config
         )
 
     def get_datahose_loop(self) -> AbstractDatahoseLoop:
@@ -177,7 +178,7 @@ class ServiceFactory:
                 DatafeedApi(self._agent_client),
                 self._session_service,
                 self._auth_session,
-                self._config
+                self._config,
             )
 
     def get_health_service(self) -> HealthService:
@@ -186,9 +187,7 @@ class ServiceFactory:
         :return: a new HealthService instance
         """
         return HealthService(
-            AgentSystemApi(self._agent_client),
-            SignalsApi(self._agent_client),
-            self._config.retry
+            AgentSystemApi(self._agent_client), SignalsApi(self._agent_client), self._config.retry
         )
 
     def get_presence_service(self) -> PresenceService:
@@ -197,11 +196,16 @@ class ServiceFactory:
         :return: a new PresenceService instance
         """
         return PresenceService(
-            PresenceApi(self._pod_client),
-            self._auth_session,
-            self._config.retry
+            PresenceApi(self._pod_client), self._auth_session, self._config.retry
         )
 
+    def get_agent_version_service(self) -> AgentVersionService:
+        """Returns a fully initialized AgentVersionService
+
+        :return: a new AgentVersionService instance
+        """
+        return AgentVersionService(SignalsApi(self._agent_client), self._config.retry)
+
 
 class OboServiceFactory:
     """Factory responsible for creating BDK service instances for OBO-enabled endpoints only:
@@ -215,10 +219,7 @@ class OboServiceFactory:
     """
 
     def __init__(
-            self,
-            api_client_factory: ApiClientFactory,
-            auth_session: AuthSession,
-            config: BdkConfig
+        self, api_client_factory: ApiClientFactory, auth_session: AuthSession, config: BdkConfig
     ):
         self._pod_client = api_client_factory.get_pod_client()
         self._agent_client = api_client_factory.get_agent_client()
@@ -234,7 +235,7 @@ class OboServiceFactory:
             UserApi(self._pod_client),
             UsersApi(self._pod_client),
             self._auth_session,
-            self._config.retry
+            self._config.retry,
         )
 
     def get_message_service(self) -> OboMessageService:
@@ -245,8 +246,9 @@ class OboServiceFactory:
         return OboMessageService(
             MultiAttachmentsMessagesApi(self._agent_client),
             MessageSuppressionApi(self._pod_client),
+            self._pod_client,
             self._auth_session,
-            self._config.retry
+            self._config.retry,
         )
 
     def get_connection_service(self) -> OboConnectionService:
@@ -255,9 +257,7 @@ class OboServiceFactory:
         :return: a new OboConnectionService instance.
         """
         return OboConnectionService(
-            ConnectionApi(self._pod_client),
-            self._auth_session,
-            self._config.retry
+            ConnectionApi(self._pod_client), self._auth_session, self._config.retry
         )
 
     def get_stream_service(self) -> OboStreamService:
@@ -270,7 +270,8 @@ class OboServiceFactory:
             RoomMembershipApi(self._pod_client),
             ShareApi(self._agent_client),
             self._auth_session,
-            self._config.retry)
+            self._config.retry,
+        )
 
     def get_presence_service(self) -> OboPresenceService:
         """Returns a fully initialized OboPresenceService
@@ -278,9 +279,7 @@ class OboServiceFactory:
         :return: a new OboPresenceService instance
         """
         return OboPresenceService(
-            PresenceApi(self._pod_client),
-            self._auth_session,
-            self._config.retry
+            PresenceApi(self._pod_client), self._auth_session, self._config.retry
         )
 
     def get_signal_service(self) -> OboSignalService:
@@ -289,7 +288,5 @@ class OboServiceFactory:
         :return: a new OboSignalService instance
         """
         return OboSignalService(
-            SignalsApi(self._agent_client),
-            self._auth_session,
-            self._config.retry
+            SignalsApi(self._agent_client), self._auth_session, self._config.retry
         )

symphony/bdk/core/symphony_bdk.py

@@ -1,5 +1,5 @@
-"""Module containing SymphonyBdk class which acts as en entry point for all BDK services.
-"""
+"""Module containing SymphonyBdk class which acts as en entry point for all BDK services."""
+
 import functools
 import logging
 
@@ -9,7 +9,7 @@ from symphony.bdk.core.auth.authenticator_factory import AuthenticatorFactory
 from symphony.bdk.core.auth.exception import AuthInitializationError
 from symphony.bdk.core.auth.ext_app_authenticator import ExtensionAppAuthenticator
 from symphony.bdk.core.client.api_client_factory import ApiClientFactory
-from symphony.bdk.core.config.exception import BotNotConfiguredError, BdkConfigError
+from symphony.bdk.core.config.exception import BdkConfigError, BotNotConfiguredError
 from symphony.bdk.core.extension import ExtensionService
 from symphony.bdk.core.service.application.application_service import ApplicationService
 from symphony.bdk.core.service.connection.connection_service import ConnectionService
@@ -55,6 +55,7 @@ def app_service(func):
     :return: the value returned by the decorated function with the passed arguments.
     :raise: BdkConfigError if the app is not configured.
     """
+
     @functools.wraps(func)
     def check_if_app_configured_and_call_function(*args, **kwds):
         symphony_bdk = args[0]
@@ -66,8 +67,7 @@ def app_service(func):
 
 
 class SymphonyBdk:
-    """BDK entry point
-    """
+    """BDK entry point"""
 
     async def __aenter__(self):
         return self
@@ -80,7 +80,9 @@ class SymphonyBdk:
         self._api_client_factory = ApiClientFactory(config)
         self._pod_client = self._api_client_factory.get_pod_client()
         self._agent_client = self._api_client_factory.get_agent_client()
-        self._authenticator_factory = AuthenticatorFactory(config, api_client_factory=self._api_client_factory)
+        self._authenticator_factory = AuthenticatorFactory(
+            config, api_client_factory=self._api_client_factory
+        )
 
         self._bot_session = None
         self._ext_app_authenticator = None
@@ -101,12 +103,17 @@ class SymphonyBdk:
         if self._config.bot.is_authentication_configured():
             self._initialize_bot_services()
         else:
-            logging.info("Bot (service account) credentials have not been configured."
-                         "You can however use services in OBO mode if app authentication is configured.")
+            logging.info(
+                "Bot (service account) credentials have not been configured."
+                "You can however use services in OBO mode if app authentication is configured."
+            )
 
     def _initialize_bot_services(self):
-        self._bot_session = AuthSession(self._authenticator_factory.get_bot_authenticator())
-        self._service_factory = ServiceFactory(self._api_client_factory, self._bot_session, self._config)
+        bot_authenticator = self._authenticator_factory.get_bot_authenticator()
+        self._bot_session = AuthSession(bot_authenticator)
+        self._service_factory = ServiceFactory(
+            self._api_client_factory, self._bot_session, self._config
+        )
         self._user_service = self._service_factory.get_user_service()
         self._message_service = self._service_factory.get_message_service()
         self._connection_service = self._service_factory.get_connection_service()
@@ -122,7 +129,10 @@ class SymphonyBdk:
         self._activity_registry = ActivityRegistry(self._session_service)
         self._datafeed_loop.subscribe(self._activity_registry)
         # initialises extension service and register decorated extensions
-        self._extension_service = ExtensionService(self._api_client_factory, self._bot_session, self._config)
+        self._extension_service = ExtensionService(
+            self._api_client_factory, self._bot_session, self._config
+        )
+        bot_authenticator.agent_version_service = self._service_factory.get_agent_version_service()
 
     @bot_service
     def bot_session(self) -> AuthSession:
@@ -140,7 +150,9 @@ class SymphonyBdk:
         :return: a new instance of :class:`symphony.bdk.core.auth.ext_app_authenticator.ExtensionAppAuthenticator`
         """
         if self._ext_app_authenticator is None:
-            self._ext_app_authenticator = self._authenticator_factory.get_extension_app_authenticator()
+            self._ext_app_authenticator = (
+                self._authenticator_factory.get_extension_app_authenticator()
+            )
         return self._ext_app_authenticator
 
     @app_service
@@ -150,11 +162,16 @@ class SymphonyBdk:
         :return: The obo authentication session
         """
         if user_id is not None:
-            return self._authenticator_factory.get_obo_authenticator().authenticate_by_user_id(user_id)
+            return self._authenticator_factory.get_obo_authenticator().authenticate_by_user_id(
+                user_id
+            )
         if username is not None:
-            return self._authenticator_factory.get_obo_authenticator().authenticate_by_username(username)
-        raise AuthInitializationError("At least user_id or username should be given to OBO authenticate the "
-                                      "extension app")
+            return self._authenticator_factory.get_obo_authenticator().authenticate_by_username(
+                username
+            )
+        raise AuthInitializationError(
+            "At least user_id or username should be given to OBO authenticate the extension app"
+        )
 
     @app_service
     def obo_services(self, obo_session: OboAuthSession) -> OboServices:
@@ -270,6 +287,5 @@ class SymphonyBdk:
         return self._extension_service
 
     async def close_clients(self):
-        """Close all the existing api clients created by the api client factory.
-        """
+        """Close all the existing api clients created by the api client factory."""
         await self._api_client_factory.close_clients()

symphony/bdk/ext/group.py

@@ -1,10 +1,14 @@
 from typing import AsyncGenerator
 
-from symphony.bdk.core.extension import BdkAuthenticationAware, BdkApiClientFactoryAware, BdkExtensionServiceProvider, \
-    BdkConfigAware
+from symphony.bdk.core.extension import (
+    BdkApiClientFactoryAware,
+    BdkAuthenticationAware,
+    BdkConfigAware,
+    BdkExtensionServiceProvider,
+)
 from symphony.bdk.core.retry import retry
 from symphony.bdk.core.retry.strategy import is_network_or_minor_error, is_unauthorized
-from symphony.bdk.core.service.pagination import offset_based_pagination, cursor_based_pagination
+from symphony.bdk.core.service.pagination import cursor_based_pagination
 from symphony.bdk.core.service.user.user_util import extract_tenant_id
 from symphony.bdk.gen.group_api.group_api import GroupApi
 from symphony.bdk.gen.group_model.add_member import AddMember
@@ -34,10 +38,10 @@ async def refresh_bearer_token_if_unauthorized(retry_state):
     return False
 
 
-class SymphonyGroupBdkExtension(BdkAuthenticationAware, BdkApiClientFactoryAware, BdkConfigAware,
-                                BdkExtensionServiceProvider):
-    """Extension for Symphony Groups APIs
-    """
+class SymphonyGroupBdkExtension(
+    BdkAuthenticationAware, BdkApiClientFactoryAware, BdkConfigAware, BdkExtensionServiceProvider
+):
+    """Extension for Symphony Groups APIs"""
 
     def __init__(self):
         self._api_client_factory = None
@@ -58,15 +62,15 @@ class SymphonyGroupBdkExtension(BdkAuthenticationAware, BdkApiClientFactoryAware
 
 
 class SymphonyGroupService:
-    """Service class for managing Symphony Groups
-    """
+    """Service class for managing Symphony Groups"""
 
     def __init__(self, api_client_factory, auth_session, retry_config):
         self._api_client_factory = api_client_factory
         self._auth_session = auth_session
         self._retry_config = retry_config
-        self._oauth_session = OAuthSession(self._api_client_factory.get_login_client(), self._auth_session,
-                                           self._retry_config)
+        self._oauth_session = OAuthSession(
+            self._api_client_factory.get_login_client(), self._auth_session, self._retry_config
+        )
         client = self._api_client_factory.get_client("/profile-manager")
         client.configuration.auth_settings = self._oauth_session.get_auth_settings
 
@@ -85,8 +89,14 @@ class SymphonyGroupService:
         return await self._group_api.insert_group(x_symphony_host="", create_group=create_group)
 
     @retry(retry=refresh_bearer_token_if_unauthorized)
-    async def list_groups(self, status: Status = None, before: str = None, after: str = None, limit: int = None,
-                          sort_order: SortOrder = None) -> GroupList:
+    async def list_groups(
+        self,
+        status: Status = None,
+        before: str = None,
+        after: str = None,
+        limit: int = None,
+        sort_order: SortOrder = None,
+    ) -> GroupList:
         """List groups of type SDL
         See: `List all groups of specified type <https://developers.symphony.com/restapi/reference/listgroups>`_
 
@@ -112,10 +122,7 @@ class SymphonyGroupService:
         return await self._group_api.list_groups(**kwargs)
 
     async def list_all_groups(
-              self,
-              status: Status = None,
-              chunk_size: int = 100,
-              max_number: int = None
+        self, status: Status = None, chunk_size: int = 100, max_number: int = None
     ) -> AsyncGenerator[ReadGroup, None]:
         """Returns an asynchronous generator of groups of type SDL
         See: `List all groups of specified type <https://developers.symphony.com/restapi/reference/listgroups>`_
@@ -128,15 +135,15 @@ class SymphonyGroupService:
         """
 
         async def groups_one_page(limit, after=None):
-            result = await self.list_groups(status=status,
-                                            limit=limit,
-                                            after=after)
-            return result.data, getattr(result.pagination.cursors, 'after', None)
+            result = await self.list_groups(status=status, limit=limit, after=after)
+            return result.data, getattr(result.pagination.cursors, "after", None)
 
         return cursor_based_pagination(groups_one_page, chunk_size, max_number)
 
     @retry(retry=refresh_bearer_token_if_unauthorized)
-    async def update_group(self, if_match: str, group_id: str, update_group: UpdateGroup) -> ReadGroup:
+    async def update_group(
+        self, if_match: str, group_id: str, update_group: UpdateGroup
+    ) -> ReadGroup:
         """Update an existing group
         See: `Update a group <https://developers.symphony.com/restapi/reference/updategroup>`_
 
@@ -145,8 +152,9 @@ class SymphonyGroupService:
         :param update_group: the group fields to be updated
         :return: the updated group
         """
-        return await self._group_api.update_group(x_symphony_host="", if_match=if_match, group_id=group_id,
-                                                  update_group=update_group)
+        return await self._group_api.update_group(
+            x_symphony_host="", if_match=if_match, group_id=group_id, update_group=update_group
+        )
 
     @retry(retry=refresh_bearer_token_if_unauthorized)
     async def update_avatar(self, group_id: str, image: str) -> ReadGroup:
@@ -159,7 +167,9 @@ class SymphonyGroupService:
         :return: the updated group
         """
         upload_avatar = UploadAvatar(image=image)
-        return await self._group_api.update_avatar(x_symphony_host="", group_id=group_id, upload_avatar=upload_avatar)
+        return await self._group_api.update_avatar(
+            x_symphony_host="", group_id=group_id, upload_avatar=upload_avatar
+        )
 
     @retry(retry=refresh_bearer_token_if_unauthorized)
     async def get_group(self, group_id: str) -> ReadGroup:
@@ -181,13 +191,13 @@ class SymphonyGroupService:
         :return: the updated group
         """
         member = Member(member_id=user_id, member_tenant=extract_tenant_id(user_id))
-        return await self._group_api.add_member_to_group(x_symphony_host="", group_id=group_id,
-                                                         add_member=AddMember(member=member))
+        return await self._group_api.add_member_to_group(
+            x_symphony_host="", group_id=group_id, add_member=AddMember(member=member)
+        )
 
 
 class OAuthSession:
-    """Used to handle the bearer token needed to call Groups endpoints.
-    """
+    """Used to handle the bearer token needed to call Groups endpoints."""
 
     def __init__(self, login_client, session, retry_config):
         self._authentication_api = AuthenticationApi(login_client)
@@ -197,10 +207,10 @@ class OAuthSession:
 
     @retry
     async def refresh(self):
-        """Refreshes internal Bearer authentication token from the bot sessionToken.
-        """
-        jwt_token = await self._authentication_api.idm_tokens_post(await self._auth_session.session_token,
-                                                                   scope="profile-manager")
+        """Refreshes internal Bearer authentication token from the bot sessionToken."""
+        jwt_token = await self._authentication_api.idm_tokens_post(
+            await self._auth_session.session_token, scope="profile-manager"
+        )
         self._bearer_token = jwt_token.access_token
 
     async def get_auth_settings(self):
@@ -208,12 +218,17 @@ class OAuthSession:
 
         :return: the map of auth_settings containing the header value with the current bearer token
         """
-        return {"bearerAuth": {"in": "header", "type": "bearer", "key": "Authorization",
-                               "value": "Bearer " + await self._get_bearer_token()}}
+        return {
+            "bearerAuth": {
+                "in": "header",
+                "type": "bearer",
+                "key": "Authorization",
+                "value": "Bearer " + await self._get_bearer_token(),
+            }
+        }
 
     async def _get_bearer_token(self):
-        """Returns the bearer token
-        """
+        """Returns the bearer token"""
         if self._bearer_token is None:
             await self.refresh()
         return self._bearer_token