aws_tool.py

#!/usr/bin/env python
# tool to automate various AWS commands
# TODO: transition to argparse

import boto3
import sys
import os
import time

from operator import itemgetter

import util as u

LIMIT_TO_KEY='nexus'

def list_instances():
  print("Region", u.get_region())
  ec2 = u.create_ec2_resource()
  instances = [(u.seconds_from_datetime(i.launch_time), i) for i in ec2.instances.all()]
  sorted_instances = sorted(instances, key=itemgetter(0))

  for (seconds, instance) in sorted_instances:
    hours_ago = (time.time()-seconds)/3600
    hours_ago+=8 # adjust for time being in UTC
    if instance.state['Name'] not in ('running', 'terminating'):
      continue
    if not (LIMIT_TO_KEY in instance.key_name):
      continue
    
    print("%4s %20s %10s %20s %s %s" %(int(hours_ago),
                            u.get_name(instance.tags), instance.instance_type,
                            instance.public_ip_address,
                                       instance.private_ip_address,
                                    instance.id
    ))

def get_instance(fragment):
  ec2 = u.create_ec2_resource()
  instances = [(u.seconds_from_datetime(i.launch_time), i) for i in ec2.instances.all()]
  # latest instance first
  sorted_instances = reversed(sorted(instances, key=itemgetter(0)))
  for (seconds, instance) in sorted_instances:
    name = u.get_name(instance.tags)
    if fragment in u.get_name(instance.tags):
      hours_ago = (time.time()-seconds)/3600
      hours_ago+=8 # adjust for time being in UTC
      print("Found instance %s launched %.1f hours ago" %(name, hours_ago,))
      return instance
  print("Found nothing matching", fragment)
  
def main():
  if len(sys.argv) < 2:
    mode = 'list'
  else:
    mode = sys.argv[1]

  if mode == 'list' or mode == 'ls':
    list_instances()
  elif mode == 'reboot':
    task_fragment = sys.argv[2]
    instance = get_instance(task_fragment)
    print("Rebooting "+u.get_name(instance.tags))
    instance.reboot()
  else:
    assert False, "Unknown mode "+mode
      
if __name__=='__main__':
  main()