def __init__(self, disk_spec, vm_name, machine_type):
    super(AzureDisk, self).__init__(disk_spec)
    self.host_caching = FLAGS.azure_host_caching
    self.name = None
    self.vm_name = vm_name
    self.lun = None

    if self.disk_type == PREMIUM_STORAGE:
      self.metadata = PREMIUM_STORAGE_METADATA
    elif self.disk_type == STANDARD_DISK:
      self.metadata = {
          disk.MEDIA: disk.HDD,
          disk.REPLICATION: AZURE_REPLICATION_MAP[FLAGS.azure_storage_type],
          disk.LEGACY_DISK_TYPE: disk.STANDARD
      }
    elif self.disk_type == disk.LOCAL:
      media = disk.SSD if LocalDiskIsSSD(machine_type) else disk.HDD

      self.metadata = {
          disk.MEDIA: media,
          disk.REPLICATION: disk.NONE,
          disk.LEGACY_DISK_TYPE: disk.LOCAL
      }
dataset/ETHPy150Open GoogleCloudPlatform/PerfKitBenchmarker/perfkitbenchmarker/providers/azure/azure_disk.py/AzureDisk.__init__/original