def get_disk_info()

in stratozone-aws-export.py [0:0]

• 25 lines of code
• 6 McCabe index (conditional complexity)

def get_disk_info(vm_id, block_device_list, root_device_name):
  """Get attached disk data.

  Args:
    vm_id: Instance ID
    block_device_list: list of attached disks
    root_device_name: name of the primary (OS) disk

  Returns:
      Disk create date.
  """
  disk_count = 0

  try:
    disk_create_date = datetime.datetime.now()

    for block_device in block_device_list:
      disk = stratozonedict.vm_disk.copy()

      volume = client.describe_volumes(
          VolumeIds=[block_device['Ebs']['VolumeId'],]).get('Volumes')

      disk['MachineId'] = vm_id
      disk['DiskLabel'] = block_device['DeviceName']
      disk['SizeInGib'] = volume[0]['Size']
      disk['StorageTypeLabel'] = volume[0]['VolumeType']

      vm_disk_list.append(disk)
      disk_count = disk_count + 1
      if root_device_name == block_device['DeviceName']:
        disk_create_date = block_device['Ebs']['AttachTime']

    if disk_count == 0:
      generate_disk_data(vm_id)

    return disk_create_date

  except Exception as e:
    if disk_count == 0:
      generate_disk_data(vm_id)

    logging.error('error in get_disk_info')
    logging.error(e)
    return disk_create_date