db-api-example/users.py at main · qnix-databricks/db-api-example · GitHub

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
import requests

from context import ProvContext


class UsersApi:
    def __init__(self, provCtx):
        self.provCtx = provCtx

    def me(self):
        response = requests.get(self.provCtx.api_endpoint('api/2.0/preview/scim/v2/Me'),
                                headers = self.provCtx.auth_headers())
        if response.status_code == 200:
            return response.json()
        else:
            print(f'[users.me] status: {response.status_code},\ntext: {response.text}')
            return None

    def users(self):
        response = requests.get(self.provCtx.api_endpoint('api/2.0/preview/scim/v2/Users'),
                                headers = self.provCtx.auth_headers())
        if response.status_code == 200:
            return response.json()
        else:
            print(f'[users.me] status: {response.status_code},\ntext: {response.text}')
            return None


if __name__ == '__main__':
    provCtx = ProvContext.get()
    usersApi = UsersApi(provCtx)

    ## List all the users in the workspace
    res = usersApi.users()
    # print(res)
    for u in res['Resources']:
        try:
            print('==>> User:', u['displayName'], '\n',
                'id:', u['id'], ', userName', u['userName'], '\n',
                'entitlements:', u['entitlements'], '\n',
                )
        except:
            pass