55 lines
1.2 KiB
Python
Executable file
55 lines
1.2 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
import sys
|
|
import io
|
|
import pyheif
|
|
import piexif
|
|
from PIL import Image
|
|
from pathlib import Path
|
|
|
|
class Heic2Jpeg:
|
|
def __init__(self, path):
|
|
self.path = Path(path)
|
|
self.img = None
|
|
|
|
def decodeImage(self):
|
|
self.himage = pyheif.read_heif(self.path)
|
|
self.img = Image.frombytes(
|
|
self.himage.mode,
|
|
self.himage.size,
|
|
self.himage.data,
|
|
"raw",
|
|
self.himage.mode,
|
|
self.himage.stride)
|
|
|
|
def getExif(self):
|
|
for metadata in self.himage.metadata or []:
|
|
if metadata['type'] == 'Exif':
|
|
# pyheif.read_heif() rotates the picture
|
|
# we need to remove the Orientation from EXIF
|
|
exif_dict = piexif.load(metadata['data'])
|
|
if exif_dict["0th"] and exif_dict["0th"].get(piexif.ImageIFD.Orientation):
|
|
exif_dict["0th"][piexif.ImageIFD.Orientation] = 1
|
|
|
|
return piexif.dump(exif_dict)
|
|
|
|
def safe(self, quality=85):
|
|
if self.img == None:
|
|
self.decodeImage()
|
|
|
|
path = self.path.with_suffix('.jpg')
|
|
|
|
self.img.save(
|
|
path,
|
|
format="JPEG",
|
|
quality=quality,
|
|
exif=self.getExif())
|
|
return path.__str__()
|
|
|
|
if __name__ == "__main__":
|
|
if len(sys.argv) != 2:
|
|
print("Usage: heic2jpeg.py path/to/picture.heic")
|
|
else:
|
|
img_path = sys.argv[1]
|
|
h2j = Heic2Jpeg(img_path)
|
|
print(h2j.safe())
|