blob: 8ca5f42e6351ef9db1baff1e07eb18c3da4f8631 (
plain)
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
|
# -*- coding: utf-8 -*-
"""
Display the current AWS bill.
##### WARNING #####
This module generate some costs on the AWS bill. Take care about the cache_timout to limit these fees !
##### WARNING #####
Configuration parameters:
- aws_access_key_id : Your AWS access key
- aws_secret_access_key : Your AWS secret key
- aws_account_id : the root ID of the AWS account. Can be find here https://console.aws.amazon.com/billing/home#/account
- s3_bucket_name : the bucket where billing files are sent by AWS. Follow this article to activate this feature : http://docs.aws.amazon.com/awsaccountbilling/latest/aboutv2/detailed-billing-reports.html
- billing_file : csv file location
- cache_timeout : how often we refresh this module in seconds
@author nawadanp
"""
import csv
import datetime
import boto
from boto.s3.connection import Key
from time import time
class Py3status:
# available configuration parameters
aws_access_key_id = ''
aws_secret_access_key = ''
aws_account_id = ''
s3_bucket_name = ''
billing_file = '/tmp/.aws_billing.csv'
cache_timeout = 3600
def _get_bill_amount(self):
# Billing file name, generated by Amazon itself
# Format : 123456789012-aws-billing-csv-yyyy-mm.csv
s3_file_key = self.aws_account_id + '-aws-billing-csv-' + \
datetime.datetime.now().strftime(
'%Y') + '-' + datetime.datetime.now().strftime('%m') + '.csv'
i = 0
# Connection to s3 service
try:
conn = boto.connect_s3(
self.aws_access_key_id, self.aws_secret_access_key)
except:
return 'conn_error'
# Connection to the bucket
try:
bucket = conn.get_bucket(self.s3_bucket_name)
except:
return 'bucket_error'
# Fetch the objects keys and get the billing file
try:
k = Key(bucket)
k.key = s3_file_key
k.get_contents_to_filename(self.billing_file)
k.close
except:
return 'key_error'
# Parse the file and get the InvoiceTotal amount
try:
with open(self.billing_file, 'rb') as f:
reader = csv.reader(f)
for row in reader:
if ''.join(row).find('InvoiceTotal') == -1:
continue
i = i + 1
return row[-1]
except:
return 'csv_error'
return False
def aws_bill(self, i3s_output_list, i3s_config):
response = {
'cached_until': time() + self.cache_timeout,
'full_text': '',
'color': i3s_config['color_bad']
}
bill_amount = self._get_bill_amount()
if bill_amount == 'csv_error':
response['full_text'] = 'Bad CSV file'
elif bill_amount == 'key_error':
response['full_text'] = 'Key not found in the bucket'
elif bill_amount == 'bucket_error':
response['full_text'] = 'Check the bucket name or your AWS keys'
elif bill_amount == 'conn_error':
response['full_text'] = 'Check your internet access'
elif bill_amount is not False:
response['full_text'] = str(bill_amount) + '$'
response['color'] = i3s_config['color_good']
else:
response['full_text'] = 'Global error - WTF exception'
return response
if __name__ == "__main__":
"""
Test this module by calling it directly.
"""
from time import sleep
x = Py3status()
config = {
'color_good': '#00FF00',
'color_bad': '#FF0000',
}
while True:
print(x.aws_bill([], config))
sleep(1)
|
