add type hints
This commit is contained in:
parent
45ea0df8b6
commit
9b3366e2a1
24
writer.py
24
writer.py
|
@ -29,7 +29,7 @@ class PictureMetadata:
|
||||||
orientation: Optional[int] = 1
|
orientation: Optional[int] = 1
|
||||||
|
|
||||||
class Writer():
|
class Writer():
|
||||||
def __init__(self, picture: bytes):
|
def __init__(self, picture: bytes) -> None:
|
||||||
self.content = picture
|
self.content = picture
|
||||||
self.image = pyexiv2.ImageData(picture)
|
self.image = pyexiv2.ImageData(picture)
|
||||||
self.exif = self.image.read_exif()
|
self.exif = self.image.read_exif()
|
||||||
|
@ -40,20 +40,20 @@ class Writer():
|
||||||
def __enter__(self):
|
def __enter__(self):
|
||||||
return self
|
return self
|
||||||
|
|
||||||
def __exit__(self, *args):
|
def __exit__(self, *args) -> None:
|
||||||
self.image.close()
|
self.image.close()
|
||||||
|
|
||||||
def apply(self):
|
def apply(self) -> None:
|
||||||
self.image.modify_exif(self.updated_exif)
|
self.image.modify_exif(self.updated_exif)
|
||||||
self.image.modify_xmp(self.updated_xmp)
|
self.image.modify_xmp(self.updated_xmp)
|
||||||
|
|
||||||
def close(self):
|
def close(self) -> None:
|
||||||
self.image.close()
|
self.image.close()
|
||||||
|
|
||||||
def get_Bytes(self):
|
def get_Bytes(self) -> bytes:
|
||||||
return self.image.get_bytes()
|
return self.image.get_bytes()
|
||||||
|
|
||||||
def writePictureMetadata(self, metadata: PictureMetadata):
|
def writePictureMetadata(self, metadata: PictureMetadata) -> None:
|
||||||
"""
|
"""
|
||||||
Override exif metadata on raw picture and return updated bytes
|
Override exif metadata on raw picture and return updated bytes
|
||||||
"""
|
"""
|
||||||
|
@ -70,7 +70,7 @@ class Writer():
|
||||||
if metadata.picture_type is not None:
|
if metadata.picture_type is not None:
|
||||||
self.add_img_projection(metadata)
|
self.add_img_projection(metadata)
|
||||||
|
|
||||||
def add_lat_lon(self, metadata: PictureMetadata):
|
def add_lat_lon(self, metadata: PictureMetadata) -> None:
|
||||||
"""
|
"""
|
||||||
Add latitude and longitude values in GPSLatitude + GPSLAtitudeRef and GPSLongitude + GPSLongitudeRef
|
Add latitude and longitude values in GPSLatitude + GPSLAtitudeRef and GPSLongitude + GPSLongitudeRef
|
||||||
"""
|
"""
|
||||||
|
@ -82,7 +82,7 @@ class Writer():
|
||||||
self.updated_exif["Exif.GPSInfo.GPSLongitudeRef"] = "E" if metadata.longitude > 0 else "W"
|
self.updated_exif["Exif.GPSInfo.GPSLongitudeRef"] = "E" if metadata.longitude > 0 else "W"
|
||||||
self.updated_exif["Exif.GPSInfo.GPSLongitude"] = self._to_exif_dms(metadata.longitude)
|
self.updated_exif["Exif.GPSInfo.GPSLongitude"] = self._to_exif_dms(metadata.longitude)
|
||||||
|
|
||||||
def add_altitude(self, metadata: PictureMetadata, precision: int = 1000):
|
def add_altitude(self, metadata: PictureMetadata, precision: int = 1000) -> None:
|
||||||
"""
|
"""
|
||||||
Add altitude value in GPSAltitude and GPSAltitudeRef
|
Add altitude value in GPSAltitude and GPSAltitudeRef
|
||||||
"""
|
"""
|
||||||
|
@ -93,7 +93,7 @@ class Writer():
|
||||||
self.updated_exif['Exif.GPSInfo.GPSAltitude'] = f"{int(abs(altitude * precision))} / {precision}"
|
self.updated_exif['Exif.GPSInfo.GPSAltitude'] = f"{int(abs(altitude * precision))} / {precision}"
|
||||||
self.updated_exif['Exif.GPSInfo.GPSAltitudeRef'] = negative_altitude
|
self.updated_exif['Exif.GPSInfo.GPSAltitudeRef'] = negative_altitude
|
||||||
|
|
||||||
def add_direction(self, metadata: PictureMetadata, ref: str = 'T', precision: int = 1000):
|
def add_direction(self, metadata: PictureMetadata, ref: str = 'T', precision: int = 1000) -> None:
|
||||||
"""
|
"""
|
||||||
Add direction value in GPSImgDirection and GPSImgDirectionRef
|
Add direction value in GPSImgDirection and GPSImgDirectionRef
|
||||||
"""
|
"""
|
||||||
|
@ -103,7 +103,7 @@ class Writer():
|
||||||
self.updated_exif['Exif.GPSInfo.GPSImgDirection'] = f"{int(abs(direction % 360.0 * precision))} / {precision}"
|
self.updated_exif['Exif.GPSInfo.GPSImgDirection'] = f"{int(abs(direction % 360.0 * precision))} / {precision}"
|
||||||
self.updated_exif['Exif.GPSInfo.GPSImgDirectionRef'] = ref
|
self.updated_exif['Exif.GPSInfo.GPSImgDirectionRef'] = ref
|
||||||
|
|
||||||
def add_gps_datetime(self, metadata: PictureMetadata):
|
def add_gps_datetime(self, metadata: PictureMetadata) -> None:
|
||||||
"""
|
"""
|
||||||
Add GPSDateStamp and GPSTimeStamp
|
Add GPSDateStamp and GPSTimeStamp
|
||||||
"""
|
"""
|
||||||
|
@ -121,7 +121,7 @@ class Writer():
|
||||||
self.updated_exif["Exif.GPSInfo.GPSDateStamp"] = utc_dt.strftime("%Y:%m:%d")
|
self.updated_exif["Exif.GPSInfo.GPSDateStamp"] = utc_dt.strftime("%Y:%m:%d")
|
||||||
self.updated_exif["Exif.GPSInfo.GPSTimeStamp"] = utc_dt.strftime("%H/1 %M/1 %S/1")
|
self.updated_exif["Exif.GPSInfo.GPSTimeStamp"] = utc_dt.strftime("%H/1 %M/1 %S/1")
|
||||||
|
|
||||||
def add_datetimeoriginal(self, metadata: PictureMetadata):
|
def add_datetimeoriginal(self, metadata: PictureMetadata) -> None:
|
||||||
"""
|
"""
|
||||||
Add date time in Exif DateTimeOriginal and SubSecTimeOriginal tags
|
Add date time in Exif DateTimeOriginal and SubSecTimeOriginal tags
|
||||||
"""
|
"""
|
||||||
|
@ -137,7 +137,7 @@ class Writer():
|
||||||
if metadata.capture_time.microsecond != 0:
|
if metadata.capture_time.microsecond != 0:
|
||||||
self.updated_exif["Exif.Photo.SubSecTimeOriginal"] = metadata.capture_time.strftime("%f")
|
self.updated_exif["Exif.Photo.SubSecTimeOriginal"] = metadata.capture_time.strftime("%f")
|
||||||
|
|
||||||
def add_img_projection(self, metadata: PictureMetadata):
|
def add_img_projection(self, metadata: PictureMetadata) -> None:
|
||||||
"""
|
"""
|
||||||
Add image projection type (equirectangular for spherical image, ...) in xmp GPano.ProjectionType
|
Add image projection type (equirectangular for spherical image, ...) in xmp GPano.ProjectionType
|
||||||
"""
|
"""
|
||||||
|
|
Loading…
Reference in New Issue