exifdata.py

#!/usr/bin/python3 -ttu
#
# Use exiftool to read data from image and translate to Python dictionary
#
# Copyright 2021 Michael T. Babcock <mtb-code@mikebabcock.ca>
# Released under MIT License, enjoy

import sys, os
from subprocess import run
import json
import pprint

class EXIFException(Exception):
    pass

class EXIFData:
    def __init__(self, filename):
        if not filename == None:
            self.readexifdata(filename)

    def readexifdata(self, filename):
        exifret = run(['exiftool', '-json', filename], capture_output=True)
        if not exifret.returncode == 0:
            raise EXIFException("exiftool returned {}".format(exifret.returncode))
        self.exifjson = json.loads(exifret.stdout)[0]

    def __getitem__(self, key):
        if key in self.exifjson:
            return self.exifjson[key]
        return None

def test(testfile):
    data = EXIFData(testfile)
    pprint.pprint(data.exifjson)
    print("Width: {}".format(data['ImageWidth']))

if __name__ == '__main__':
    test(sys.argv[1])