Fabric Config for Starting / Stopping EC2 Instances with SSH and Jupyter Access

fabfile.py

# import fabrics API functions
from fabric.api import *
from requests import get

# Update with your security group and instance ID
SECURITY_GROUP = "sg-9bdc1fd0"
INSTANCE_ID = "i-06c072747655b16e5"


def modify_security_group(enable=False):
    ip = get('https://api.ipify.org').text
    CIDR = ip + "/32"
    if enable:
        aws_command = "aws ec2 authorize-security-group-ingress --group-id " + SECURITY_GROUP
    else:
        aws_command = "aws ec2 revoke-security-group-ingress --group-id  " + SECURITY_GROUP
    aws_command += "  --protocol tcp --port 22 --cidr " + CIDR
    local(aws_command)
    if enable:
        aws_command = "aws ec2 authorize-security-group-ingress --group-id " + SECURITY_GROUP
    else:
        aws_command = "aws ec2 revoke-security-group-ingress --group-id  " + SECURITY_GROUP
    aws_command += "  --protocol tcp --port 8888 --cidr " + CIDR
    local(aws_command)


def enable_access():
    modify_security_group(enable=True)


def disable_access():
    modify_security_group(enable=False)


def start_work():
    local("aws ec2 start-instances --instance-ids " + INSTANCE_ID)
    enable_access()


def stop_work():
    local("aws ec2 stop-instances --instance-ids " + INSTANCE_ID)
    disable_access()


def get_status():
    local("aws ec2 describe-instances --instance-ids " + INSTANCE_ID)