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
|
# costexplorer.py
import boto3
import datetime
from tabulate import tabulate
def get_section(config):
profile = config["aws"].get("profile")
region = config["aws"]["region"]
session = boto3.Session(
profile_name=profile if profile else None, region_name=region
)
client = session.client("ce")
today = datetime.date.today()
start = (today - datetime.timedelta(days=1)).strftime("%Y-%m-%d")
end = today.strftime("%Y-%m-%d")
response = client.get_cost_and_usage(
TimePeriod={"Start": start, "End": end},
Granularity="DAILY",
Metrics=["UnblendedCost"],
GroupBy=[{"Type": "DIMENSION", "Key": "SERVICE"}],
)
results = response["ResultsByTime"][0]
date = results["TimePeriod"]["Start"]
estimated = results.get("Estimated", False)
rows = []
groups = results.get("Groups", [])
if groups:
for group in groups:
service = group["Keys"][0]
cost = (
group.get("Metrics", {}).get("UnblendedCost", {}).get("Amount", "0.00")
)
rows.append([service, f"${float(cost):.2f}"])
else:
rows.append(["No service-level data available", ""])
total_cost = results.get("Total", {}).get("UnblendedCost", {}).get("Amount", None)
if total_cost is None:
total_cost = sum(
float(group.get("Metrics", {}).get("UnblendedCost", {}).get("Amount", 0.0))
for group in groups
)
rows.append(["TOTAL", f"${float(total_cost):.2f}"])
table = tabulate(
rows,
headers=["Service", "Cost"],
tablefmt="simple_grid",
colalign=("left", "right"),
)
lines = [
f"AWS Billing Report for {date}",
f"[https://{config['aws'].get('region')}.console.aws.amazon.com/costmanagement/]",
table,
]
if estimated:
lines.append("\nNote: Costs are estimated and may change.")
return "\n".join(lines)
|
