oauth: Add project to saved credentials when using the notebook authe…

…nticator.

PiperOrigin-RevId: 668535162

python/ee/oauth.py

@@ -14,6 +14,7 @@
 import http.server
 import json
 import os
+import re
 import shutil
 import subprocess
 import sys
@@ -226,6 +227,14 @@ def _in_jupyter_shell() -> bool:
     return False
 
 
+def _project_number_from_client_id(client_id: Optional[str]) -> Optional[str]:
+  """Returns the project number associated with the given OAuth client ID."""
+  # Client IDs are of the form:
+  # PROJECTNUMBER-BASE32STUFF.apps.googleusercontent.com.
+  match = re.search(r'(\d+)-', client_id or '')
+  return match.group(1) if match else None
+
+
 def _obtain_and_write_token(
     auth_code: Optional[str] = None,
     code_verifier: Optional[str] = None,
@@ -259,6 +268,9 @@ def _obtain_and_write_token(
   token = request_token(auth_code.strip(), code_verifier, **client_info)
   client_info['refresh_token'] = token
   client_info['scopes'] = scopes
+  project = _project_number_from_client_id(client_info.get('client_id'))
+  if project:
+    client_info['project'] = project
   write_private_json(get_credentials_path(), client_info)
   print('\nSuccessfully saved authorization token.')