debrouwere / askmanta Goto Github PK

import os
import logging
import manta 

logging.basicConfig()

MANTA_URL = os.getenv('MANTA_URL')
MANTA_USER = os.getenv('MANTA_URL')
MANTA_KEY_ID = os.getenv('MANTA_KEY_ID')

class MantaConnectionError(Exception):
  pass

def _check_env():
  """
  Make sure environmental variables are present.
  """
  if not MANTA_URL:
    raise MantaConnectionError(
      'You must set MANTA_URL as an environmental variable.'
    )

  if not MANTA_USER:
    raise MantaConnectionError(
      'You must set MANTA_USER as an environmental variable.'
    )

  if 'localhost' not in MANTA_URL:
    if not MANTA_KEY_ID:
      raise MantaConnectionError(
        'For a non-local client, you must set'
        'MANTA_KEY_ID as an environmental variable.'
      )  

def connect():
  """
  connect to manta 
  """
  _check_env()

  if 'localhost' in MANTA_URL:
    client = manta.MantaClient(MANTA_URL, MANTA_USER, internal=True)
  else:
    signer = manta.SSHAgentSigner(MANTA_KEY_ID)
    client = manta.MantaClient(MANTA_URL, MANTA_USER, MANTA_KEY_ID)

  return client

# HACK for now, should figure out how 
# to announce errors on init of various classes.
client = connect()