in CommonLayerCode/datalake-library/python/datalake_library/octagon/client.py [0:0]
def build(self):
""" Client initialization method """
# Initialization here
if self.run_in_fargate:
if "AWS_ACCESS_KEY" in os.environ and "AWS_SECRET_ACCESS_KEY" in os.environ:
aws_access_key = os.environ.get("AWS_ACCESS_KEY")
aws_secret_access_key = os.environ.get("AWS_SECRET_ACCESS_KEY")
boto3.setup_default_session(
profile_name=self.profile,
region_name=self.region,
aws_access_key_id=aws_access_key,
aws_secret_access_key=aws_secret_access_key,
)
else:
msg = "Environment variables AWS_ACCESS_KEY and AWS_SECRET_ACCESS_KEY are not set"
self.logger.error(msg)
raise ValueError(msg)
elif self.run_in_lambda:
pass
else:
boto3.setup_default_session(profile_name=self.profile, region_name=self.region)
self.account_id = boto3.client("sts").get_caller_identity().get("Account")
self.dynamodb = boto3.resource("dynamodb")
self.sns = boto3.client("sns")
self.config = ConfigParser(self.configuration_file, self.configuration_instance)
self.meta = OctagonMetadata(self.metadata_file)
self.initialized = True
return self