Add a main function to the Admin SDK Quickstarts.

Eric Koleda · Eric Koleda · commit fa36ca4e0636 · 2018-07-24T15:47:14.000-04:00
diff --git a/admin_sdk/directory/quickstart.py b/admin_sdk/directory/quickstart.py
@@ -14,33 +14,45 @@
 
 # [START admin_sdk_directory_quickstart]
 """
-Shows basic usage of the Admin SDK Directory API. Lists of first 10 users in the
-domain.
+Shows basic usage of the Admin SDK Directory API. Lists of first 10 users in
+the domain.
 """
 from __future__ import print_function
 from apiclient.discovery import build
 from httplib2 import Http
 from oauth2client import file, client, tools
 
-# Setup the Admin SDK Directory API
+# If modifying these scopes, delete the file token.json.
 SCOPES = 'https://www.googleapis.com/auth/admin.directory.user'
-store = file.Storage('token.json')
-creds = store.get()
-if not creds or creds.invalid:
-    flow = client.flow_from_clientsecrets('credentials.json', SCOPES)
-    creds = tools.run_flow(flow, store)
-service = build('admin', 'directory_v1', http=creds.authorize(Http()))
-
-# Call the Admin SDK Directory API
-print('Getting the first 10 users in the domain')
-results = service.users().list(customer='my_customer', maxResults=10,
-                               orderBy='email').execute()
-users = results.get('users', [])
-
-if not users:
-    print('No users in the domain.')
-else:
-    print('Users:')
-    for user in users:
-        print('{0} ({1})'.format(user['primaryEmail'], user['name']['fullName']))
+
+
+def main():
+    """Shows basic usage of the Google Admin SDK Directory API.
+
+    Outputs a list of first 10 users in the domain.
+    """
+    store = file.Storage('token.json')
+    creds = store.get()
+    if not creds or creds.invalid:
+        flow = client.flow_from_clientsecrets('credentials.json', SCOPES)
+        creds = tools.run_flow(flow, store)
+    service = build('admin', 'directory_v1', http=creds.authorize(Http()))
+
+    # Call the Admin SDK Directory API
+    print('Getting the first 10 users in the domain')
+    results = service.users().list(customer='my_customer', maxResults=10,
+                                   orderBy='email').execute()
+    users = results.get('users', [])
+
+    if not users:
+        print('No users in the domain.')
+    else:
+        print('Users:')
+        for user in users:
+            print(u'{0} ({1})'.format(user['primaryEmail'],
+                                      user['name']['fullName']))
+
+
+if __name__ == '__main__':
+    main()
 # [END admin_sdk_directory_quickstart]
diff --git a/admin_sdk/reports/quickstart.py b/admin_sdk/reports/quickstart.py
@@ -18,35 +18,41 @@
 events.
 """
 from __future__ import print_function
-from apiclient.discovery import build, http
+from apiclient.discovery import build
 from httplib2 import Http
 from oauth2client import file, client, tools
-import StringIO
-import random
 
-import apiclient
-from email import Utils
-from email import MIMEText
-
-# Setup the Admin SDK Reports API
+# If modifying these scopes, delete the file token.json.
 SCOPES = 'https://www.googleapis.com/auth/admin.reports.audit.readonly'
-store = file.Storage('token.json')
-creds = store.get()
-if not creds or creds.invalid:
-    flow = client.flow_from_clientsecrets('credentials.json', SCOPES)
-    creds = tools.run_flow(flow, store)
-service = build('admin', 'reports_v1', http=creds.authorize(Http()))
-
-print('Getting the last 10 login events')
-results = service.activities().list(userKey='all', applicationName='login',
-    maxResults=10).execute()
-activities = results.get('items', [])
-
-if not activities:
-    print('No logins found.')
-else:
-    print('Logins:')
-    for activity in activities:
-        print('{0}: {1} ({2})'.format(activity['id']['time'],
-            activity['actor']['email'], activity['events'][0]['name']))
+
+
+def main():
+    """Shows basic usage of the Google Admin SDK Reports API.
+
+    Outputs a list of last 10 login events.
+    """
+    store = file.Storage('token.json')
+    creds = store.get()
+    if not creds or creds.invalid:
+        flow = client.flow_from_clientsecrets('credentials.json', SCOPES)
+        creds = tools.run_flow(flow, store)
+    service = build('admin', 'reports_v1', http=creds.authorize(Http()))
+
+    print('Getting the last 10 login events')
+    results = service.activities().list(userKey='all', applicationName='login',
+                                        maxResults=10).execute()
+    activities = results.get('items', [])
+
+    if not activities:
+        print('No logins found.')
+    else:
+        print('Logins:')
+        for activity in activities:
+            print(u'{0}: {1} ({2})'.format(activity['id']['time'],
+                                           activity['actor']['email'],
+                                           activity['events'][0]['name']))
+
+
+if __name__ == '__main__':
+    main()
 # [END admin_sdk_reports_quickstart]
diff --git a/admin_sdk/reseller/quickstart.py b/admin_sdk/reseller/quickstart.py
@@ -18,35 +18,41 @@
 events.
 """
 from __future__ import print_function
-from apiclient.discovery import build, http
+from apiclient.discovery import build
 from httplib2 import Http
 from oauth2client import file, client, tools
-import StringIO
-import random
 
-import apiclient
-from email import Utils
-from email import MIMEText
-
-# Setup the Admin SDK Reports API
+# If modifying these scopes, delete the file token.json.
 SCOPES = 'https://www.googleapis.com/auth/admin.reports.audit.readonly'
-store = file.Storage('token.json')
-creds = store.get()
-if not creds or creds.invalid:
-    flow = client.flow_from_clientsecrets('credentials.json', SCOPES)
-    creds = tools.run_flow(flow, store)
-service = build('admin', 'reports_v1', http=creds.authorize(Http()))
-
-print('Getting the last 10 login events')
-results = service.activities().list(userKey='all', applicationName='login',
-    maxResults=10).execute()
-activities = results.get('items', [])
-
-if not activities:
-    print('No logins found.')
-else:
-    print('Logins:')
-    for activity in activities:
-        print('{0}: {1} ({2})'.format(activity['id']['time'],
-            activity['actor']['email'], activity['events'][0]['name']))
+
+
+def main():
+    """Shows basic usage of the Google Admin-SDK Reseller API.
+
+    Outputs the first 10 subscriptions you manage.
+    """
+    store = file.Storage('token.json')
+    creds = store.get()
+    if not creds or creds.invalid:
+        flow = client.flow_from_clientsecrets('credentials.json', SCOPES)
+        creds = tools.run_flow(flow, store)
+    service = build('admin', 'reports_v1', http=creds.authorize(Http()))
+
+    print('Getting the last 10 login events')
+    results = service.activities().list(userKey='all', applicationName='login',
+                                        maxResults=10).execute()
+    activities = results.get('items', [])
+
+    if not activities:
+        print('No logins found.')
+    else:
+        print('Logins:')
+        for activity in activities:
+            print(u'{0}: {1} ({2})'.format(activity['id']['time'],
+                                           activity['actor']['email'],
+                                           activity['events'][0]['name']))
+
+
+if __name__ == '__main__':
+    main()
 # [END admin_sdk_reseller_quickstart]
