[Findit] Cloud Endpoints API to receive reports on flakes.

appengine/findit/common/base_handler.py

 # Copyright 2014 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 import collections
 import json
 import logging
 
 from google.appengine.api import users
 import jinja2
 import webapp2
 
+from common import auth_util
 from common import constants
 
 
 JINJA_ENVIRONMENT = jinja2.Environment(
     loader=jinja2.FileSystemLoader(constants.HTML_TEMPLATE_DIR),
     extensions=['jinja2.ext.autoescape'],
     autoescape=True)
 
 
 def ToJson(data):
@@ skipping 18 matching lines @@
     if (self.request.headers.get('X-AppEngine-QueueName') or
         self.request.headers.get('X-AppEngine-Cron')):
       # Requests from task queues or cron jobs could access all HTTP endpoints.
       return True
     elif self.PERMISSION_LEVEL == Permission.ANYONE:
       return True
     elif self.PERMISSION_LEVEL == Permission.CORP_USER:
       # Only give access to google accounts or admins.
       return self.IsCorpUserOrAdmin()
     elif self.PERMISSION_LEVEL == Permission.ADMIN:
-      return users.is_current_user_admin()
+      return auth_util.IsCurrentUserAdmin()
     else:
       logging.error('Unknown permission level: %s' % self.PERMISSION_LEVEL)
       return False
 
   def IsCorpUserOrAdmin(self):
     """Returns True if the user logged in with corp account or as admin."""
-    user = users.get_current_user()
-    return ((user and user.email().endswith('@google.com')) or
-        users.is_current_user_admin())
+    user_email = auth_util.GetUserEmail()
+    return ((user_email and user_email.endswith('@google.com')) or
+        auth_util.IsCurrentUserAdmin())
 
   @staticmethod
   def CreateError(error_message, return_code=500):
     logging.error('Error occurred: %s', error_message)
     return {
         'template': 'error.html',
         'data': {'error_message': error_message},
         'return_code': return_code,
     }
 
@@ skipping 66 matching lines @@
     else:
       content_type = 'text/html'
 
     if cache_expiry is not None:
       self.response.headers['cache-control'] = (
           'max-age=%s, public' % cache_expiry)
     self.response.headers['Content-Type'] = content_type
     self.response.write(data)
 
   def GetLoginUrl(self):
-    if self.request.referer:
-      return users.create_login_url(self.request.referer)
-    else:
-      return users.create_login_url(self.request.uri)
+    """Returns the login url."""
+    return users.create_login_url(self.request.referer or self.request.uri)
 
   def _Handle(self, handler_func):
     try:
       if not self._HasPermission():
         template = 'error.html'
         data = {
             'error_message':
                 ('Either not login or no permission. '
                  'Please login with your google.com account.'),
             'login_url': self.GetLoginUrl(),
@@ skipping 19 matching lines @@
       return_code = 500
       cache_expiry = None
 
     self._SendResponse(template, data, return_code, cache_expiry)
 
   def get(self):
     self._Handle(self.HandleGet)
 
   def post(self):
     self._Handle(self.HandlePost)