aws_daily_cost_email_lambda/lambda_function.py at main · ijazrushad/aws_daily_cost_email_lambda · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
"""
AWS Daily Cost Reporting Lambda

This script runs on a daily schedule to fetch AWS cost data, including
month-to-date, daily, and forecasted costs. It then formats this data into a
professional HTML report and sends it to a specified recipient via Amazon SES.
"""
import os
import datetime
import boto3

# Initialize AWS clients outside the handler for better performance
ce = boto3.client('ce', region_name='ap-southeast-1')
ses = boto3.client('ses', region_name='ap-southeast-1')


# pylint: disable=too-many-locals
def lambda_handler(_event, context):
    """
    Main function to fetch AWS cost data, format it, and send it via email.
    This function is triggered by an Amazon EventBridge schedule.
    """
    try:
        # --- 1. CONFIGURATION ---
        sender_email = os.environ.get('SENDER_EMAIL')
        recipient_email = os.environ.get('RECIPIENT_EMAIL')
        aws_account_id = context.invoked_function_arn.split(":")[4]

        # --- 2. DATE CALCULATIONS ---
        today = datetime.datetime.now()
        yesterday = today - datetime.timedelta(days=1)
        start_of_month = today.strftime('%Y-%m-01')
        end_of_period = today.strftime('%Y-%m-%d')
        yesterday_str = yesterday.strftime('%Y-%m-%d')
        future_date = today + datetime.timedelta(days=30)
        forecast_end_date = future_date.strftime('%Y-%m-%d')

        # --- 3. AWS API CALLS ---
        service_breakdown = ce.get_cost_and_usage(
            TimePeriod={'Start': start_of_month, 'End': end_of_period},
            Granularity='MONTHLY',
            Metrics=['UnblendedCost'],
            GroupBy=[{'Type': 'DIMENSION', 'Key': 'SERVICE'}]
        )
        forecast_response = ce.get_cost_forecast(
            TimePeriod={'Start': end_of_period, 'End': forecast_end_date},
            Metric='UNBLENDED_COST',
            Granularity='MONTHLY'
        )
        forecasted_cost = float(forecast_response['Total']['Amount'])
        daily_cost_response = ce.get_cost_and_usage(
            TimePeriod={'Start': yesterday_str, 'End': end_of_period},
            Granularity='DAILY',
            Metrics=['UnblendedCost']
        )
        daily_cost = float(
            daily_cost_response['ResultsByTime'][0]['Total']['UnblendedCost']['Amount']
        ) if daily_cost_response['ResultsByTime'] else 0.0

        # --- 4. PROCESS DATA ---
        service_costs, total_cost = {}, 0.0
        for result in service_breakdown['ResultsByTime'][0]['Groups']:
            service_name = result['Keys'][0]
            cost_amount = float(result['Metrics']['UnblendedCost']['Amount'])
            if cost_amount > 0.0:
                service_costs[service_name] = cost_amount
                total_cost += cost_amount

        sorted_services = sorted(
            service_costs.items(), key=lambda item: item[1], reverse=True
        )
        max_cost = sorted_services[0][1] if sorted_services else 1.0

        # --- 5. BUILD AND SEND EMAIL ---
        subject = f"AWS Cost Summary for {aws_account_id} - {end_of_period}"
        body = build_html_body(
            total_cost, forecasted_cost, daily_cost, sorted_services, max_cost
        )
        send_email(sender_email, recipient_email, subject, body)
        return {'statusCode': 200, 'body': 'Email sent successfully!'}

    # pylint: disable=broad-exception-caught
    except Exception as e:
        print(f"An error occurred in the handler: {str(e)}")
        return {'statusCode': 500, 'body': f'An error occurred: {str(e)}'}


def build_html_body(total, forecast, daily, services, max_cost):
    """
    Builds the HTML content for the email report.
    """
    # Using a multi-line string for better readability
    html_template = """
    <html>
    <head>
    <style>
        body {{ font-family: Arial, sans-serif; color: #333; margin: 0;
               padding: 0; background-color: #f7f7f7;}}
        .container {{ max-width: 800px; margin: 20px auto; padding: 20px;
                     border: 1px solid #ddd; border-radius: 8px;
                     background-color: #fff; box-shadow: 0 2px 4px rgba(0,0,0,0.1); }}
        .header {{ font-size: 24px; color: #d9534f; border-bottom: 2px solid #f0f0f0;
                  padding-bottom: 10px; margin-bottom: 20px; }}
        .summary-box {{ display: flex; justify-content: space-around;
                       background-color: #f9f9f9; padding: 20px; text-align: center;
                       border-radius: 8px; margin-bottom: 30px; }}
        .summary-item h3 {{ margin: 0; font-size: 16px; color: #555; font-weight: normal; }}
        .summary-item p {{ margin: 5px 0 0; font-size: 28px; font-weight: bold;
                          color: #0275d8; }}
        h2 {{ font-size: 20px; color: #333; border-bottom: 1px solid #eee;
             padding-bottom: 5px; }}
        table {{ width: 100%; border-collapse: collapse; margin-top: 15px;}}
        th, td {{ padding: 12px; border: 1px solid #ddd; text-align: left; }}
        th {{ background-color: #f2f2f2; }}
        .bar-container {{ width: 100%; background-color: #f1f1f1;
                         border-radius: 4px; height: 22px; }}
        .bar {{ height: 22px; background-color: #4CAF50; border-radius: 4px;
               text-align: right; color: white; line-height: 22px; }}
    </style>
    </head>
    <body>
    <div class="container">
        <div class="header">AWS Cost Report</div>
        <div class="summary-box">
            <div class="summary-item">
                <h3>Month-to-Date Cost</h3>
                <p>${total:,.2f}</p>
            </div>
            <div class="summary-item">
                <h3>Last 24h Cost</h3>
                <p>${daily:,.2f}</p>
            </div>
            <div class="summary-item">
                <h3>Forecasted Monthly Cost</h3>
                <p>${forecast:,.2f}</p>
            </div>
        </div>
        <h2>Service-wise Cost Breakdown</h2>
        <table>
            <tr>
                <th style="width:40%;">Service</th>
                <th style="width:20%;">Cost (USD)</th>
                <th>Cost Distribution</th>
            </tr>
            {service_rows}
        </table>
    </div>
    </body>
    </html>
    """

    service_rows = ""
    for service, cost in services:
        bar_width = (cost / max_cost) * 100 if max_cost > 0 else 0
        service_rows += f"""
            <tr>
                <td>{service}</td>
                <td style="text-align: right;">${cost:,.2f}</td>
                <td>
                    <div class="bar-container">
                        <div class="bar" style="width: {bar_width:.2f}%;
                                                background-color: #5cb85c;"></div>
                    </div>
                </td>
            </tr>
        """
    return html_template.format(
        total=total, daily=daily, forecast=forecast, service_rows=service_rows
    )


def send_email(source, to, subject, body):
    """
    Sends an email using AWS SES.
    """
    try:
        response = ses.send_email(
            Source=source,
            Destination={'ToAddresses': [to]},
            Message={
                'Subject': {'Data': subject, 'Charset': 'UTF-8'},
                'Body': {'Html': {'Data': body, 'Charset': 'UTF-8'}}
            }
        )
        print(f"Email sent successfully. Message ID: {response['MessageId']}")
    except Exception as e:
        print(f"Error sending email: {str(e)}")
        raise e