Mercurial > repos > shellac > guppy_basecaller
comparison env/bin/elbadmin @ 0:26e78fe6e8c4 draft
"planemo upload commit c699937486c35866861690329de38ec1a5d9f783"
author | shellac |
---|---|
date | Sat, 02 May 2020 07:14:21 -0400 |
parents | |
children |
comparison
equal
deleted
inserted
replaced
-1:000000000000 | 0:26e78fe6e8c4 |
---|---|
1 #!/Users/pldms/Development/Projects/2020/david-matthews-galaxy/guppy_basecaller/env/bin/python3 | |
2 # Copyright (c) 2009 Chris Moyer http://coredumped.org/ | |
3 # | |
4 # Permission is hereby granted, free of charge, to any person obtaining a | |
5 # copy of this software and associated documentation files (the | |
6 # "Software"), to deal in the Software without restriction, including | |
7 # without limitation the rights to use, copy, modify, merge, publish, dis- | |
8 # tribute, sublicense, and/or sell copies of the Software, and to permit | |
9 # persons to whom the Software is furnished to do so, subject to the fol- | |
10 # lowing conditions: | |
11 # | |
12 # The above copyright notice and this permission notice shall be included | |
13 # in all copies or substantial portions of the Software. | |
14 # | |
15 # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS | |
16 # OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABIL- | |
17 # ITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT | |
18 # SHALL THE AUTHOR BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, | |
19 # WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, | |
20 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS | |
21 | |
22 # | |
23 # Elastic Load Balancer Tool | |
24 # | |
25 VERSION = "0.2" | |
26 usage = """%prog [options] [command] | |
27 Commands: | |
28 list|ls List all Elastic Load Balancers | |
29 delete <name> Delete ELB <name> | |
30 get <name> Get all instances associated with <name> | |
31 create <name> Create an ELB; -z and -l are required | |
32 add <name> <instances> Add <instances> in ELB <name> | |
33 remove|rm <name> <instances> Remove <instances> from ELB <name> | |
34 reap <name> Remove terminated instances from ELB <name> | |
35 enable|en <name> <zone> Enable Zone <zone> for ELB <name> | |
36 disable <name> <zone> Disable Zone <zone> for ELB <name> | |
37 addl <name> Add listeners (specified by -l) to the ELB | |
38 <name> | |
39 rml <name> <port> Remove Listener(s) specified by the port on | |
40 the ELB <name> | |
41 """ | |
42 | |
43 | |
44 def find_elb(elb, name): | |
45 try: | |
46 elbs = elb.get_all_load_balancers(name) | |
47 except boto.exception.BotoServerError as se: | |
48 if se.code == 'LoadBalancerNotFound': | |
49 elbs = [] | |
50 else: | |
51 raise | |
52 | |
53 if len(elbs) < 1: | |
54 print "No load balancer by the name of %s found" % name | |
55 return None | |
56 elif len(elbs) > 1: | |
57 print "More than one elb matches %s?" % name | |
58 return None | |
59 | |
60 # Should not happen | |
61 if name not in elbs[0].name: | |
62 print "No load balancer by the name of %s found" % name | |
63 return None | |
64 | |
65 return elbs[0] | |
66 | |
67 | |
68 def list(elb): | |
69 """List all ELBs""" | |
70 print "%-20s %s" % ("Name", "DNS Name") | |
71 print "-" * 80 | |
72 for b in elb.get_all_load_balancers(): | |
73 print "%-20s %s" % (b.name, b.dns_name) | |
74 | |
75 def check_valid_region(conn, region): | |
76 if conn is None: | |
77 print 'Invalid region (%s)' % region | |
78 sys.exit(1) | |
79 | |
80 def get(elb, name): | |
81 """Get details about ELB <name>""" | |
82 | |
83 b = find_elb(elb, name) | |
84 if b: | |
85 print "=" * 80 | |
86 print "Name: %s" % b.name | |
87 print "DNS Name: %s" % b.dns_name | |
88 if b.canonical_hosted_zone_name: | |
89 chzn = b.canonical_hosted_zone_name | |
90 print "Canonical hosted zone name: %s" % chzn | |
91 if b.canonical_hosted_zone_name_id: | |
92 chznid = b.canonical_hosted_zone_name_id | |
93 print "Canonical hosted zone name id: %s" % chznid | |
94 print | |
95 | |
96 print "Health Check: %s" % b.health_check | |
97 print | |
98 | |
99 print "Listeners" | |
100 print "---------" | |
101 print "%-8s %-8s %s" % ("IN", "OUT", "PROTO") | |
102 for l in b.listeners: | |
103 print "%-8s %-8s %s" % (l[0], l[1], l[2]) | |
104 | |
105 print | |
106 | |
107 print " Zones " | |
108 print "---------" | |
109 for z in b.availability_zones: | |
110 print z | |
111 | |
112 print | |
113 | |
114 # Make map of all instance Id's to Name tags | |
115 import boto | |
116 from boto.compat.six import iteritems | |
117 if not options.region: | |
118 ec2 = boto.connect_ec2() | |
119 else: | |
120 ec2 = boto.ec2.connect_to_region(options.region) | |
121 check_valid_region(ec2, options.region) | |
122 | |
123 instance_health = b.get_instance_health() | |
124 instances = [state.instance_id for state in instance_health] | |
125 | |
126 names = dict((k,'') for k in instances) | |
127 for i in ec2.get_only_instances(): | |
128 if i.id in instances: | |
129 names[i.id] = i.tags.get('Name', '') | |
130 | |
131 name_column_width = max([4] + [len(v) for k,v in iteritems(names)]) + 2 | |
132 | |
133 print "Instances" | |
134 print "---------" | |
135 print "%-12s %-15s %-*s %s" % ("ID", | |
136 "STATE", | |
137 name_column_width, "NAME", | |
138 "DESCRIPTION") | |
139 for state in instance_health: | |
140 print "%-12s %-15s %-*s %s" % (state.instance_id, | |
141 state.state, | |
142 name_column_width, names[state.instance_id], | |
143 state.description) | |
144 | |
145 print | |
146 | |
147 | |
148 def create(elb, name, zones, listeners): | |
149 """Create an ELB named <name>""" | |
150 l_list = [] | |
151 for l in listeners: | |
152 l = l.split(",") | |
153 if l[2] == 'HTTPS': | |
154 l_list.append((int(l[0]), int(l[1]), l[2], l[3])) | |
155 else: | |
156 l_list.append((int(l[0]), int(l[1]), l[2])) | |
157 | |
158 b = elb.create_load_balancer(name, zones, l_list) | |
159 return get(elb, name) | |
160 | |
161 | |
162 def delete(elb, name): | |
163 """Delete this ELB""" | |
164 b = find_elb(elb, name) | |
165 if b: | |
166 b.delete() | |
167 print "Load Balancer %s deleted" % name | |
168 | |
169 | |
170 def add_instances(elb, name, instances): | |
171 """Add <instance> to ELB <name>""" | |
172 b = find_elb(elb, name) | |
173 if b: | |
174 b.register_instances(instances) | |
175 return get(elb, name) | |
176 | |
177 | |
178 def remove_instances(elb, name, instances): | |
179 """Remove instance from elb <name>""" | |
180 b = find_elb(elb, name) | |
181 if b: | |
182 b.deregister_instances(instances) | |
183 return get(elb, name) | |
184 | |
185 | |
186 def reap_instances(elb, name): | |
187 """Remove terminated instances from elb <name>""" | |
188 b = find_elb(elb, name) | |
189 if b: | |
190 for state in b.get_instance_health(): | |
191 if (state.state == 'OutOfService' and | |
192 state.description == 'Instance is in terminated state.'): | |
193 b.deregister_instances([state.instance_id]) | |
194 return get(elb, name) | |
195 | |
196 | |
197 def enable_zone(elb, name, zone): | |
198 """Enable <zone> for elb""" | |
199 b = find_elb(elb, name) | |
200 if b: | |
201 b.enable_zones([zone]) | |
202 return get(elb, name) | |
203 | |
204 | |
205 def disable_zone(elb, name, zone): | |
206 """Disable <zone> for elb""" | |
207 b = find_elb(elb, name) | |
208 if b: | |
209 b.disable_zones([zone]) | |
210 return get(elb, name) | |
211 | |
212 | |
213 def add_listener(elb, name, listeners): | |
214 """Add listeners to a given load balancer""" | |
215 l_list = [] | |
216 for l in listeners: | |
217 l = l.split(",") | |
218 l_list.append((int(l[0]), int(l[1]), l[2])) | |
219 b = find_elb(elb, name) | |
220 if b: | |
221 b.create_listeners(l_list) | |
222 return get(elb, name) | |
223 | |
224 | |
225 def rm_listener(elb, name, ports): | |
226 """Remove listeners from a given load balancer""" | |
227 b = find_elb(elb, name) | |
228 if b: | |
229 b.delete_listeners(ports) | |
230 return get(elb, name) | |
231 | |
232 | |
233 if __name__ == "__main__": | |
234 try: | |
235 import readline | |
236 except ImportError: | |
237 pass | |
238 import boto | |
239 import sys | |
240 from optparse import OptionParser | |
241 from boto.mashups.iobject import IObject | |
242 parser = OptionParser(version=VERSION, usage=usage) | |
243 parser.add_option("-z", "--zone", | |
244 help="Operate on zone", | |
245 action="append", default=[], dest="zones") | |
246 parser.add_option("-l", "--listener", | |
247 help="Specify Listener in,out,proto", | |
248 action="append", default=[], dest="listeners") | |
249 parser.add_option("-r", "--region", | |
250 help="Region to connect to", | |
251 action="store", dest="region") | |
252 | |
253 (options, args) = parser.parse_args() | |
254 | |
255 if len(args) < 1: | |
256 parser.print_help() | |
257 sys.exit(1) | |
258 | |
259 if not options.region: | |
260 elb = boto.connect_elb() | |
261 else: | |
262 import boto.ec2.elb | |
263 elb = boto.ec2.elb.connect_to_region(options.region) | |
264 check_valid_region(elb, options.region) | |
265 | |
266 print "%s" % (elb.region.endpoint) | |
267 | |
268 command = args[0].lower() | |
269 if command in ("ls", "list"): | |
270 list(elb) | |
271 elif command == "get": | |
272 get(elb, args[1]) | |
273 elif command == "create": | |
274 if not options.listeners: | |
275 print "-l option required for command create" | |
276 sys.exit(1) | |
277 if not options.zones: | |
278 print "-z option required for command create" | |
279 sys.exit(1) | |
280 create(elb, args[1], options.zones, options.listeners) | |
281 elif command == "delete": | |
282 delete(elb, args[1]) | |
283 elif command in ("add", "put"): | |
284 add_instances(elb, args[1], args[2:]) | |
285 elif command in ("rm", "remove"): | |
286 remove_instances(elb, args[1], args[2:]) | |
287 elif command == "reap": | |
288 reap_instances(elb, args[1]) | |
289 elif command in ("en", "enable"): | |
290 enable_zone(elb, args[1], args[2]) | |
291 elif command == "disable": | |
292 disable_zone(elb, args[1], args[2]) | |
293 elif command == "addl": | |
294 if not options.listeners: | |
295 print "-l option required for command addl" | |
296 sys.exit(1) | |
297 add_listener(elb, args[1], options.listeners) | |
298 elif command == "rml": | |
299 if not args[2:]: | |
300 print "port required" | |
301 sys.exit(2) | |
302 rm_listener(elb, args[1], args[2:]) |