watch_cloudwatch_logs.py

watch_cloudwatch_logs.py

#!/usr/bin/env python3

import click
import boto3
import time

logs_client = boto3.client('logs')


@click.command()
@click.option("--group", type=str)
@click.option("--query", type=str)
@click.option("--start", type=int, default=86400)
@click.option("--end", help='start and end are relative', type=int, default=0)
def query_logs(group, start, end, query):
    now = int(time.time())
    start_time = now - start
    end_time = now - end

    query_id = logs_client.start_query(
        logGroupName=group,
        startTime=start_time,
        endTime=end_time,
        queryString=query,
    )['queryId']

    resp = None
    while True:
        resp = logs_client.get_query_results(queryId=query_id)
        if resp['status'] == 'Running':
            time.sleep(1)
            continue
        break

    results = resp['results']
    for result in results:
        result = {field['field']: field['value'] for field in result}
        print(f"{result['@timestamp']} {result['@message']}".strip())


if __name__ == "__main__":
    query_logs()