AlibabaCloud ECSを定期的に自動再起動する設定方法

# coding=utf-8
# If the Python sdk is not installed, run 'sudo pip install aliyun-python-sdk-ecs'.
# Make sure you're using the latest sdk version.
# Run 'sudo pip install --upgrade aliyun-python-sdk-ecs' to upgrade.

import json
import sys
import base64
import time
import logging
from aliyunsdkcore.client import AcsClient
from aliyunsdkcore.acs_exception.exceptions import ClientException
from aliyunsdkcore.acs_exception.exceptions import ServerException
from aliyunsdkecs.request.v20140526.RunCommandRequest import RunCommandRequest
from aliyunsdkecs.request.v20140526.DescribeInvocationResultsRequest import DescribeInvocationResultsRequest
from aliyunsdkecs.request.v20140526.RebootInstancesRequest import RebootInstancesRequest
from aliyunsdkecs.request.v20140526.DescribeInstancesRequest import DescribeInstancesRequest

# Configure the log output formatter
logging.basicConfig(level=logging.INFO,
                    format="%(asctime)s %(name)s [%(levelname)s]: %(message)s",
                    datefmt='%m-%d %H:%M')

logger = logging.getLogger()

access_key = '<yourAccessKey ID>'  # The AccessKey ID you obtained.
access_key_secret = '<yourAccessKey Secret>'  # The AccessKey secret you obtained.
region_id = '<yourRegionId>'  # The region ID you obtained.

client = AcsClient(access_key, access_key_secret, region_id)

def base64_decode(content, code='utf-8'):
    if sys.version_info.major == 2:
        return base64.b64decode(content)
    else:
        return base64.b64decode(content).decode(code)

def get_invoke_result(invoke_id):
    request = DescribeInvocationResultsRequest()
    request.set_accept_format('json')

    request.set_InvokeId(invoke_id)
    response = client.do_action_with_exception(request)
    response_details = json.loads(response)["Invocation"]["InvocationResults"]["InvocationResult"]
    dict_res = { detail.get("InstanceId",""):{"status": detail.get("InvocationStatus",""),"output":base64_decode(detail.get("Output",""))}  for detail in response_details }
    return dict_res

def get_instances_status(instance_ids):
    request = DescribeInstancesRequest()
    request.set_accept_format('json')
    request.set_InstanceIds(instance_ids)
    response = client.do_action_with_exception(request)
    response_details = json.loads(response)["Instances"]["Instance"]
    dict_res = { detail.get("InstanceId",""):{"status":detail.get("Status","")} for detail in response_details }
    return dict_res

def run_command(cmdtype,cmdcontent,instance_ids,timeout=60):
    """
    cmdtype: the command type, which can be RunBatScript, RunPowerShellScript, or RunShellScript.
    cmdcontent: the command content.
    instance_ids: the IDs of the instances on which you want to run the command.
    """
    try:
        request = RunCommandRequest()
        request.set_accept_format('json')

        request.set_Type(cmdtype)
        request.set_CommandContent(cmdcontent)
        request.set_InstanceIds(instance_ids)
        # The timeout period for running the command. Unit: seconds. Default value: 60. Specify this parameter based on the command that you want to run.
        request.set_Timeout(timeout)
        response = client.do_action_with_exception(request)
        invoke_id = json.loads(response).get("InvokeId")
        return invoke_id
    except Exception as e:
        logger.error("run command failed")

def reboot_instances(instance_ids,Force=False):
    """
    instance_ids: the IDs of the instances that you want to restart.
    Force: specifies whether to forcibly restart the instances. Default value: False.
    """
    request = RebootInstancesRequest()
    request.set_accept_format('json')
    request.set_InstanceIds(instance_ids)
    request.set_ForceReboot(Force)
    response = client.do_action_with_exception(request)

def wait_invoke_finished_get_out(invoke_id,wait_count,wait_interval):
    for i in range(wait_count):
        result = get_invoke_result(invoke_id)
        if set([res["status"] for _,res in result.items()]) & set(["Running","Pending","Stopping"]):
            time.sleep(wait_interval)
        else:
            return result
    return result

def wait_instance_reboot_ready(ins_ids,wait_count,wait_interval):
    for i in range(wait_count):
        result = get_instances_status(ins_ids)
        if set([res["status"] for _,res in result.items()]) != set(["Running"]):
            time.sleep(wait_interval)
        else:
            return result
    return result

def run_task():
    # Specify the type of the command.
    cmdtype = "RunShellScript"
    # Specify the content of the command.
    cmdcontent = """
    #!/bin/bash
    echo helloworld
    """
    # Specify the timeout period.
    timeout = 60
    # Specify the IDs of the instances on which you want to run the command. After the command is run on these instances, these instances are restarted.
    ins_ids= ["i-bp185fcs****","i-bp14wwh****","i-bp13jbr****"]

    # Run the command.
    invoke_id = run_command(cmdtype,cmdcontent,ins_ids,timeout)
    logger.info("run command,invoke-id:%s" % invoke_id)

    # Wait for the command to finishing running. Query the command running state 10 times at an interval of 5 seconds. Specify the number of queries and the query interval based on the actual requirements.
    invoke_result = wait_invoke_finished_get_out(invoke_id,10,5)
    for ins_id,res in invoke_result.items():
        logger.info("instance %s command execute finished,status: %s,output:%s" %(ins_id,res["status"],res["output"]))

    # Restart the instances.
    logger.warn("reboot instance Now")
    reboot_instances(ins_ids)

    time.sleep(5)
    # Wait for the instances to enter the Running state. Query the instance states 30 times at an interval of 10 seconds.
    reboot_result = wait_instance_reboot_ready(ins_ids,30,10)
    logger.warn("reboot instance Finished")
    for ins_id,res in reboot_result.items():
        logger.info("instance %s status: %s" %(ins_id,res["status"]))

if __name__ == '__main__':
    run_task()