Disk modification changes completed

Author: samperay

diskExtensionEC2.py

@@ -5,13 +5,7 @@
 import paramiko
 import sys
 from pprint import pprint
-
-class Error(Exception):
-  pass
-
-class VolumeModificationRateExceeded(Error):
-  pass
-
+from time import sleep
 
 aws_mgmt_console = boto3.session.Session(profile_name="default",region_name="ap-south-1")
 
@@ -21,8 +15,11 @@ class VolumeModificationRateExceeded(Error):
 # Get Required Info from Instance
 for each_item in ec2_client.describe_instances()['Reservations']:
   for each_instance in each_item['Instances']:
-    public_ip = each_instance['PublicIpAddress']
-    public_dns_name = each_instance['PublicDnsName']
+    if each_instance['State'] == "Running":
+      public_ip = each_instance['PublicIpAddress']
+      public_dns_name = each_instance['PublicDnsName']
+    else:
+      continue
 
 # Get list of Volume ID and its associated Instance ID
 response = ec2_client.describe_volumes()['Volumes']
@@ -34,6 +31,11 @@ class VolumeModificationRateExceeded(Error):
   az = each_item['AvailabilityZone']
   current_state = each_item['State']
   volume_type = each_item['VolumeType']
+
+  print("Current Size: ",current_size)
+  print("az:",az)
+  print("current_state: ",current_state)
+  print("volume_type: ",volume_type)
 
   # Get instance details from volume
   for each_instance in each_item['Attachments']:
@@ -47,37 +49,40 @@ class VolumeModificationRateExceeded(Error):
 print("Device Name:", device)
 print("Current Disk Size:", current_size)
 
-if current_size == 12:
+
+if current_size == 10:
   print("Disk already same size as target size.. exiting")
   sys.exit(0)
 else:
   print("Attempting to increasing volume size from AWS ...")
   volumemodify = ec2_client.modify_volume(
     DryRun = False,
     VolumeId = volume_id,
-    Size = 12,
+    Size = 10,
     VolumeType = volume_type
   )
+  waiter = ec2_client.get_waiter('volume_in_use')
+  waiter.wait(VolumeIds=[volume_id])
   print("Volume has been modified from AWS System..")
 
 # Logging into the instance to get disk partitions using paramiko
 
-key = paramiko.RSAKey.from_private_key_file("/home/samperay/MyLinuxEC2KeyPair.pem")
-client = paramiko.SSHClient()
-client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
-client.connect(hostname=public_ip, username="ec2-user", pkey=key)
+# key = paramiko.RSAKey.from_private_key_file("/home/samperay/MyLinuxEC2KeyPair.pem")
+# client = paramiko.SSHClient()
+# client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
+# client.connect(hostname=public_ip, username="ec2-user", pkey=key)
 
-localfile = "/home/samperay/Documents/gitprojects/aws-python/getDiskInfo.py"
-remotefile = "/tmp/getDiskInfo.py"
+# localfile = "/home/samperay/Documents/gitprojects/aws-python/getDiskInfo.py"
+# remotefile = "/tmp/getDiskInfo.py"
 
-# Copy file locally to remote host
-sftp = client.open_sftp()
-sftp.put(localfile, remotefile)
-sftp.close()
+# # Copy file locally to remote host
+# sftp = client.open_sftp()
+# sftp.put(localfile, remotefile)
+# sftp.close()
 
-#Execute a command(cmd) after connecting/ssh to an instance
-stdin,stdout,stderr = client.exec_command("chmod +x /tmp/getDiskInfo.py; python /tmp/getDiskInfo.py")
-print(stdout.read())
+# #Execute a command(cmd) after connecting/ssh to an instance
+# stdin,stdout,stderr = client.exec_command("chmod +x /tmp/getDiskInfo.py; python /tmp/getDiskInfo.py")
+# print(stdout.read())
 
-# close the client connection once the job is done
-client.close()
+# # close the client connection once the job is done
+# client.close()

getDiskInfo.py

@@ -27,10 +27,9 @@ def partitions(disk):
 device, Type, size, used, available, percent, mountpoint = output.split("\n")[1].split()
 
 if Type == 'xfs':
-  os.system('sudo xfs_growfs -d '+mountpoint)
+  os.system('sudo xfs_growfs -d'+mountpoint)
   print('file system has been extended')
 elif Type == 'ext3' or 'ext4':
   print('Need to work on ext file systems')
 
-sys.exit(0)
-
+sys.exit(0)