113 lines
3.2 KiB
Python
113 lines
3.2 KiB
Python
import subprocess
|
|
from datetime import datetime
|
|
from typing import List
|
|
|
|
_file_date_format = "%Y:%m:%d %H:%M:%S%z"
|
|
_original_date_format = "%Y:%m:%d %H:%M:%S.%f%z"
|
|
_original_date_format_fallback = "%Y:%m:%d %H:%M:%S.%f"
|
|
|
|
_date_fields = {
|
|
'file_modification_date/time': [_file_date_format],
|
|
'file_access_date/time': [_file_date_format],
|
|
'file_inode_change_date/time': [_file_date_format],
|
|
'date/time_original': [_original_date_format, _original_date_format_fallback],
|
|
'create_date': [_original_date_format, _original_date_format_fallback],
|
|
'modify_date': [_original_date_format, _original_date_format_fallback],
|
|
}
|
|
|
|
_integer_fields = [
|
|
'image_width',
|
|
'image_height',
|
|
'iso',
|
|
'shutter_count',
|
|
'jpg_from_raw_start',
|
|
'jpg_from_raw_length',
|
|
'thumbnail_offset',
|
|
'thumbnail_length',
|
|
'sr2_sub_ifd_offset',
|
|
'sr2_sub_ifd_length',
|
|
'exif_image_width',
|
|
'exif_image_height',
|
|
'shutter_count_2',
|
|
'sony_iso',
|
|
'iso_auto_min',
|
|
'iso_auto_max',
|
|
'bits_per_sample',
|
|
'strip_byte_counts',
|
|
'rows_per_strip',
|
|
'strip_offsets',
|
|
'x_resolution',
|
|
'y_resolution',
|
|
'samples_per_pixel',
|
|
'sequence_file_number',
|
|
'digital_zoom_ratio',
|
|
'sequence_image_number',
|
|
'focus_position_2',
|
|
]
|
|
|
|
_decimal_fields = [
|
|
'aperture',
|
|
'megapixels',
|
|
'light_value',
|
|
'blue_balance',
|
|
'sony_f_number',
|
|
'sony_max_aperture_value',
|
|
'sony_f_number_2',
|
|
'f_number',
|
|
'max_aperture_value',
|
|
'brightness_value',
|
|
'stops_above_base_iso',
|
|
]
|
|
|
|
|
|
def _parse_datetime(raw_value: str, available_formats: List[str]) -> datetime:
|
|
for available_format in available_formats:
|
|
try:
|
|
return datetime.strptime(raw_value, available_format)
|
|
except ValueError:
|
|
continue
|
|
|
|
raise ValueError(f"Could not parse datetime '{raw_value}' ({available_formats})")
|
|
|
|
|
|
def execute_exiftool(img_file: str) -> dict:
|
|
res = subprocess.run(
|
|
['exiftool', img_file],
|
|
capture_output=True,
|
|
text=True,
|
|
)
|
|
|
|
if res.returncode != 0:
|
|
raise Exception(res.stderr)
|
|
|
|
exif_metadata = {}
|
|
|
|
for line in res.stdout.splitlines():
|
|
parts = line.split(':', 1)
|
|
exif_metadata[parts[0].strip().lower().replace(' ', '_')] = parts[1].strip()
|
|
|
|
for (field, date_formats) in _date_fields.items():
|
|
if field in exif_metadata:
|
|
try:
|
|
exif_metadata[field] = _parse_datetime(exif_metadata[field], date_formats)
|
|
except ValueError as e:
|
|
print(f'Failed to convert {field} ({exif_metadata[field]}) to a datetime.')
|
|
raise e
|
|
|
|
for field in _integer_fields:
|
|
if field in exif_metadata:
|
|
try:
|
|
exif_metadata[field] = int(exif_metadata[field])
|
|
except ValueError as e:
|
|
print(f'Failed to convert {field} ({exif_metadata[field]}) to an integer.')
|
|
raise e
|
|
|
|
for field in _decimal_fields:
|
|
if field in exif_metadata:
|
|
try:
|
|
exif_metadata[field] = float(exif_metadata[field])
|
|
except ValueError as e:
|
|
print(f'Failed to convert {field} ({exif_metadata[field]}) to a float.')
|
|
raise e
|
|
|
|
return exif_metadata
|