from authalligator_client.client import Client
from authalligator_client.enums import ProviderType
from authalligator_client.exceptions import AccountError

client = Client(token='my-secret-token', service_url='authalligator.example.com')

try:
    authorize_result = client.authorize_account(
        provider=ProviderType.MICROSOFT,
        authorization_code='my-code',
        redirect_uri='mysite.example.com/oauth2/microsoft/callback',
    )
    print("Got an access token + key for refreshing!")
    print("Access Key:", authorize_result.access_key)
    print("Username:", authorize_result.account.username)
    print("Provider:", authorize_result.account.provider)
    print("Access Token:", authorize_result.account.access_token)
    print("Access Token Expiry:", authorize_result.account.access_token_expires_at)
except AccountError as e:
    print("encountered an error!", e.code, e.message)

try:
    account_result = client.query_account(
        provider=authorize_result.account.provider,
        username=authorize_result.account.username,
        account_key=authorize_result.account_key,
    )
    print("New Access Token:". account_result.access_token)
except AccountError as e:
    print("Encountered an error!", e.code, e.message)