forked from lacework/python-sdk
-
Notifications
You must be signed in to change notification settings - Fork 0
/
example_team_users.py
79 lines (53 loc) · 1.92 KB
/
example_team_users.py
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
# -*- coding: utf-8 -*-
"""
Example script showing how to use the LaceworkClient class.
"""
import logging
from dotenv import load_dotenv
from laceworksdk import LaceworkClient
logging.basicConfig(level=logging.DEBUG)
load_dotenv()
def standard_user_example(client: LaceworkClient):
"""
Example of create/update/delete and group management for standard user
"""
# Create user
data = client.team_users.create("test user", "[email protected]", "test-company")
guid = data["data"]["userGuid"]
logging.debug(f'user guid created:\n{guid}')
# Get one user
client.team_users.get(guid)
# Update user
client.team_users.update(guid, user_enabled=0)
# Add user to group
client.user_groups.add_users("LACEWORK_USER_GROUP_POWER_USER", [guid])
# Remove user from group
client.user_groups.remove_users("LACEWORK_USER_GROUP_POWER_USER", [guid])
# Delete user
client.team_users.delete(guid)
def service_user_example(client: LaceworkClient):
"""
Example of create/update/delete and group management for service user
"""
# Create user
data = client.team_users.create("test service user", description="test service user", type="ServiceUser")
guid = data["data"]["userGuid"]
logging.debug(f'user guid created:\n{guid}')
# Get one user
client.team_users.get(guid)
# Update user
client.team_users.update(guid, user_enabled=0)
# Add user to group
client.user_groups.add_users("LACEWORK_USER_GROUP_POWER_USER", [guid])
# Remove user from group
client.user_groups.remove_users("LACEWORK_USER_GROUP_POWER_USER", [guid])
# Delete user
client.team_users.delete(guid)
if __name__ == "__main__":
# Instantiate a LaceworkClient instance
lacework_client = LaceworkClient()
# TeamUsers API
# Get users
lacework_client.team_users.get()
standard_user_example(lacework_client)
service_user_example(lacework_client)