import sys
import boto3

DEV_ID = 'XXX'
DEV2_ID = 'XXX'

mode = sys.argv[1]
instance_name = sys.argv[2]
if instance_name == 'dev':
    instance_ids = [DEV_ID]
elif instance_name == 'dev2':
    instance_ids = [DEV2_ID]
elif instance_name == 'all':
    instance_ids = [DEV_ID, DEV2_ID]

client = boto3.client('ec2', region_name='ap-northeast-1')
if mode == 'check':
    res = client.describe_instances(InstanceIds=instance_ids)
    for r in res['Reservations']:
        i = r['Instances'][0]
        if i['InstanceId'] == DEV_ID:
            print('dev', i['State'])
        elif i['InstanceId'] == DEV2_ID:
            print('kaggle', i['State'])
elif mode == 'start':
    client.start_instances(InstanceIds=instance_ids)
elif mode == 'stop':
    client.stop_instances(InstanceIds=instance_ids)