mirror of
https://github.com/Derpy-Leggies/OnlyLegs.git
synced 2025-01-09 08:05:19 +00:00
99 lines
2.9 KiB
Python
99 lines
2.9 KiB
Python
"""
|
|
OnlyLegs - Metadata Parser
|
|
Parse metadata from images if available
|
|
otherwise get some basic information from the file
|
|
"""
|
|
import os
|
|
|
|
from PIL import Image
|
|
from PIL.ExifTags import TAGS
|
|
|
|
from .helpers import *
|
|
from .mapping import *
|
|
|
|
|
|
class Metadata:
|
|
"""
|
|
Metadata parser
|
|
"""
|
|
def __init__(self, file_path):
|
|
"""
|
|
Initialize the metadata parser
|
|
"""
|
|
self.file_path = file_path
|
|
img_exif = {}
|
|
|
|
try:
|
|
file = Image.open(file_path)
|
|
tags = file._getexif()
|
|
img_exif = {}
|
|
|
|
for tag, value in TAGS.items():
|
|
if tag in tags:
|
|
img_exif[value] = tags[tag]
|
|
|
|
img_exif['FileName'] = os.path.basename(file_path)
|
|
img_exif['FileSize'] = os.path.getsize(file_path)
|
|
img_exif['FileFormat'] = img_exif['FileName'].split('.')[-1]
|
|
img_exif['FileWidth'], img_exif['FileHeight'] = file.size
|
|
|
|
file.close()
|
|
except TypeError:
|
|
img_exif['FileName'] = os.path.basename(file_path)
|
|
img_exif['FileSize'] = os.path.getsize(file_path)
|
|
img_exif['FileFormat'] = img_exif['FileName'].split('.')[-1]
|
|
img_exif['FileWidth'], img_exif['FileHeight'] = file.size
|
|
|
|
self.encoded = img_exif
|
|
|
|
def yoink(self):
|
|
"""
|
|
Yoinks the metadata from the image
|
|
"""
|
|
if not os.path.isfile(self.file_path):
|
|
return None
|
|
return self.format_data(self.encoded)
|
|
|
|
@staticmethod
|
|
def format_data(encoded_exif):
|
|
"""
|
|
Formats the data into a dictionary
|
|
"""
|
|
exif = {
|
|
'Photographer': {},
|
|
'Camera': {},
|
|
'Software': {},
|
|
'File': {},
|
|
}
|
|
|
|
# Thanks chatGPT xP
|
|
for key, value in encoded_exif.items():
|
|
for mapping_name, mapping_val in EXIF_MAPPING:
|
|
if key in mapping_val:
|
|
if len(mapping_val[key]) == 2:
|
|
# the helper function works, so not sure why it triggers pylint
|
|
exif[mapping_name][mapping_val[key][0]] = {
|
|
'raw': value,
|
|
'formatted': (
|
|
getattr(helpers, mapping_val[key][1]) # pylint: disable=E0602
|
|
(value)
|
|
),
|
|
}
|
|
else:
|
|
exif[mapping_name][mapping_val[key][0]] = {
|
|
'raw': value,
|
|
}
|
|
continue
|
|
|
|
# Remove empty keys
|
|
if not exif['Photographer']:
|
|
del exif['Photographer']
|
|
if not exif['Camera']:
|
|
del exif['Camera']
|
|
if not exif['Software']:
|
|
del exif['Software']
|
|
if not exif['File']:
|
|
del exif['File']
|
|
|
|
return exif
|