def lambda_handler(event, context): config = DotDict({}) config.account = boto3.client("sts").get_caller_identity().get("Account") config.athena_workgroup = os.environ.get("ATHENA_WORKGROUP", "defenda_data_lake") config.athena_database = os.environ.get("ATHENA_DATABASE", "defenda_data_lake") config.athena_table = os.environ.get("ATHENA_TABLE", "events") # query status/wait for response athena_query = get_athena_query(config) logger.debug(athena_query) cursor = connect(work_group=config.athena_workgroup).cursor() cursor.execute(athena_query) logger.debug("Query finished: {}".format(cursor.state)) return