Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
#!/usr/bin/python
import sys
import os
import boto
from pprint import pprint
# You can uncomment and set these, or set the env variables AWSAccessKeyId & AWSSecretKey
# AWS_ACCESS_KEY_ID="aaaaaaaaaaaaaaaaaaaa"
# AWS_SECRET_ACCESS_KEY="bbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbb"
try:
AWS_ACCESS_KEY_ID
except NameError:
try:
AWS_ACCESS_KEY_ID=os.environ['AWSAccessKeyId']
AWS_SECRET_ACCESS_KEY=os.environ['AWSSecretKey']
except KeyError:
print "Please set env variable"
sys.exit(1)
ec2_conn = boto.connect_ec2(AWS_ACCESS_KEY_ID, AWS_SECRET_ACCESS_KEY)
reservations = ec2_conn.get_all_instances()
running_instances = {}
for reservation in reservations:
for instance in reservation.instances:
if instance.state != "running":
sys.stderr.write("Disqualifying instance %s: not running\n" % ( instance.id ) )
elif instance.spot_instance_request_id:
sys.stderr.write("Disqualifying instance %s: spot\n" % ( instance.id ) )
else:
if instance.vpc_id:
print "Does not support vpc yet, please be careful when trusting these results"
else:
az = instance.placement
instance_type = instance.instance_type
running_instances[ (instance_type, az ) ] = running_instances.get( (instance_type, az ) , 0 ) + 1
# pprint( running_instances )
reserved_instances = {}
for reserved_instance in ec2_conn.get_all_reserved_instances():
if reserved_instance.state != "active":
sys.stderr.write( "Excluding reserved instances %s: no longer active\n" % ( reserved_instance.id ) )
else:
az = reserved_instance.availability_zone
instance_type = reserved_instance.instance_type
reserved_instances[( instance_type, az) ] = reserved_instances.get ( (instance_type, az ), 0 ) + reserved_instance.instance_count
# pprint( reserved_instances )
# this dict will have a positive number if there are unused reservations
# and negative number if an instance is on demand
instance_diff = dict([(x, reserved_instances[x] - running_instances.get(x, 0 )) for x in reserved_instances])
# instance_diff only has the keys that were present in reserved_instances. There's probably a cooler way to add a filtered dict here
for placement_key in running_instances:
if not placement_key in reserved_instances:
instance_diff[placement_key] = -running_instances[placement_key]
# pprint ( instance_diff )
unused_reservations = dict((key,value) for key, value in instance_diff.iteritems() if value > 0)
if unused_reservations == {}:
print "Congratulations, you have no unused reservations"
else:
for unused_reservation in unused_reservations:
print "UNUSED RESERVATION!\t(%s)\t%s\t%s" % ( unused_reservations[ unused_reservation ], unused_reservation[0], unused_reservation[1] )
print ""
unreserved_instances = dict((key,-value) for key, value in instance_diff.iteritems() if value < 0)
if unreserved_instances == {}:
print "Congratulations, you have no unreserved instances"
else:
for unreserved_instance in unreserved_instances:
print "Instance not reserved:\t(%s)\t%s\t%s" % ( unreserved_instances[ unreserved_instance ], unreserved_instance[0], unreserved_instance[1] )
qty_running_instances = reduce( lambda x, y: x+y, running_instances.values() )
qty_reserved_instances = reduce( lambda x, y: x+y, reserved_instances.values() )
print "\n(%s) running on-demand instances\n(%s) reservations" % ( qty_running_instances, qty_reserved_instances )
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
You can’t perform that action at this time.