def add_disk(self, source_file, device): if device == "cdrom": self.disks.append(Disk.cdrom_init(source_file)) elif device == "disk": self.disks.append(Disk.disk_init(source_file))