脚本要实现的功能：输入instance id
1：将所有的volume take snapshot
2: 获取public ip 并登陆机器执行 ps 命令记录patch前进程状态已经端口状态
3：获取机器所在的elb
4: 从elb中移除当前机器
5：检查snapshots是否完成
6：snapshots完成后patching
7: patching完成后将instance加回到elb
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 |
#!/usr/bin/python # vim: expandtab:tabstop=4:shiftwidth=4 ''' script to get ecr info ''' # Reason: disable invalid-name because pylint does not like our naming convention # pylint: disable=invalid-name import time import boto3 import sys import argparse def get_volume(ec2, instanceId): result = [] instance = ec2.Instance(instanceId) volumes = instance.volumes.all() for volume in volumes: print("Volume attached to this instance is :" + volume.id) result.append(volume.id) return result def take_snapByInstance(client, instanceId): response = client.create_snapshots( Description='string', InstanceSpecification={ 'InstanceId': instanceId, 'ExcludeBootVolume': False }, TagSpecifications=[ { 'ResourceType': 'snapshot', 'Tags': [ { 'Key': 'orginName', 'Value': 'patch backup'+ instanceId }, ] }, ], DryRun=False, CopyTagsFromSource='volume' ) print("Creating new snapshots for instances:" + response['Snapshots'][0]['SnapshotId']) return response['Snapshots'][0]['SnapshotId'] def get_publicIp(ec2, instanceId): instance = ec2.Instance(instanceId) publicIp = instance.public_ip_address return publicIp def take_screenshotOfProcess(public_ip): print("Please run this command on your local machine") print('ssh -t ' + public_ip + ' "sudo netstat -tnpl > disk.listen"') print('ssh -t ' + public_ip + ' "sudo ps auxf > disk.ps"') def get_elbInfo(client_elb, ec2, instanceId): bals = client_elb.describe_load_balancers() for elb in bals['LoadBalancerDescriptions']: #print('ELB DNS Name : ' + elb['DNSName']) #check if the elb is the elb of instance if instanceId in elb['Instances']: print("found elb " + elb['DNSName']) else: pass def remove_fromElb(client_elb, elb, instanceId): response = client_elb.deregister_instances_from_load_balancer( LoadBalancerName='elb', Instances=[ { 'InstanceId': instanceId }, ] ) def add_backElb(client_elb, elb, instanceId): response = client.register_instances_with_load_balancer( LoadBalancerName= elb, Instances=[ { 'InstanceId': instanceId }, ] ) def check_snapStatus(ec2, snaps): snapshot = ec2.Snapshot(snaps) snapshot.load() print(snapshot.state) return snapshot.state def main(ec2, client, instanceId, client_elb): print("going to paching instanceid: " + instanceId) #get volumes volumes = get_volume(ec2, instanceId) #get public ip public_ip = get_publicIp(ec2, instanceId) #take snapshot snaps = take_snapByInstance(client, instanceId) #take screenshot of procss and port take_screenshotOfProcess(public_ip) #get elb info elb = False #elb = get_elbInfo(client_elb, ec2, instanceId) #remove from elb if elb: ans_remove = input("Are you sure to remove the instance from the elb now? Yes/No") if ans_remove == 'Yes': #remove from instance remove_fromElb(client_elb, elb, instanceId) #check snapshot status snapshotStatus = '' check_snapStatus(ec2, snaps) print("checking staus of snapshots") while True: snapshotStatus = check_snapStatus(ec2, snaps) print(snapshotStatus) if snapshotStatus == 'completed': break else: time.sleep(10) #paching paching_cmd = 'Your paching command' print(paching_cmd) #add to elb if elb: ans_add = input("please confirm the patching is over , input yes to continue") if ans_add == 'Yes': add_backElb(client_elb, elb, instanceId) if __name__ == "__main__": ec2 = boto3.resource('ec2', region_name='us-east-1') client = boto3.client('ec2', region_name='us-east-1') client_elb = boto3.client('elb', region_name='us-east-1') main(ec2, client, 'i-abcasdfa111122', client_elb) |
注意，本脚本并未包含链接机器并执行命令的部分，仅仅是打印出命令，需要手动执行 take_screenshotOfProcess 已经patch的命令，此部分也参考之前的文章 ，完全自动化，不需要手动执行
另外Patch命令脚本中并未给出
Latest posts by Zhiming Zhang (see all)
- aws eks node 自动化扩展工具 Karpenter - 8月 10, 2022
- ReplicationController and ReplicaSet in Kubernetes - 12月 20, 2021
- public key fingerprint - 5月 27, 2021