Пример #1
0
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