shims.py

import sys
import importlib
"""Stub versions of optional submodules which may fail to clone."""


class Shim(object):
  def __init__(self):
    self.is_shim = True


class version(Shim):
  def get_version(self):
    return None


class simplewrap(Shim):
  class Wrapper(object):
    def __init__(self):
      self.width = 80
    def wrap(self, string):
      return string


class phone(Shim):
  class Call(object):
    def __init__(self,
                 script_path,
                 version,
                 run_id=None,
                 domain=None,
                 timeout=None,
                 secure=None,
                 platform=None,
                 test=False,
                 fail='exception'):
      pass
    def send_data(self, event_type, run_data={}, run_time=None):
      pass


def get_module_or_shim(module_path):
  """Load the given module, or if not possible, return a stub."""
  try:
    return importlib.import_module(module_path)
  except ImportError:
    sys.stderr.write('Error importing module '+module_path+'. Some functionality may be missing.\n')
  module_name = module_path.split('.')[-1]
  try:
    shim = globals()[module_name]
  except KeyError:
    sys.stderr.write('Error: cannot find a shim named "'+module_name+'".\n')
    raise
  try:
    return shim()
  except TypeError:
    sys.stderr.write('Error: problem loading shim "'+module_name+'".\n')
    raise