使用UIImagePickerController时获取文件扩展名的简便方法

Question

func imagePickerController(_ picker: UIImagePickerController, didFinishPickingMediaWithInfo info: [String : Any]) {
    if let pickedImage = info[UIImagePickerControllerOriginalImage] as? UIImage {
        imageView.contentMode = .scaleAspectFit
        imageView.isHidden = false
        imageView.image = pickedImage
        var imageData: Data?
        if let url = info[UIImagePickerControllerImageURL] as? URL {
            if url.isFileURL {
                let pathExtension = url.pathExtension
                switch pathExtension {
                case "jpeg":
                    imageData =  UIImageJPEGRepresentation(pickedImage, 0)
                    break
                case "png":
                    imageData =  UIImagePNGRepresentation(pickedImage)
                    break
                default:
                    break
                }
            }
        }

        if let data =  imageData {
            let bytes = data.count
            print("pickedImage size in bytes:\(bytes)")
            let Mb = (Double(bytes) / 1024.0) / 1024.0
            print("pickedImage size in Mb:\(Mb)")
        }

        dismiss(animated: true, completion: nil)
    }


}