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