Add a timeout to oauth redirect flow, so it does not hang forever

Author: mafrosis

gcsa/_services/authentication.py

@@ -4,7 +4,7 @@
 from typing import List
 
 from googleapiclient import discovery
-from google_auth_oauthlib.flow import InstalledAppFlow
+from google_auth_oauthlib.flow import InstalledAppFlow, WSGITimeout
 from google.auth.transport.requests import Request
 from google.auth.credentials import Credentials
 
@@ -107,7 +107,12 @@ def _get_credentials(
             else:
                 credentials_path = os.path.join(credentials_dir, credentials_file)
                 flow = InstalledAppFlow.from_client_secrets_file(credentials_path, scopes)
-                credentials = flow.run_local_server(host=host, port=port, bind_addr=bind_addr)
+                try:
+                    credentials = flow.run_local_server(
+                        host=host, port=port, bind_addr=bind_addr, timeout_seconds=120
+                    )
+                except WSGITimeout:
+                    print('Authentication flow timed out. Please try again.')
 
             if save_token:
                 with open(token_path, 'wb') as token_file: