ankitbko / python-requests-msal

Sample app integrating python requests library with MSAL to get AD token using Device Code Flow

python-requests-msal

Sample app integrating python requests library with MSAL to get AD token using Device Code Flow

Usage

Feel free to copy and use auth.py and resr_service.py in your own project. The files depends upon following PyPI -

• msal
• requests

Initialize DeviceCodeFlowTokenAuth by passing auth_config object to it. auth_config must have

• client_id: Azure AD Application ID
• authority: Azure AD authority
• scope: Scope to send while requesting for token

Pass the initialized DeviceCodeFlowTokenAuth to RestService. RestService will use it to set Authorization token before sending request.

In addition to above, change _DEFAULT_TOKEN_CACHE_DIR in DeviceCodeFlowTokenAuth to any folder of your choosing where you would like the token to be chached. More detaisl below.

How does it work

DeviceCodeFlowTokenAuth accepts configuration which contains client_id, authority and scope to get access token. It first checks if a valid access token or refresh token is present in cache. The in-memory cache is serialized and stored as a file named token in a folder specified by _DEFAULT_TOKEN_CACHE_DIR variable in home folder (default ~/.myapp).

The code will try to read the token from the folder above and populate its TokenCache. If valid access token exist it will use it. Otherwise if a valid refresh token exists it will use it to get a fresh access token and save it in the cache. If neither of them are valid it will initiate device code flow to fetch fresh tokens. The device code flow will print the code generated to console.