统计AWS的ec2的ReservedInstance使用情况

1. #!/usr/bin/env python
   
2. # -*- coding: utf-8 -*-
   
3. #
   
4. # Date: Jul 25,2016
   
5. 
   
6. '''
   
7. ### Purpose ####
   
8. 
   
9. Correlating the reservations you currently have active with your running instances is a manual, time-consuming, and error prone process.
   
10. This quick little Python script uses boto3 to inspect your reserved instances and running instances to determine if you currently have any reserved instances which are not being used. Additionally, it will give you a list of non-reserved instances which could benefit from additional reserved instance allocations.
    
11. 
    
12. ### prequisite ####
    
13. 
    
14. 1. requirement(optional): boto3 and termcolor
    
15. $ pip install boto3 termcolor (maybe need sudo privilege)
    
16. 
    
17. python >= 2.7
    
18. 
    
19. 2. aws credentials configuration
    
20. See here http://docs.aws.amazon.com/cli/latest/userguide/cli-chap-getting-started.html
    
21. If you have multiple profiles, you can configure additional, named profiles by using the --profile option.
    
22. 
    
23. $ aws configure --profile pcf-China
    
24. AWS Access Key ID [None]: xxxxxxxxxxxxxxxxxxxxxxxx
    
25. AWS Secret Access Key [None]: xxxxxxxxxxxxxxxxxxxxx
    
26. Default region name [None]: cn-north-1
    
27. Default output format [None]: text
    
28. 
    
29. ##### something else that need your attention !!!!!!
    
30. 
    
31. 1. if your Reserved Instance account and Running Instance account are same one, probably you don't need to specify a profile name as below, yet you need to re-write the code under
    
32. 
    
33. the initialization of an instance will be :
    
34. 
    
35. x = Ec2()
36. 
    
37. which means a profile name isn't required,it will read from the `default` credentials.
    
38. 
    
39. 2. Probably your profile name is different from mine, please modify them on below section
    
40. 
    
41. ### aws profile specifications ###
    
42. 
    
43. in this script, my reserve profile is reservation, and running instance profile is pcf-China and concourse.
    
44. 
    
45. '''
    
46. try:
    
47.    import sys
    
48.    import boto3
    
49.    from termcolor import colored
    
50. except ImportError:
    
51.    print "either boto3 or termcolor isn't installed, please run \"sudo pip install boto3 termcolor\""
    
52.    sys.exit(1)
    
53. 
    
54. ### aws profile specifications ###
    
55. 
    
56. ri_acc_id = ["reservation"]
    
57. rn_acc_id = ["pcf-China", "concourse"]
    
58. 
    
59. class Ec2 :
    
60.     def __init__(self, profile=None) :
    
61.         self.profile = profile
    
62.         self.session = boto3.session.Session(profile_name = self.profile )
    
63.         self.client = self.session.client("ec2")
    
64.         self.RI = {}
    
65.         self.running_instances = {}
    
66. 
    
67.     def Current_ec2_instances(self) :
    
68. 
    
69.         for i in self.client.describe_instances(Filters=[{'Name': 'instance-state-name', 'Values': ['running','stopped']}])["Reservations"] :
    
70.             tmp = i["Instances"]
    
71.             az_type = tmp[0]["Placement"]["AvailabilityZone"],tmp[0]["InstanceType"]
    
72.             self.running_instances[az_type]=self.running_instances.get(az_type,0) + 1
    
73.         # current ec2 instances, including running/stopped, excluding pending/terminated.
    
74.         #print self.running_instances
    
75.         return self.running_instances
    
76. 
    
77.     def RI_ec2(self) :
    
78. 
    
79.         target = self.client.describe_reserved_instances()["ReservedInstances"]
    
80.         for i in target :
    
81.             # current reserved ec2 instances, excluding pending-payment | active | payment-failed | retired
    
82.             if i["State"] == "active" :
    
83.                 az_type = i["AvailabilityZone"],i["InstanceType"]
    
84.                 self.RI[az_type] = i["InstanceCount"]
    
85.         #print self.RI
    
86.         return self.RI
    
87. 
    
88. #####################################################
    
89. 
    
90. 
    
91. def do_calc(ri,running) :
    
92. 
    
93.     for key in ri :
    
94.         balance = ri[key] - running.get(key,0) # asumme it's zero, if key doesn't appear on RI
    
95.         if balance < 0 :
    
96.             print "%s\t\t%s" %(key,colored(balance,"red"))
    
97.         else :
    
98.             print "%s\t\t%d" %(key,balance)
    
99.         try :
    
100.             del running[key]
     
101.         except Exception,e :
     
102.             pass
     
103.     print "-" * 60
     
104.     print colored("Region\t\tInstanceType\t\treserved-instance-need-to-buy","red")
     
105.     print "-" * 60
     
106.     if running.items() != {}:
     
107.         for k,v in running.items() :
     
108.             print "%s\t\t%d" % (k,v)
     
109. 
     
110. if __name__ == "__main__" :
     
111. 
     
112. ### ec2 ###
     
113.     print "#" * 33
     
114.     print "# EC2 statistic #"
     
115.     print "#" * 33
     
116.    # ec2 statistics
     
117.     ri_instance_sum ={}
     
118.     for i in ri_acc_id :
     
119.         reserve_ins = Ec2(i)
     
120.         y = reserve_ins.RI_ec2()
     
121.         # print "### reserved instances details : %s ###" %colored(i,"green")
     
122.         # print "-" * 60
     
123.         #print colored("Region\t\tInstanceType\t\tAmount","green")
     
124.         #print "-" * 60
     
125.         for k,v in y.items():
     
126.             #print "%s\t\t%d" % (k,v)
     
127.             if k not in ri_instance_sum :
     
128.                 ri_instance_sum[k] = v
     
129.             else :
     
130.                 ri_instance_sum[k] = ri_instance_sum[k] + v
     
131. 
     
132.     rn_instance_sum = {}
     
133.     for i in rn_acc_id :
     
134.         x = Ec2(i)
     
135.         #print "\n### running instances details : %s ###" %colored(i,"green")
     
136.         #print "-" * 60
     
137.         #print colored("Region\t\tInstanceType\t\tAmount","green")
     
138.         #print "-" * 60
     
139.         for k,v in x.Current_ec2_instances().items() :
     
140. 
     
141.             #print "%s\t\t%d"%(k,v)
     
142. 
     
143.             if k not in rn_instance_sum:
     
144.                 rn_instance_sum[k] =v
     
145.             else :
     
146.                 rn_instance_sum[k] = rn_instance_sum[k] + v
     
147. 
     
148.     print "-" * 60
     
149.     print colored("Region\t\tInstanceType\t\tBalance","green")
     
150.     print "-" * 60
     
151.     print colored("\npositive integer means reserved instances remains un-used,while negative integer \nmeans this type of reserved instance is out of capacity, needs to purchase more...","yellow")
     
152.     print "-" * 60
     
153. 
     
154.     do_calc(ri_instance_sum,rn_instance_sum)
     
155. 
     

主要是熟悉AWS的Python的boto3。非常的强大，当然也希望官方能给出统计，这样就省得麻烦了。