Source code for globus_sdk.tokenstorage.memory_adapter
from __future__ import annotations
import typing as t
from globus_sdk.services.auth import OAuthTokenResponse
from globus_sdk.tokenstorage.base import StorageAdapter
[docs]
class MemoryAdapter(StorageAdapter):
"""
A token storage adapter which stores tokens in process memory.
Tokens are lost when the process exits.
"""
def __init__(self) -> None:
self._tokens: dict[str, dict[str, t.Any]] = {}
[docs]
def store(self, token_response: OAuthTokenResponse) -> None:
self._tokens.update(token_response.by_resource_server)
[docs]
def get_token_data(self, resource_server: str) -> dict[str, t.Any] | None:
return self._tokens.get(resource_server)