File: sample_uamqp_authenticate_client_by_jwt.py

package info (click to toggle)
azure-uamqp-python 1.6.11-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 35,564 kB
  • sloc: ansic: 184,383; cpp: 7,738; python: 7,731; cs: 5,767; sh: 983; xml: 298; makefile: 34
file content (45 lines) | stat: -rw-r--r-- 1,607 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
#-------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for
# license information.
#-------------------------------------------------------------------------
"""
Examples to show how to authenticate Client by jwt token.
"""

from collections import namedtuple

from uamqp import authentication, SendClient, Message

AccessToken = namedtuple("AccessToken", ["token", "expires_on"])


# Define get_token callback which shall be passed to the JWTTokenAuth for token generation
def get_token():
    token = "<token>"  # the token used for authentication.
    expires_on = 253402261199  # The timestamp at which the JWT token will expire formatted as seconds since epoch.
    return AccessToken(token, expires_on)


def authenticate_client_by_jwt():
    # Create the JWTTokenAuth object
    auth_uri = "<amqp endpoint uri for authentication>"  # The AMQP endpoint URI for authentication.
    token_audience = "<token audience>"  # The token audience field.
    auth = authentication.JWTTokenAuth(
        audience=token_audience,
        uri=auth_uri,
        get_token=get_token
    )

    # Instantiate the SendClient with the JWTTokenAuth object
    target = "<target amqp service endpoint>"  # The target AMQP service endpoint.
    send_client = SendClient(target=target, auth=auth)

    # Send a message
    message = Message(b'data')
    send_client.send_message(message)
    send_client.close()


if __name__ == "__main__":
    authenticate_client_by_jwt()