Filter instances by project. Also stop and start instances.

Author: qualitycodebase

README.md

@@ -10,4 +10,7 @@ In my case "shotty" or anyname you provide with config file, uses the configurat
 `aws configure --profile shotty`
 
 ## How to run it
-`pipenv run python shotty/shotty.py`
+`pipenv run python shotty/shotty.py <command> <--project=PROJECT>`
+
+*command* is list, start, or stop
+*project* is optional

shotty/shotty.py

@@ -4,10 +4,29 @@
 session = boto3.Session(profile_name='shotty')
 ec2 = session.resource('ec2')
 
-@click.command()
-def list_instances():
+def filter_instances(project):
+    instances = []
+
+    if project:
+        filters = [{'Name':'tag:Project', 'Values':[project]}]
+        instances = ec2.instances.filter(Filters=filters)
+    else:
+        instances = ec2.instances.all()
+
+    return instances
+
+@click.group()
+def instances():
+    """Commands for instances"""
+@instances.command('list')
+@click.option('--project', default=None,
+     help="Only instances for project (tag Project:<name>)")
+def list_instances(project):
     "List EC2 instances"
-    for i in ec2.instances.all():
+
+    instances = filter_instances(project)
+
+    for i in instances:
         print(', '.join((
         i.id,
         i.instance_type,
@@ -16,5 +35,31 @@ def list_instances():
         i.public_dns_name)))
     return
 
+@instances.command('stop')
+@click.option('--project', default=None,
+  help='Only instances for project')
+def stop_instances(project):
+    "Stop EC2 instances"
+
+    instances = filter_instances(project)
+
+    for i in instances:
+        print("Stopping {0}...".format(i.id))
+        i.stop()
+    return
+
+@instances.command('start')
+@click.option('--project', default=None,
+  help='Only instances for project')
+def start_instances(project):
+    "Stop EC2 instances"
+
+    instances = filter_instances(project)
+
+    for i in instances:
+        print("Starting {0}...".format(i.id))
+        i.start()
+    return
+
 if __name__ == '__main__':
-    list_instances()
+    instances()