-
Notifications
You must be signed in to change notification settings - Fork 0
/
find_aliases_for_users.py
99 lines (85 loc) · 3.35 KB
/
find_aliases_for_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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
from __future__ import print_function
import pickle
import time
import os.path
from googleapiclient.discovery import build
from google_auth_oauthlib.flow import InstalledAppFlow
from google.auth.transport.requests import Request
# If modifying these scopes, delete the file token.pickle.
SCOPES = ["https://www.googleapis.com/auth/admin.directory.user"]
### Steps:
### 1. Get list of full names of all users of which aliases should be deleted and append to a list
### 2. Call Google API/user accounts, match names and find all aliases to delete, add to a list
### 3. Add exceptions if alias includes words
### 4. Call Google API/user aliases, delete aliases from a list
### 5. ???
### 6. Profit?
# Examples
# Send a request - created_account = self.get_service().users().insert(body=body).execute()
# Format output - print(u'{0} ({1})'.format(user['primaryEmail'], user['name']['fullName']))
# print(user['aliases'])
def main():
print("Admin SDK Directory API script")
creds = None
# The file token.pickle stores the user's access and refresh tokens, and is
# created automatically when the authorization flow completes for the first
# time.
if os.path.exists("token.pickle"):
with open("token.pickle", "rb") as token:
creds = pickle.load(token)
# If there are no (valid) credentials available, let the user log in.
if not creds or not creds.valid:
if creds and creds.expired and creds.refresh_token:
creds.refresh(Request())
else:
flow = InstalledAppFlow.from_client_secrets_file("credentials.json", SCOPES)
creds = flow.run_local_server(port=0)
# Save the credentials for the next run
with open("token.pickle", "wb") as token:
pickle.dump(creds, token)
service = build("admin", "directory_v1", credentials=creds)
# Get list of user full names seperated by newline
data = open("input.csv", "r", encoding="utf-8")
fullNamesDirty = data.readlines()
data.close()
fullNames = []
for name in fullNamesDirty:
fullNames.append(name.strip("\n"))
# Call the Admin SDK Directory API
results = (
service.users()
.list(
customer="my_customer",
maxResults=500, # pylint: disable=maybe-no-member
orderBy="email",
)
.execute()
)
users = results.get("users", [])
outdatedAliases = []
print("---- Output: ----")
if not users:
print("No users in the domain!")
else:
print("Users:")
for user in users:
for name in fullNames:
if user["name"]["fullName"].lower() == name.lower():
try:
for alias in user["aliases"]:
if (
alias not in outdatedAliases
# Words to exclude from aliases
and alias.find("leder") == -1
and alias.find("kasserer") == -1
):
outdatedAliases.append(alias)
except KeyError:
pass
# Add output aliases to output.csv
f = open("output.csv", "w")
for alias in outdatedAliases:
f.writelines(alias + "\n")
f.close()
if __name__ == "__main__":
main()