Created
May 15, 2018 09:26
-
-
Save tkw1536/b9947dfc62dd78ba9f8e3372df8a9174 to your computer and use it in GitHub Desktop.
Ugly hack to generate stats via the Google Directory API
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
from __future__ import print_function | |
import httplib2 | |
import os | |
from apiclient import discovery | |
from oauth2client import client | |
from oauth2client import tools | |
from oauth2client.file import Storage | |
try: | |
import argparse | |
flags = argparse.ArgumentParser(parents=[tools.argparser]).parse_args() | |
except ImportError: | |
flags = None | |
# If modifying these scopes, delete your previously saved credentials | |
# at ~/.credentials/admin-directory_v1-python-quickstart.json | |
SCOPES = 'https://www.googleapis.com/auth/admin.directory.user' | |
CLIENT_SECRET_FILE = 'client_secret.json' | |
APPLICATION_NAME = 'Directory API Python Quickstart' | |
def get_credentials(): | |
"""Gets valid user credentials from storage. | |
If nothing has been stored, or if the stored credentials are invalid, | |
the OAuth2 flow is completed to obtain the new credentials. | |
Returns: | |
Credentials, the obtained credential. | |
""" | |
home_dir = os.path.expanduser('~') | |
credential_dir = os.path.join(home_dir, '.credentials') | |
if not os.path.exists(credential_dir): | |
os.makedirs(credential_dir) | |
credential_path = os.path.join(credential_dir, | |
'admin-directory_v1-python-quickstart.json') | |
store = Storage(credential_path) | |
credentials = store.get() | |
if not credentials or credentials.invalid: | |
flow = client.flow_from_clientsecrets(CLIENT_SECRET_FILE, SCOPES) | |
flow.user_agent = APPLICATION_NAME | |
if flags: | |
credentials = tools.run_flow(flow, store, flags) | |
else: # Needed only for compatibility with Python 2.6 | |
credentials = tools.run(flow, store) | |
print('Storing credentials to ' + credential_path) | |
return credentials | |
def main(): | |
"""Shows basic usage of the Google Admin SDK Directory API. | |
Creates a Google Admin SDK API service object and outputs a list of first | |
10 users in the domain. | |
""" | |
credentials = get_credentials() | |
http = credentials.authorize(httplib2.Http()) | |
service = discovery.build('admin', 'directory_v1', http=http) | |
request = service.users().list(customer='my_customer', orderBy='email') | |
orgs = {} | |
total = 0 | |
while request is not None: | |
result = request.execute() | |
users = result.get('users', []) | |
for u in users: | |
unit = u['orgUnitPath'] | |
if unit in orgs: | |
orgs[unit] += 1 | |
else: | |
orgs[unit] = 1 | |
total += 1 | |
request = service.users().list_next(request, result) | |
for o in sorted(orgs): | |
print("{}: {}".format(o, orgs[o])) | |
print("-----------") | |
print("Total: {}".format(total)) | |
if __name__ == '__main__': | |
main() |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment