云服务器CVM批量关机脚本-Python

#! /usr/bin/env python
# -*- coding: utf-8 -*-

# 本工具主要功能：批量关闭云服务器CVM

# region的地域列表：https://cloud.tencent.com/document/api/214/30670#.E5.9C.B0.E5.9F.9F.E5.88.97.E8.A1.A8

import json, sys ,os, time, datetime
from tencentcloud.common import credential
from tencentcloud.common.profile.client_profile import ClientProfile
from tencentcloud.common.profile.http_profile import HttpProfile
from tencentcloud.common.exception.tencent_cloud_sdk_exception import TencentCloudSDKException
from tencentcloud.cvm.v20170312 import cvm_client, models

reload(sys)
sys.setdefaultencoding("utf-8")

now= str(datetime.datetime.now().strftime("%Y 年 %m 月 %d 日  -  %H:%M:%S"))
all_region = ['ap-bangkok','ap-beijing','ap-chengdu','ap-chongqing','ap-guangzhou','ap-hongkong','ap-mumbai',
          'ap-nanjing','ap-seoul','ap-shanghai','ap-shanghai-fsi','ap-shenzhen-fsi','ap-singapore',
          'ap-tokyo','eu-frankfurt','eu-moscow','na-ashburn','na-siliconvalley','na-toronto']

def print_help():
    print"""
输入 ./shutdown.py help 来查看如何使用此工具

关闭云服务器CVM的命令: ./shutdown.py shut <访问密钥，以Secretld 空格 SecreKey 的格式 保存文本即可> <需关机的云服务器IP地址 的格式 保存文本即可>

#温馨提示：此工具会在当前目录下生成关机操作的日志文件
"""
def cvm_shutdown(Secretld,SecreKey,region,InstanceId):

    try:
        cred = credential.Credential(SecretId,SecretKey)
        httpProfile = HttpProfile()
        httpProfile.endpoint = "cvm.tencentcloudapi.com"

        clientProfile = ClientProfile()
        clientProfile.httpProfile = httpProfile
        client = cvm_client.CvmClient(cred, region, clientProfile) 
       
        req = models.StopInstancesRequest()
        params = {
        "InstanceIds": [ str(InstanceId) ]
        # "ForceStop": True, 是否强制关机 此参数选填
        # "StopType": "SOFT_FIRST", 关闭实例的模式 此参数选填
        # "StoppedMode": "STOP_CHARGING" 是否开启关机不收费 此参数选填

        }

        req.from_json_string(json.dumps(params))

        resp = client.StopInstances(req) 
        return resp.to_json_string()

    except TencentCloudSDKException as err: 
        return err

def _ID_region_from_eip(SecretId,SecretKey,region,offset):

	try:
		cred = credential.Credential(SecretId, SecretKey)
		httpProfile = HttpProfile()
		httpProfile.endpoint = "cvm.tencentcloudapi.com"

		clientProfile = ClientProfile()
		clientProfile.httpProfile = httpProfile
		client = cvm_client.CvmClient(cred, region, clientProfile)

		req = models.DescribeInstancesRequest()
		params = {
			"Limit": 20,
			"Offset": offset
		}
		req.from_json_string(json.dumps(params))

		resp = client.DescribeInstances(req)
		return resp.to_json_string()

	except TencentCloudSDKException as err:
		return err

def log_log(log_content, path):
	
	print "\033[0;31m 本工具最近一次运行的时间: "+now +'\033[0m \n\n'
	
	fin=open(path,'a') # 追加原始内容的写入
	content = log_content + "\033[0;31m 本工具最近一次运行的时间: "+now +'\033[0m \n\n'
	fin.write(content)
	fin.close()

if __name__ == '__main__':
	try:
		if_sys_argv_exit = sys.argv[1]
	except Exception:
		print_help()
		exit()

	if sys.argv[1] == 'help':
		print_help()
		exit()

	else:
		font = open('%s'%sys.argv[2])
		font = font.readline()
		token = font.split()
		SecretId = token[0]
		SecretKey = token[1]

		if sys.argv[1] == 'shut':
			try:
				path = sys.argv[4]
			except Exception:
				path = str(os.path.dirname(os.path.abspath(__file__))) + "/shutdown_cvm.log"

			content = ''
			ip_dict={ }

			for city in all_region:
				result = _ID_region_from_eip(SecretId,SecretKey,city,0)

				try:
					result_dict = json.loads(result)
				except Exception:
					print result
					print "because of above reason, this tool is crashed, please try later"
					content += str(result) +'\n'
					content += "because of above reason, this tool is crashed, please try later"
					log_log(content,path)
					exit()

				TotalCount = result_dict['TotalCount'] #本次查询到了多少CVM

				left_times = TotalCount/20 + 1
				offset = 0

				while left_times>0:
					result = _ID_region_from_eip(SecretId,SecretKey,city,offset)
					try:
						result_dict = json.loads(result)
					except Exception:
						print result
						print "because of above reason, this tool is crashed, please try later"
						content += str(result) + '\n'
						content += "because of above reason, this tool is crashed, please try later"
						log_log(content,path)
						exit()


					for i in result_dict['InstanceSet']:
						InstanceId = i['InstanceId']

						try:
							TF_eip = i['PublicIpAddresses'][0]
						except Exception:
							continue

						for j_ip in i['PublicIpAddresses']:
							eip = j_ip
							ip_dict[eip] = [ InstanceId  , city ]

					left_times -= 1
					offset += 20

			#print ip_dict

			fout=open('%s'%sys.argv[3])
			while 1:
				time.sleep(0.12)
				line = fout.readline()
				if not line :
					break

				line  = line.strip()
				try:
					top_word = line[0]
				except Exception:
					continue

				if top_word=='#':
					continue

				all_conf = line.split()
				ip = all_conf[0]

				if '.' not in all_conf[0]:
					print line + ' 当前帐号下找不到该服务器 '
					print ' '
					content += line + ' 当前帐号下找不到该服务器 查询时间: ' + now + '\n\n'
					continue

				if ip not in ip_dict:
					print line + ' 当前帐号下找不到该服务器 '
					print ' '
					content +=  line + ' 当前帐号下找不到该服务器 查询时间: '+ now + '\n\n'
					continue

				result = cvm_shutdown(SecretId,SecretKey,region=ip_dict[ip][1],InstanceId=ip_dict[ip][0])

				try:
					result_dict = json.loads(result)
					RequestId = result_dict['RequestId']
					print ' 云服务器IP ' + line + ' ' + ip_dict[ip][1] + ' 地域下 ' + ' 实例ID为 '+ ip_dict[ip][0] + ' 的云服务器关机成功 '
					print ' '
					content += ' 云服务器IP ' + line + ' ' + ip_dict[ip][1] + ' 地域下 ' + ' 实例ID为 '+ ip_dict[ip][0] + ' 的云服务器在 ' + now + ' 点 关机成功' + '\n\n'
					
				except Exception:
					print line + ' '+ip_dict[ip][0]+' '+ip_dict[ip][1]+ ' is fail '
					print result
					print ' '
					content +=  line + ' '+ip_dict[ip][0]+' '+ip_dict[ip][1]+ ' is fail time:' + now + '\n' + str(result) + '\n\n'
					
			log_log(content,path)
		else:
			print '\n'
			print ' 当前目录下没有找到这个 %s ' % sys.argv[1] + ' 这个文件 '
			print '\n'
			print_help()