python/google/appengine/ext/ndb/google_imports.py

   1 """Dynamically decide from where to import Google App Engine modules.
   2
   3 All other NDB code should import its Google App Engine modules from
   4 this module.  If necessary, add new imports here (in both places).
   5 """
   6
   7 try:
   8   from google.appengine.datastore import entity_pb
   9   normal_environment = True
  10 except ImportError:
  11   from google3.storage.onestore.v3 import entity_pb
  12   normal_environment = False
  13
  14 if normal_environment:
  15   from google.appengine.api.blobstore import blobstore as api_blobstore
  16   from google.appengine.api import apiproxy_rpc
  17   from google.appengine.api import apiproxy_stub_map
  18   from google.appengine.api import datastore
  19   from google.appengine.api import datastore_errors
  20   from google.appengine.api import datastore_types
  21   from google.appengine.api import memcache
  22   from google.appengine.api import namespace_manager
  23   from google.appengine.api import prospective_search
  24   from google.appengine.api import taskqueue
  25   from google.appengine.api import urlfetch
  26   from google.appengine.api import users
  27   from google.appengine.api.prospective_search import prospective_search_pb
  28   from google.appengine.datastore import datastore_query
  29   from google.appengine.datastore import datastore_rpc
  30   # This line will fail miserably for any app using auto_import_fixer
  31   # because auto_import_fixer only set up simple alias between
  32   # google and google3. But entity_pb is move to a different path completely.
  33   from google.appengine.datastore import entity_pb
  34   from google.appengine.ext.blobstore import blobstore as ext_blobstore
  35   from google.appengine.ext import db
  36   from google.appengine.ext import gql
  37   from google.appengine.runtime import apiproxy_errors
  38   from google.net.proto import ProtocolBuffer
  39 else:
  40   from google3.apphosting.api.blobstore import blobstore as api_blobstore
  41   from google3.apphosting.api import apiproxy_rpc
  42   from google3.apphosting.api import apiproxy_stub_map
  43   from google3.apphosting.api import datastore
  44   from google3.apphosting.api import datastore_errors
  45   from google3.apphosting.api import datastore_types
  46   from google3.apphosting.api import memcache
  47   from google3.apphosting.api import namespace_manager
  48   from google3.apphosting.api import taskqueue
  49   from google3.apphosting.api import urlfetch
  50   from google3.apphosting.api import users
  51   from google3.apphosting.datastore import datastore_query
  52   from google3.apphosting.datastore import datastore_rpc
  53   from google3.storage.onestore.v3 import entity_pb
  54   from google3.apphosting.ext.blobstore import blobstore as ext_blobstore
  55   from google3.apphosting.ext import db
  56   from google3.apphosting.ext import gql
  57   from google3.apphosting.runtime import apiproxy_errors
  58   from google3.net.proto import ProtocolBuffer
  59   # Prospective search is optional.
  60   try:
  61     from google3.apphosting.api import prospective_search
  62     from google3.apphosting.api.prospective_search import prospective_search_pb
  63   except ImportError:
  64     pass