diff options
Diffstat (limited to 'logster/logster_helper.py')
| -rw-r--r-- | logster/logster_helper.py | 58 |
1 files changed, 58 insertions, 0 deletions
diff --git a/logster/logster_helper.py b/logster/logster_helper.py index c8b3ca1..8846c8f 100644 --- a/logster/logster_helper.py +++ b/logster/logster_helper.py @@ -19,6 +19,13 @@ ### along with Logster. If not, see <http://www.gnu.org/licenses/>. ### +import os +import httplib +import base64 +import hashlib +import hmac + +from urllib import urlencode, quote_plus from time import time class MetricObject(object): @@ -51,3 +58,54 @@ class LockingError(Exception): """ Exception raised for errors creating or destroying lockfiles. """ pass +class CloudWatch: + """ Specify Amazon CloudWatch params """ + def __init__(self, key, secret_key, metric): + conn = httplib.HTTPConnection("169.254.169.254") + conn.request("GET", "/latest/meta-data/instance-id") + instance_id = conn.getresponse().read() + print "instance-id = %s" % instance_id + + self.base_url = "monitoring.ap-northeast-1.amazonaws.com" + self.key = os.getenv('AWS_ACCESS_KEY_ID', key) + self.secret_key = os.getenv('AWS_SECRET_ACCESS_KEY_ID', secret_key) + self.params = {'Namespace': 'logster', + 'MetricData.member.1.MetricName': metric.name, + 'MetricData.member.1.Value': metric.value, + 'MetricData.member.1.Unit': metric.units, + 'MetricData.member.1.Dimensions.member.1.Name': 'InstanceID', + 'MetricData.member.1.Dimensions.member.1.Value': instance_id} + + self.url_params = self.params + self.url_params['AWSAccessKeyId'] = self.key + self.url_params['Action'] = 'PutMetricData' + self.url_params['SignatureMethod'] = 'HmacSHA256' + self.url_params['SignatureVersion'] = '2' + self.url_params['Version'] = '2010-08-01' + self.url_params['Timestamp'] = metric.timestamp + + def get_signed_url(self): + """ build signed parameters following http://docs.amazonwebservices.com/AmazonCloudWatch/latest/APIReference/API_PutMetricData.html """ + keys = self.url_params.keys() + keys.sort() + values = map(self.url_params.get, keys) + url_string = urlencode(zip(keys,values)) + + string_to_sign = "GET\n%s\n/\n%s" % (self.base_url, url_string) + signature = hmac.new( key=self.secret_key, msg=string_to_sign, digestmod=hashlib.sha256).digest() + signature = base64.encodestring(signature).strip() + urlencoded_signature = quote_plus(signature) + url_string += "&Signature=%s" % urlencoded_signature + + return "/?" + url_string + + def put_data(self): + signedURL = self.get_signed_url() + print signedURL + conn = httplib.HTTPConnection(self.base_url) + conn.request("GET", signedURL) + res = conn.getresponse() + print res.status, res.reason + print res.getheaders() + print res.read() + |