Rietveld schedules builds on buildbucket

appengine/chromium_rietveld/codereview/auth_utils.py

 # Copyright 2013 Google Inc.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
 #
 #     http://www.apache.org/licenses/LICENSE-2.0
 #
 # Unless required by applicable law or agreed to in writing, software
 # distributed under the License is distributed on an "AS IS" BASIS,
@@ skipping 76 matching lines @@
           config is in the datastore, else the tuple (None, None, []).
     """
     config = cls.get_by_id(cls.GLOBAL_KEY)
     if config is None:
       return None, None, []
     else:
       return (config.client_id, config.client_secret,
               config.additional_client_ids)
 
 
+class OwnClientId(ndb.Model):
+  """Model to store Rietveld's own API client ids."""
+  client_id = ndb.StringProperty()
+
+
+def get_own_web_client_id():
+  """Returns client id for the web client."""
+  return OwnClientId.get_or_insert('web').client_id
+
+
 def _get_client_id(tries=3):
   """Call oauth.get_client_id() and retry if it times out."""
   for attempt in xrange(tries):
     try:
       return oauth.get_client_id(EMAIL_SCOPE)
     except apiproxy_errors.DeadlineExceededError:
       logging.error('get_client_id() timed out on attempt %r', attempt)
       if attempt == tries - 1:
         raise
 
-  
+
 def get_current_rietveld_oauth_user():
   """Gets the current OAuth 2.0 user associated with a request.
 
   This user must be intending to reach this application, so we check the token
   info to verify this is the case.
 
   Returns:
     A users.User object that was retrieved from the App Engine OAuth library if
         the token is valid, otherwise None.
   """
@@ skipping 66 matching lines @@
   if cookie_user_is_admin:
     return cookie_user_is_admin
 
   # oauth.is_current_user_admin is not sufficient, we must first check that the
   # OAuth 2.0 user has a token minted for this application.
   rietveld_user = get_current_rietveld_oauth_user()
   if rietveld_user is None:
     return False
 
   return oauth.is_current_user_admin(EMAIL_SCOPE)