def set_region_name(self, region_name=None): 'we need to set a region name if it is not set yet' env = Env() self.region_name = region_name if self.region_name is None: self.region_name = env.get_env('AWS_REGION')
def set_region(self, region_name=None): '''we need to set a region if it is unset at this point''' env = Env() self.region_name = region_name if region_name is None: self.region_name = env.get_env('AWS_REGION') return self.region_name
def _set_master_account(self, account_number=None): 'set the AWS Organization master account number' try: env = Env() sts = Sts() acc_num = sts.get_account_id() self.master_account = acc_num or account_number or env.get_env( 'AW_MASTER') except Exception as e: raise Exception('Could not find master account info', e) if self.master_account is None: raise Exception('Master Account not set, check AW_MASTER')
def test_env_get(): os.environ['TEST_ENV'] = 'just a test' env = Env() assert env.get_env('TEST_ENV') == 'just a test' assert env.get_env('ENV_TEST') is None os.environ.pop('TEST_ENV')
def test_get_env(env): env = Env() assert env.get_env('BLARGH') == 'hello!' assert env.get_env('HGRALB') is None