How to get access to fresh OAuth tokens from integrations?

[ramnes]

Hey there, thanks for the great work on Authlib!

I'm using the Starlette integration. I would like to update my token in database whenever Authlib gets a fresh token from a client.get call, so that it won't have to refresh on the next request if the newer token is still valid.

So far, the only way that I've found to get access to a renewed token that I can save somewhere and reuse is with something in the lines of:

oauth = OAuth()
oauth.register(...)
partner_client = oauth.create_client(partner_name)

async def ensure_token(token):
    async with partner_client._get_oauth_client(token=token) as client:
        await client.ensure_active_token()
        return client.token

...

@app.route("/")
async def do_stuff():
    token = database.get_token()
    new_token = await ensure_token(token)
    if new_token and new_token != token:
        database.replace_token(new_token)
        token = new_token
    partner_client.get(..., token=token)

This seems quite hackish to me and I guess there is a much better way but I can't find it. Any hint? :)