Correct missing Exif data

Format missed information
This commit is contained in:
Michał Gdula 2023-01-31 22:08:37 +00:00
parent da1579555b
commit c02d618844
6 changed files with 266 additions and 179 deletions

View file

@ -2,38 +2,65 @@ import PIL
from PIL import Image
from PIL.ExifTags import TAGS, GPSTAGS
from datetime import datetime
import os
class metadata:
def yoink(filename):
exif = metadata.getFile(filename)
file_size = os.path.getsize(filename)
file_name = os.path.basename(filename)
file_resolution = Image.open(filename).size
if exif:
formatted = metadata.format(exif)
unformatted_exif = metadata.format(exif, file_size, file_name, file_resolution)
else:
return None
# No EXIF data, get some basic informaton from the file
unformatted_exif = {
'File': {
'Name': {
'type': 'text',
'raw': file_name
},
'Size': {
'type': 'number',
'raw': file_size,
'formatted': metadata.human_size(file_size)
},
'Format': {
'type': 'text',
'raw': file_name.split('.')[-1]
},
'Width': {
'type': 'number',
'raw': file_resolution[0]
},
'Height': {
'type': 'number',
'raw': file_resolution[1]
},
}
}
return metadata.deleteEmpty(formatted)
def deleteEmpty(dict):
new_dict = {}
formatted_exif = {}
for section in dict:
for section in unformatted_exif:
tmp = {}
for value in dict[section]:
if dict[section][value]['raw'] != None:
if isinstance(dict[section][value]['raw'], PIL.TiffImagePlugin.IFDRational):
dict[section][value]['raw'] = dict[section][value]['raw'].__float__()
elif isinstance(dict[section][value]['raw'], bytes):
dict[section][value]['raw'] = dict[section][value]['raw'].decode('utf-8')
for value in unformatted_exif[section]:
if unformatted_exif[section][value]['raw'] != None:
raw_type = unformatted_exif[section][value]['raw']
if isinstance(raw_type, PIL.TiffImagePlugin.IFDRational):
raw_type = raw_type.__float__()
elif isinstance(raw_type, bytes):
raw_type = raw_type.decode('utf-8')
tmp[value] = dict[section][value]
tmp[value] = unformatted_exif[section][value]
if len(tmp) > 0:
new_dict[section] = tmp
formatted_exif[section] = tmp
return new_dict
return formatted_exif
def getFile(filename):
try:
@ -54,9 +81,9 @@ class metadata:
return exif
except Exception as e:
return None
return False
def format(raw):
def format(raw, file_size, file_name, file_resolution):
exif = {}
exif['Photographer'] = {
@ -72,11 +99,6 @@ class metadata:
'type': 'text',
'raw': raw["ImageDescription"]["raw"]
},
'Date Digitized': {
'type': 'date',
'raw': raw["DateTimeDigitized"]["raw"],
'formatted': metadata.date(raw["DateTimeDigitized"]["raw"])
},
'Copyright': {
'type': 'text',
'raw': raw["Copyright"]["raw"]
@ -91,6 +113,14 @@ class metadata:
'type': 'text',
'raw': raw['Make']['raw']
},
'Camera Type': {
'type': 'text',
'raw': raw['BodySerialNumber']['raw']
},
'Lens Make': {
'type': 'text',
'raw': raw['LensMake']['raw'],
},
'Lense Model': {
'type': 'text',
'raw': raw['LensModel']['raw'],
@ -110,24 +140,28 @@ class metadata:
'raw': raw['DateTime']['raw'],
'formatted': metadata.date(raw['DateTime']['raw'])
},
}
exif['Software'] = {
'Software': {
'Date Digitized': {
'type': 'date',
'raw': raw["DateTimeDigitized"]["raw"],
'formatted': metadata.date(raw["DateTimeDigitized"]["raw"])
},
'Time Offset': {
'type': 'text',
'raw': raw['Software']['raw']
'raw': raw["OffsetTime"]["raw"]
},
'Colour Space': {
'type': 'number',
'raw': raw['ColorSpace']['raw'],
'formatted': metadata.colorSpace(raw['ColorSpace']['raw'])
'Time Offset - Original': {
'type': 'text',
'raw': raw["OffsetTimeOriginal"]["raw"]
},
'Compression': {
'type': 'number',
'raw': raw['Compression']['raw'],
'formatted': metadata.compression(raw['Compression']['raw'])
'Time Offset - Digitized': {
'type': 'text',
'raw': raw["OffsetTimeDigitized"]["raw"]
},
'Date Original': {
'type': 'date',
'raw': raw["DateTimeOriginal"]["raw"],
'formatted': metadata.date(raw["DateTimeOriginal"]["raw"])
},
}
exif['Photo'] = {
'FNumber': {
'type': 'fnumber',
'raw': raw["FNumber"]["raw"],
@ -135,11 +169,13 @@ class metadata:
},
'Focal Length': {
'type': 'focal',
'raw': raw["FocalLength"]["raw"]
'raw': raw["FocalLength"]["raw"],
'formatted': metadata.focal(raw["FocalLength"]["raw"])
},
'Focal Length - Film': {
'Focal Length (35mm format)': {
'type': 'focal',
'raw': raw["FocalLengthIn35mmFilm"]["raw"]
'raw': raw["FocalLengthIn35mmFilm"]["raw"],
'formatted': metadata.focal(raw["FocalLengthIn35mmFilm"]["raw"])
},
'Max Aperture': {
'type': 'fnumber',
@ -220,16 +256,55 @@ class metadata:
'type': 'number',
'raw': raw["SceneType"]["raw"],
'formatted': metadata.sceneType(raw["SceneType"]["raw"])
},
},
'Rating': {
'type': 'number',
'raw': raw["Rating"]["raw"],
'formatted': metadata.rating(raw["Rating"]["raw"])
},
'Rating Percent': {
'type': 'number',
'raw': raw["RatingPercent"]["raw"],
'formatted': metadata.ratingPercent(raw["RatingPercent"]["raw"])
},
}
exif['Software'] = {
'Software': {
'type': 'text',
'raw': raw['Software']['raw']
},
'Colour Space': {
'type': 'number',
'raw': raw['ColorSpace']['raw'],
'formatted': metadata.colorSpace(raw['ColorSpace']['raw'])
},
'Compression': {
'type': 'number',
'raw': raw['Compression']['raw'],
'formatted': metadata.compression(raw['Compression']['raw'])
},
}
exif['File'] = {
'Name': {
'type': 'text',
'raw': file_name
},
'Size': {
'type': 'number',
'raw': file_size,
'formatted': metadata.human_size(file_size)
},
'Format': {
'type': 'text',
'raw': file_name.split('.')[-1]
},
'Width': {
'type': 'number',
'raw': raw["ImageWidth"]["raw"]
'raw': file_resolution[0]
},
'Height': {
'type': 'number',
'raw': raw["ImageLength"]["raw"]
'raw': file_resolution[1]
},
'Orientation': {
'type': 'number',
@ -250,8 +325,27 @@ class metadata:
'formatted': metadata.resolutionUnit(raw["ResolutionUnit"]["raw"])
},
}
#exif['Raw'] = {}
#for key in raw:
# try:
# exif['Raw'][key] = {
# 'type': 'text',
# 'raw': raw[key]['raw'].decode('utf-8')
# }
# except:
# exif['Raw'][key] = {
# 'type': 'text',
# 'raw': str(raw[key]['raw'])
# }
return exif
def human_size(num, suffix="B"):
for unit in ["", "Ki", "Mi", "Gi", "Ti", "Pi", "Ei", "Zi"]:
if abs(num) < 1024.0:
return f"{num:3.1f}{unit}{suffix}"
num /= 1024.0
return f"{num:.1f}Yi{suffix}"
def date(date):
date_format = '%Y:%m:%d %H:%M:%S'
@ -281,7 +375,10 @@ class metadata:
def focal(value):
if value != None:
return str(value[0] / value[1]) + 'mm'
try:
return str(value[0] / value[1]) + 'mm'
except:
return str(value) + 'mm'
else:
return None
@ -297,7 +394,6 @@ class metadata:
65535: 'Uncalibrated',
0: 'Reserved'
}
try:
return types[int(value)]
except:
@ -328,7 +424,6 @@ class metadata:
93: 'Flash fired, auto mode, return light not detected, red-eye reduction mode',
95: 'Flash fired, auto mode, return light detected, red-eye reduction mode'
}
try:
return types[int(value)]
except:
@ -346,7 +441,6 @@ class metadata:
7: 'Portrait mode',
8: 'Landscape mode'
}
try:
return types[int(value)]
except:
@ -363,7 +457,6 @@ class metadata:
6: 'Partial',
255: 'Other'
}
try:
return types[int(value)]
except:
@ -375,7 +468,6 @@ class metadata:
2: 'Inch',
3: 'Centimeter'
}
try:
return types[int(value)]
except:
@ -405,7 +497,6 @@ class metadata:
24: 'ISO studio tungsten',
255: 'Other light source',
}
try:
return types[int(value)]
except:
@ -418,7 +509,6 @@ class metadata:
2: 'Portrait',
3: 'Night scene',
}
try:
return types[int(value)]
except:
@ -435,7 +525,6 @@ class metadata:
0: 'Auto white balance',
1: 'Manual white balance',
}
try:
return types[int(value)]
except:
@ -447,7 +536,6 @@ class metadata:
1: 'Manual exposure',
2: 'Auto bracket',
}
try:
return types[int(value)]
except:
@ -464,7 +552,6 @@ class metadata:
6: 'Recommended Exposure Index and ISO Speed',
7: 'Standard Output Sensitivity, Recommended Exposure Index and ISO Speed',
}
try:
return types[int(value)]
except:
@ -479,9 +566,55 @@ class metadata:
def compression(value):
types = {
1: 'Uncompressed',
6: 'JPEG compression',
2: 'CCITT 1D',
3: 'T4/Group 3 Fax',
4: 'T6/Group 4 Fax',
5: 'LZW',
6: 'JPEG (old-style)',
7: 'JPEG',
8: 'Adobe Deflate',
9: 'JBIG B&W',
10: 'JBIG Color',
99: 'JPEG',
262: 'Kodak 262',
32766: 'Next',
32767: 'Sony ARW Compressed',
32769: 'Packed RAW',
32770: 'Samsung SRW Compressed',
32771: 'CCIRLEW',
32772: 'Samsung SRW Compressed 2',
32773: 'PackBits',
32809: 'Thunderscan',
32867: 'Kodak KDC Compressed',
32895: 'IT8CTPAD',
32896: 'IT8LW',
32897: 'IT8MP',
32898: 'IT8BL',
32908: 'PixarFilm',
32909: 'PixarLog',
32946: 'Deflate',
32947: 'DCS',
33003: 'Aperio JPEG 2000 YCbCr',
33005: 'Aperio JPEG 2000 RGB',
34661: 'JBIG',
34676: 'SGILog',
34677: 'SGILog24',
34712: 'JPEG 2000',
34713: 'Nikon NEF Compressed',
34715: 'JBIG2 TIFF FX',
34718: '(MDI) Binary Level Codec',
34719: '(MDI) Progressive Transform Codec',
34720: '(MDI) Vector',
34887: 'ESRI Lerc',
34892: 'Lossy JPEG',
34925: 'LZMA2',
34926: 'Zstd',
34927: 'WebP',
34933: 'PNG',
34934: 'JPEG XR',
65000: 'Kodak DCR Compressed',
65535: 'Pentax PEF Compressed',
}
try:
return types[int(value)]
except:
@ -498,7 +631,6 @@ class metadata:
7: 'Mirror horizontal and rotate 90 CW',
8: 'Rotate 270 CW',
}
try:
return types[int(value)]
except:
@ -514,8 +646,13 @@ class metadata:
5: 'G',
6: 'B',
}
try:
return ''.join([types[int(x)] for x in value])
except:
return None
return None
def rating(value):
return str(value) + ' stars'
def ratingPercent(value):
return str(value) + '%'