def get_connection_info(): ci = namedtuple("ConnectionInfo", "dbhost, dbport, dbuser, dbpass, dbname") config = ConfigHelper() ci.dbhost = config.get_config("DB_HOST") ci.dbport = config.get_config("DB_PORT") ci.dbuser = config.get_config("DB_USER") ci.dbpass = config.get_config("DB_PASS") ci.dbname = config.get_config("DB_NAME") return ci
def __init__(self): config = ConfigHelper() self.bucket = config.get_config("GCP_BUCKET") self.account = config.get_config("GOOGLE_APPLICATION_CREDENTIALS") # all these values are necessary if not self.bucket or not self.account: raise Exception("""The GCP_BUCKET or GOOGLE_APPLICATION_CREDENTIALS are not set and there is not a .dspreview.json file in the user's home folder, please provide one of them.""") # make sure they are the same os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = self.account self._service = None
def test_get_value(self, mock_path_exists): mock_path_exists.return_value = True read_data = """{"key":"value"}""" mo = mock_open(read_data=read_data) with patch('utils.config_helper.open', mo): config = ConfigHelper() self.assertEqual( config.get_config("key"), "value", """Is is not finding the correct key-value config""")
def __enter__(self): try: config = ConfigHelper() credentials = pika.PlainCredentials(config.get_config("MQ_USER"), config.get_config("MQ_PASS")) host = config.get_config("MQ_HOST") port = config.get_config("MQ_PORT") vhost = config.get_config("MQ_VHOST") self.queue = config.get_config("MQ_QUEUE") parameters = pika.ConnectionParameters(host=host, port=port, virtual_host=vhost, credentials=credentials) self.connection = pika.BlockingConnection(parameters) self.channel = self.connection.channel() self.channel.queue_declare(queue=self.queue, durable=True) except Exception as err: logger.exception(err) raise Exception("""It was not possible to connect to the MQ, please check the connection information""") return self