ExifReader.py

import imghdr
import os

from PIL import Image
from PIL.ExifTags import TAGS


class ExifReader():
    def __init__(self, debug=False):
        self.debug = debug

    # Recording Times
    def get_recordingtimes_from_folder(self, folder):
        results = []
        for file in os.listdir(folder):
            tmp = self.get_recordingtimes_from_file(folder + file)
            if tmp:
                results.append(tmp)
        return results

    def get_recordingtimes_from_file(self, file):
        return self.get_recordingtimes_from_exifdata(self.get_exifdata_from_file(file))

    def get_recordingtimes_from_exifdata(self, data):
        results = []
        for set in data:
            if 'DateTime' in set:
                results.append(set['DateTime'])
            else:
                results.append('Kein Datum')
        return results

    def get_coordinates_from_file(self, file):
        return self.get_coordinates(self.get_exifdata_from_file(file))

    def get_coordinates_for_folder(self, folder):
        results = []
        for file in os.listdir(folder):
            try:
                if imghdr.what(folder + file):
                    tmp = self.get_coordinates_from_file(folder + file)
            except(PermissionError):
                if self.debug:
                    print('Permission to Subfolder denied')
            if tmp:
                results.append(tmp)
        return results

    def get_coordinates_for_exifdata(self, data):
        results = []
        for set in data:
            results.append(self.get_coordinates(set))
        return results

    def get_coordinates(self, exifdata):
        if 'GPSInfo' in exifdata and len(exifdata['GPSInfo']) > 4:
            nsdeg = exifdata['GPSInfo'][2][0][0]
            nsmin = exifdata['GPSInfo'][2][1][0]
            nssec = exifdata['GPSInfo'][2][2][0] / 1000
            if exifdata['GPSInfo'][1] == 'S':
                south = True
            else:
                south = False

            ewdeg = exifdata['GPSInfo'][4][0][0]
            ewmin = exifdata['GPSInfo'][4][1][0]
            ewsec = exifdata['GPSInfo'][4][2][0] / 1000
            if exifdata['GPSInfo'][3] == 'W':
                west = True
            else:
                west = False

            if self.debug:
                print(str(nsdeg) + '°' + str(nsmin) + '\'' + str(nssec) + '\"' + exifdata['GPSInfo'][1])
                print(str(ewdeg) + '°' + str(ewmin) + '\'' + str(ewsec) + '\"' + exifdata['GPSInfo'][3])

            long = ewdeg + ewmin / 60 + ewsec / 3600
            lat = nsdeg + nsmin / 60 + nssec / 3600
            if west:
                long *= -1
            if south:
                lat *= -1

            return [lat, long]
        else:
            return None

    # Exif data
    def get_exifdata_from_file(self, file):
        try:
            img = Image.open(file)
            info = img._getexif()
            ret = {}
            useless = ['MakerNote']
            for tag, value in info.items():
                decoded = TAGS.get(tag, tag)
                if decoded not in useless:
                    ret[decoded] = value
            return ret
        except(OSError):
            print("Wrong file format. Must be an image file.")

    def get_exif_data_from_folders(self, folders):
        data = []
        for folder in folders:
            data.extend(self.get_exif_data_from_folder(folder))
        return data

    def get_exif_data_from_folder(self, folder):
        results = []
        for file in os.listdir(folder):
            try:
                if imghdr.what(folder + file):
                    tmp = self.get_exifdata_from_file(folder + file)
                    if tmp:
                        results.append(tmp)
            except(PermissionError):
                if self.debug:
                    print('Permission to Subfolder denied')
        return results