utils.py 9.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247
  1. import logging
  2. import json
  3. import sys
  4. import time
  5. import requests
  6. from aliyunsdkcore.client import AcsClient
  7. from aliyunsdkecs.request.v20140526.RunInstancesRequest import RunInstancesRequest
  8. from aliyunsdkecs.request.v20140526.DescribeInstancesRequest import DescribeInstancesRequest
  9. from aliyunsdkecs.request.v20140526.DescribeNetworkInterfacesRequest import DescribeNetworkInterfacesRequest
  10. from aliyunsdkslb.request.v20140515.DescribeLoadBalancerAttributeRequest import DescribeLoadBalancerAttributeRequest
  11. from aliyunsdkecs.request.v20140526.RunCommandRequest import RunCommandRequest
  12. from aliyunsdkcore.request import CommonRequest
  13. logging.basicConfig(level=logging.INFO,
  14. format='%(asctime)s %(filename)s[line:%(lineno)d] %(levelname)s %(message)s',
  15. datefmt='%a, %d %b %Y %H:%M:%S')
  16. def send_msg_to_feishu(webhook, key_word, msg_text):
  17. """发送消息到飞书"""
  18. headers = {'Content-Type': 'application/json'}
  19. payload_message = {
  20. "msg_type": "text",
  21. "content": {
  22. "text": '{}: {}'.format(key_word, msg_text)
  23. }
  24. }
  25. response = requests.request('POST', url=webhook, headers=headers, data=json.dumps(payload_message))
  26. logging.info(response.text)
  27. def connect_client(access_key_id, access_key_secret, region_id):
  28. """
  29. 初始化账号,连接客户端
  30. :param access_key_id: access key Id, type-string
  31. :param access_key_secret: access key secret, type-string
  32. :param region_id: region_id
  33. :return: clt
  34. """
  35. try:
  36. clt = AcsClient(ak=access_key_id, secret=access_key_secret, region_id=region_id)
  37. return clt
  38. except Exception as e:
  39. # 失败,记录报错信息,发送通知,停止并退出
  40. logging.error(e)
  41. sys.exit()
  42. def build_create_instances_request(image_id, vswitch_id, security_group_id, zone_id, instance_type, instance_name,
  43. disk_size, disk_category, key_pair_name, tags):
  44. """
  45. 购买服务器参数配置
  46. :param image_id: 使用的镜像信息 type-string
  47. :param vswitch_id: 选择的交换机 type-string
  48. :param security_group_id: 当前vpc类型的安全组 type-string
  49. :param zone_id: 服务器所在区域 type-string
  50. :param instance_type: 实例规格 type-string
  51. :param instance_name: 实例命名 type-string
  52. :param disk_size: 磁盘大小,单位:G,type-string
  53. :param disk_category: 磁盘类型 type-string
  54. :param key_pair_name: 密钥对名称 type-string
  55. :param tags: 标签 type-list, eg: [{"Key": "ecs", "Value": "rov-server.prod"}, ...]
  56. :return: request
  57. """
  58. request = RunInstancesRequest()
  59. request.set_ImageId(image_id)
  60. request.set_VSwitchId(vswitch_id)
  61. request.set_SecurityGroupId(security_group_id)
  62. request.set_ZoneId(zone_id)
  63. request.set_InstanceType(instance_type)
  64. request.set_InstanceName(instance_name)
  65. request.set_SystemDiskSize(disk_size)
  66. request.set_SystemDiskCategory(disk_category)
  67. request.set_KeyPairName(key_pair_name)
  68. request.set_Tags(tags)
  69. return request
  70. def send_request(client, request):
  71. """
  72. 发送API请求
  73. :param client: 客户端连接
  74. :param request: 请求配置
  75. :return: response
  76. """
  77. request.set_accept_format('json')
  78. try:
  79. response = client.do_action_with_exception(request)
  80. response = json.loads(response)
  81. logging.info(response)
  82. return response
  83. except Exception as e:
  84. # 失败,记录报错信息,发送通知,停止并退出
  85. logging.error(e)
  86. sys.exit()
  87. def check_instance_running(client, instance_ids):
  88. """
  89. 检查服务器运行状态
  90. :param client: 客户端连接
  91. :param instance_ids: 实例id列表, type-list
  92. :return: running_count,Status为Running的实例数
  93. """
  94. try:
  95. request = DescribeInstancesRequest()
  96. request.set_InstanceIds(json.dumps(instance_ids))
  97. response = send_request(client=client, request=request)
  98. if response.get('Code') is None:
  99. instances_list = response.get('Instances').get('Instance')
  100. running_count = 0
  101. running_instances = []
  102. for instance_detail in instances_list:
  103. if instance_detail.get('Status') == "Running":
  104. running_count += 1
  105. running_instances.append(instance_detail.get('InstanceId'))
  106. return running_count, running_instances
  107. else:
  108. # 失败,记录报错信息,发送通知,停止并退出
  109. logging.error(response)
  110. sys.exit()
  111. except Exception as e:
  112. # 失败,记录报错信息,发送通知,停止并退出
  113. logging.error(e)
  114. sys.exit()
  115. def create_multiple_instances(amount,
  116. access_key_id, access_key_secret, region_id,
  117. image_id, vswitch_id, security_group_id, zone_id, instance_type, instance_name,
  118. disk_size, disk_category, key_pair_name, tags):
  119. """
  120. 创建多个ECS实例
  121. :param amount: 创建实例数 type-int 取值范围:[1, 100]
  122. :param access_key_id: access key Id, type-string
  123. :param access_key_secret: access key secret, type-string
  124. :param region_id: region_id, type-string
  125. :param image_id: 使用的镜像信息 type-string
  126. :param vswitch_id: 选择的交换机 type-string
  127. :param security_group_id: 当前vpc类型的安全组 type-string
  128. :param zone_id: 服务器所在区域 type-string
  129. :param instance_type: 实例规格 type-string
  130. :param instance_name: 实例命名 type-string
  131. :param disk_size: 磁盘大小,单位:G,type-string
  132. :param disk_category: 磁盘类型 type-string
  133. :param key_pair_name: 密钥对名称 type-string
  134. :param tags: 标签 type-list, eg: [{"Key": "ecs", "Value": "rov-server.prod"}, ...]
  135. :return:
  136. """
  137. logging.info(f"create instances start, request amount: {amount}.")
  138. # 1. 连接客户端
  139. create_instances_clt = connect_client(
  140. access_key_id=access_key_id, access_key_secret=access_key_secret, region_id=region_id
  141. )
  142. # 2. 请求参数配置
  143. request = build_create_instances_request(
  144. image_id=image_id, vswitch_id=vswitch_id, security_group_id=security_group_id, zone_id=zone_id,
  145. instance_type=instance_type, instance_name=instance_name, disk_size=disk_size, disk_category=disk_category,
  146. key_pair_name=key_pair_name, tags=tags
  147. )
  148. request.set_Amount(amount)
  149. # 3. 发送API请求,购买机器并启动
  150. response = send_request(client=create_instances_clt, request=request)
  151. if response.get('Code') is None:
  152. instance_ids = response.get('InstanceIdSets').get('InstanceIdSet')
  153. logging.info(f"success amount: {len(instance_ids)}, instance ids: {instance_ids}.")
  154. # 获取机器运行状态
  155. running_amount = 0
  156. while running_amount < amount:
  157. time.sleep(10)
  158. running_amount, running_instances = check_instance_running(client=create_instances_clt,
  159. instance_ids=instance_ids)
  160. logging.info(f"running amount: {running_amount}, running instances: {running_instances}.")
  161. return instance_ids
  162. else:
  163. # 失败,记录报错信息,发送通知,停止并退出
  164. logging.error(response)
  165. sys.exit()
  166. def run_command(client, instance_ids, command):
  167. """
  168. 批量执行命令
  169. :param client: 客户端连接
  170. :param instance_ids: 实例id列表, type-list
  171. :param command: 命令 type-string
  172. :return:
  173. """
  174. request = RunCommandRequest()
  175. request.set_accept_format('json')
  176. request.set_Type("RunShellScript")
  177. request.set_CommandContent(command)
  178. request.set_InstanceIds(instance_ids)
  179. response = send_request(client=client, request=request)
  180. logging.info(response)
  181. def get_instances(client, slb_id):
  182. """
  183. 获取slb下所有服务器信息
  184. :param client: 客户端连接
  185. :param slb_id: 负载均衡id type-string
  186. :return:
  187. """
  188. request = DescribeLoadBalancerAttributeRequest()
  189. request.set_accept_format('json')
  190. request.set_LoadBalancerId(slb_id)
  191. response = send_request(client=client, request=request)
  192. return response
  193. def get_ip_address(client, instance_id):
  194. """
  195. 获取实例IP地址
  196. :param client: 客户端连接
  197. :param instance_id: 实例id, type-string
  198. :return: ip_address, type-string
  199. """
  200. request = DescribeNetworkInterfacesRequest()
  201. request.set_accept_format('json')
  202. request.set_InstanceId(instance_id)
  203. response = send_request(client=client, request=request)
  204. ip_address = response['NetworkInterfaceSets']['NetworkInterfaceSet'][0]['PrivateIpAddress']
  205. return ip_address
  206. def set_weight_for_instances(client, slb_id, instance_id_list, weight):
  207. """
  208. 同时设置多台服务器的slb权重,权重一样
  209. :param client: 客户端连接
  210. :param slb_id: slb_id
  211. :param instance_id_list: 服务器id list
  212. :param weight: 权重值
  213. :return: None
  214. """
  215. BackendServers = [{"ServerId": instance_id, "Weight": weight} for instance_id in instance_id_list]
  216. request = CommonRequest()
  217. request.set_accept_format('json')
  218. request.set_domain('slb.aliyuncs.com')
  219. request.set_version('2014-05-15')
  220. request.set_method('POST')
  221. request.set_action_name('SetBackendServers')
  222. request.add_query_param('BackendServers', BackendServers)
  223. request.add_query_param('LoadBalancerId', slb_id)
  224. response = send_request(client=client, request=request)
  225. return response