xfs file system disk extension completed

Author: samperay

diskExtensionEC2.py

@@ -15,11 +15,10 @@
 # Get Required Info from Instance
 for each_item in ec2_client.describe_instances()['Reservations']:
   for each_instance in each_item['Instances']:
-    if each_instance['State'] == "Running":
-      public_ip = each_instance['PublicIpAddress']
-      public_dns_name = each_instance['PublicDnsName']
-    else:
-      continue
+    instance_state = each_instance['State']['Name']
+    if instance_state == 'running' and each_instance.get('PublicIpAddress') is not None:
+      public_ip = each_instance.get('PublicIpAddress')
+      print("Public_Ip:",public_ip)
 
 # Get list of Volume ID and its associated Instance ID
 response = ec2_client.describe_volumes()['Volumes']
@@ -50,19 +49,19 @@
 print("Current Disk Size:", current_size)
 
 
-if current_size == 10:
+if current_size == 12:
   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 = 10,
+    Size = 12,
     VolumeType = volume_type
   )
-  waiter = ec2_client.get_waiter('volume_in_use')
-  waiter.wait(VolumeIds=[volume_id])
+  print('Extending volume from AWS ......')
+  sleep(300)
   print("Volume has been modified from AWS System..")
 
 # Logging into the instance to get disk partitions using paramiko
@@ -85,4 +84,26 @@
 # print(stdout.read())
 
 # # close the client connection once the job is done
-# client.close()
+# client.close()
+
+# ====== Windows testing =======
+
+key = paramiko.RSAKey.from_private_key_file("E:\\aws-keys\ebsdemo.pem")
+client = paramiko.SSHClient()
+client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
+client.connect(hostname=public_ip, username="ec2-user", pkey=key)
+
+localfile = "E:\\projects\\aws-python\getDiskInfo.py"
+remotefile = "/tmp/getDiskInfo.py"
+
+# 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())
+
+# close the client connection once the job is done
+client.close()

getDiskInfo.py

@@ -17,17 +17,21 @@ def partitions(disk):
 disk = physical_drives()
 disk_part = partitions(disk[0])
 
-conv_str = ' '.join(map(str, disk_part))
-main_device = "/dev/" + conv_str
+conv_str = ' '.join(map(str, disk))
+root_device = "/dev/"+conv_str
+
+conv_str1 = ' '.join(map(str, disk_part))
+root_partition = "/dev/"+conv_str1
 
 # Fetching mount point from main_device
 
-df = subprocess.Popen(["df", "-hT", main_device], stdout=subprocess.PIPE)
+df = subprocess.Popen(["df", "-hT", root_partition], stdout=subprocess.PIPE)
 output = df.communicate()[0]
 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 growpart '+root_device+' 1')
+  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')