google_apis/google_api_keys.py

   1 #!/usr/bin/env python
   2 # Copyright (c) 2012 The Chromium Authors. All rights reserved.
   3 # Use of this source code is governed by a BSD-style license that can be
   4 # found in the LICENSE file.
   5
   6 """Python API for retrieving API keys.
   7
   8 Note that this cannot have the exact same semantics (at the moment) as
   9 the C++ API in google_api_keys.h, since it does not have access to gyp
  10 variables or preprocessor defines.
  11
  12 TODO(joi): Give this have the same semantics as the C++ API.
  13 """
  14
  15 import os
  16 import re
  17 import sys
  18
  19
  20 # The token returned when an API key is unset.
  21 DUMMY_TOKEN = 'dummytoken'
  22
  23
  24 def _GetTokenFromOfficialFile(token_name):
  25   """Parses the token from the official file if it exists, else returns None."""
  26   official_path = os.path.join(os.path.dirname(__file__),
  27                                'internal/google_chrome_api_keys.h')
  28   if not os.path.isfile(official_path):
  29     return None
  30
  31   line_regexp = '^#define\s*%s\s*"([^"]+)"' % token_name
  32   line_pattern = re.compile(line_regexp)
  33   def ParseLine(current_line):
  34     result = line_pattern.match(current_line)
  35     if result:
  36       return result.group(1)
  37     else:
  38       return None
  39
  40   with open(official_path) as f:
  41     current_line = ''
  42     for line in f:
  43       if line.endswith('\\\n'):
  44         current_line += line[:-2]
  45       else:
  46         # Last line in multi-line #define, or a line that is not a
  47         # continuation line.
  48         current_line += line
  49         token = ParseLine(current_line)
  50         if token:
  51           if current_line.count('"') != 2:
  52             raise Exception(
  53               'Embedded quotes and multi-line strings are not supported.')
  54           return token
  55         current_line = ''
  56     return None
  57
  58
  59 def _GetToken(token_name):
  60   """Returns the API token with the given name, or DUMMY_TOKEN by default."""
  61   if token_name in os.environ:
  62     return os.environ[token_name]
  63   token = _GetTokenFromOfficialFile(token_name)
  64   if token:
  65     return token
  66   else:
  67     return DUMMY_TOKEN
  68
  69
  70 def GetAPIKey():
  71   """Returns the simple API key."""
  72   return _GetToken('GOOGLE_API_KEY')
  73
  74
  75 def GetClientID(client_name):
  76   """Returns the OAuth 2.0 client ID for the client of the given name."""
  77   return _GetToken('GOOGLE_CLIENT_ID_%s' % client_name)
  78
  79
  80 def GetClientSecret(client_name):
  81   """Returns the OAuth 2.0 client secret for the client of the given name."""
  82   return _GetToken('GOOGLE_CLIENT_SECRET_%s' % client_name)
  83
  84
  85 if __name__ == "__main__":
  86   print 'GOOGLE_API_KEY=%s' % GetAPIKey()
  87   print 'GOOGLE_CLIENT_ID_MAIN=%s' % GetClientID('MAIN')
  88   print 'GOOGLE_CLIENT_SECRET_MAIN=%s' % GetClientSecret('MAIN')
  89   print 'GOOGLE_CLIENT_ID_CLOUD_PRINT=%s' % GetClientID('CLOUD_PRINT')
  90   print 'GOOGLE_CLIENT_SECRET_CLOUD_PRINT=%s' % GetClientSecret('CLOUD_PRINT')
  91   print 'GOOGLE_CLIENT_ID_REMOTING=%s' % GetClientID('REMOTING')
  92   print 'GOOGLE_CLIENT_SECRET_REMOTING=%s' % GetClientSecret('REMOTING')