db-api-example/security.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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
import requests

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

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

    def list_members(self, group):
        response = requests.get(self.provCtx.api_endpoint('api/2.0/groups/list-members'),
                                headers = self.provCtx.auth_headers(),
                                params = {"group_name": group})
        if response.status_code == 200:
            return response.json()
        else:
            print(f'[groups.list-members] status: {response.status_code},\ntext: {response.text}')
            return None

    def list_parents(self, entity):
        """ entity: user or group """
        response = requests.get(self.provCtx.api_endpoint('api/2.0/groups/list-parents'),
                                headers = self.provCtx.auth_headers(),
                                params = {"user_name": entity})
        if response.status_code == 200:
            return response.json()
        else:
            print(f'[groups.list-parents] status: {response.status_code},\ntext: {response.text}')
            return None

    def create(self, group_name):
        response = requests.post(self.provCtx.api_endpoint('api/2.0/groups/create'),
                                 headers = self.provCtx.auth_headers(),
                                 json = {"group_name": group_name})
        if response.status_code == 200:
            return response.json()
        else:
            print(f'[groups.list-members] status: {response.status_code},\ntext: {response.text}')
            return None

    def add_member(self, user_name, group_name):
        response = requests.post(self.provCtx.api_endpoint('api/2.0/groups/add-member'),
                                 headers = self.provCtx.auth_headers(),
                                 json = {
                                     "user_name": user_name,
                                     "parent_name": group_name
                                 })
        if response.status_code == 200:
            return response.json()
        else:
            print(f'[groups.add-member] status: {response.status_code},\ntext: {response.text}')
            return None

    def remove_member(self, user_name, group_name):
        response = requests.post(self.provCtx.api_endpoint('api/2.0/groups/remove-member'),
                                 headers = self.provCtx.auth_headers(),
                                 json = {
                                     "user_name": user_name,
                                     "parent_name": group_name
                                 })
        if response.status_code == 200:
            return response.json()
        else:
            print(f'[groups.remove-member] status: {response.status_code},\ntext: {response.text}')
            return None

    def delete(self, group_name):
        response = requests.post(self.provCtx.api_endpoint('api/2.0/groups/delete'),
                                 headers = self.provCtx.auth_headers(),
                                 json = {"group_name": group_name})
        if response.status_code == 200:
            return response.json()
        else:
            print(f'[groups.delete] status: {response.status_code},\ntext: {response.text}')
            return None