Add exif data

mediatools/mediafile.py

@@ -19,7 +19,9 @@
 # Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 #
 
+from datetime import datetime
 import re
+from exiftool import ExifToolHelper
 import ffmpeg
 from mediatools import log
 import mediatools.file as fil
@@ -29,6 +31,17 @@
 import mediatools.options as opt
 import mediatools.media_config as conf
 
+EXIF_LONGITUDE_TAG = "EXIF:GPSLongitude"
+EXIF_LONG_REF_TAG = "EXIF:GPSLongitudeRef"
+EXIF_LATITUDE_TAG = "EXIF:GPSLatitude"
+EXIF_LAT_REF_TAG = "EXIF:GPSLatitudeRef"
+
+EXIF_DATE_FMT = "%Y:%m:%d %H:%M:%S"
+ISO_DATE_FMT = "%Y-%m-%d %H:%M:%S"
+TIME_FMT = "%H:%M:%S"
+DATE_FORMATS = (ISO_DATE_FMT, f"{ISO_DATE_FMT}%z", EXIF_DATE_FMT, f"{EXIF_DATE_FMT}%z")
+CREATION_DATE_TAGS = ("QuickTime:CreateDate", "EXIF:DateTimeOriginal", "File:FileModifyDate")
+
 
 class MediaFile(fil.File):
     '''Media file abstraction
@@ -52,6 +65,7 @@ def __init__(self, filename):
         self.title = None
         self.bitrate = None
         self.comment = None
+        self._exif_data = None
         # self.probe()
 
     def __str__(self):
@@ -169,6 +183,108 @@ def __get_top_left__(self, width, height, **kwargs):
                 left = (iw - width) // 2
         return (top, left, pos)
 
+    def get_exif_data(self, force=False):
+        if self._exif_data is None or force:
+            with ExifToolHelper() as et:
+                self._exif_data = et.get_metadata(self.filename)[0]
+        return self._exif_data
+
+    def set_exif_creation_date(self, some_datetime):
+        if type(some_datetime) == datetime:
+            time_to_set = datetime.strftime(some_datetime, EXIF_DATE_FMT)
+        else:
+            time_to_set = some_datetime
+        with ExifToolHelper() as et:
+            et.set_tags([self.filename], tags={"DateTimeOriginal": time_to_set}, params=["-P", "-overwrite_original"])
+
+    def get_exif_creation_date(self):
+        exif_data = self.get_exif_data()
+        if "EXIF:DateTimeOriginal" in exif_data:
+            str_date = exif_data["EXIF:DateTimeOriginal"]
+        elif "File:FileModifyDate" in exif_data:
+            str_date = exif_data["File:FileModifyDate"]
+        else:
+            log.logger.warning("Can't find creation date in %s", util.json_fmt(exif_data))
+            return None
+
+        str_date = creation_date = None
+        for tag in CREATION_DATE_TAGS:
+            if tag in exif_data:
+                str_date = exif_data[tag]
+                break
+        if str_date is None:
+            log.logger.warning("Can't find creation date in %s", util.json_fmt(exif_data))
+            return None
+
+        if str_date == "0000:00:00 00:00:00":
+            str_date = "1900:01:01 00:00:00"
+
+        for fmt in DATE_FORMATS:
+            try:
+                creation_date = datetime.strptime(str_date, fmt)
+            except ValueError:
+                continue
+        if creation_date is None:
+            raise ValueError
+        return creation_date
+
+    def get_exif_device(self):
+        exif_data = self.get_exif_data()
+        device = ""
+        if "EXIF:Make" in exif_data:
+            device = exif_data["EXIF:Make"]
+            if "EXIF:Model" in exif_data:
+                device += " " + exif_data["EXIF:Model"]
+        elif "QuickTime:Author" in exif_data:
+            device = exif_data["QuickTime:Author"]
+        elif "QuickTime:CompressorName" in exif_data:
+            device = exif_data["QuickTime:CompressorName"]
+        return device
+
+    def get_exif_dimensions(self):
+        exif_data = self.get_exif_data()
+        dimensions = ""
+        if "File:ImageWidth" in exif_data and "File:ImageHeight" in exif_data:
+            dimensions = f'{exif_data["File:ImageWidth"]}x{exif_data["File:ImageHeight"]}'
+        elif "EXIF:ExifImageWidth" in exif_data and "EXIF:ExifImageHeight" in exif_data:
+            dimensions = f'{exif_data["EXIF:ExifImageWidth"]}x{exif_data["EXIF:ExifImageHeight"]}'
+        elif "QuickTime:SourceImageWidth" in exif_data and "QuickTime:SourceImageHeight" in exif_data:
+            dimensions = f'{exif_data["QuickTime:SourceImageWidth"]}x{exif_data["QuickTime:SourceImageHeight"]}'
+        elif "QuickTime:ImageWidth" in exif_data and "QuickTime:ImageHeight" in exif_data:
+            dimensions = f'{exif_data["QuickTime:ImageWidth"]}x{exif_data["QuickTime:ImageHeight"]}'
+        elif "Composite:ImageSize" in exif_data:
+            dimensions = exif_data["Composite:ImageSize"].replace(" ", "x")
+        return dimensions
+
+    def get_exif_gps_coordinates(self):
+        (lat, long) = (None, None)
+        exif_data = self.get_exif_data()
+        if EXIF_LATITUDE_TAG in exif_data:
+            lat = exif_data[EXIF_LATITUDE_TAG]
+            if exif_data.get(EXIF_LAT_REF_TAG, "N") == "S":
+                lat = "-" + lat
+        if EXIF_LONGITUDE_TAG in exif_data:
+            long = exif_data[EXIF_LONGITUDE_TAG]
+            if exif_data.get(EXIF_LONG_REF_TAG, "E") == "W":
+                long = "-" + long
+        return (float(lat), float(long))
+
+    def set_exif_gps_coordinates(self, latitude, longitude):
+        if latitude < 0:
+            latitude = -latitude
+            lat_ref = "S"
+        if longitude < 0:
+            longitude = -longitude
+            long_ref = "W"
+        with ExifToolHelper() as et:
+            et.set_tags([self.filename], tags={
+                "EXIF:GPSLatitude": latitude,
+                "EXIF:GPSLatitudeRef": lat_ref,
+                "EXIF:GPSLongitude": longitude,
+                "EXIF:GPSLongitudeRef": long_ref
+            },
+                params=["-P", "-overwrite_original"])
+
 
 # ---------------- Class methods ---------------------------------
 
@@ -300,6 +416,26 @@ def compute_fps(rate):
     return rate
 
 
+def get_exif_creation_date(filename):
+    return MediaFile(filename).get_exif_creation_date()
+
+
+def set_exif_creation_date(filename, some_datetime):
+    MediaFile(filename).set_exif_creation_date(some_datetime)
+
+
+def get_dimensions(filename):
+    return MediaFile(filename).get_dimensions()
+
+
+def get_exif_gps_coordinates(filename):
+    return MediaFile(filename).get_exif_gps_coordinates()
+
+
+def set_exif_gps_coordinates(filename, latitude, longitude):
+    MediaFile(filename).set_exif_gps_coordinates(latitude, longitude)
+
+
 def reduce_aspect_ratio(aspect_ratio, height=None):
     ''' Reduces the Aspect ratio calculation in prime factors '''
     if height is None:

mediatools/videofile.py

@@ -22,6 +22,8 @@
 '''Video file tools'''
 
 from __future__ import print_function
+import datetime
+from exiftool import ExifToolHelper
 import re
 from mediatools import log
 import mediatools.exceptions as ex
@@ -374,6 +376,62 @@ def encode(self, target_file=None, profile=None, **kwargs):
         log.logger.info("File %s encoded", target_file)
         return target_file
 
+    def set_creation_date(self, some_datetime):
+        if type(some_datetime) == datetime.datetime:
+            time_to_set = datetime.strftime(some_datetime, media.EXIF_DATE_FMT)
+        else:
+            time_to_set = some_datetime
+        p = ["-P", "-overwrite_original"]
+        with ExifToolHelper() as et:
+            et.set_tags([self.filename], tags={
+                "CreateDate": time_to_set,
+                "ModifyDate": time_to_set,
+                "DateTimeOriginal": time_to_set
+            }, params=p)
+            et.set_tags([self.filename], tags={
+                "EXIF:CreateDate": time_to_set,
+                "EXIF:ModifyDate": time_to_set,
+                "EXIF:DateTimeOriginal": time_to_set
+            }, params=p)
+            et.set_tags([self.filename], tags={
+                "Composite:SubSecCreateDate": time_to_set,
+                "Composite:SubSecDateTimeOriginal": time_to_set,
+                "Composite:SubSecModifyDate": time_to_set,
+                "Quicktime:CreateDate": time_to_set,
+                "Quicktime:DateTimeOriginal": time_to_set,
+                "QuickTime:MediaCreateDate": time_to_set,
+                "QuickTime:MediaModifyDate": time_to_set,
+                "QuickTime:TrackCreateDate": time_to_set,
+                "QuickTime:TrackModifyDate": time_to_set,
+                "QuickTime:CreateDate": time_to_set,
+                "QuickTime:ModifyDate": time_to_set
+            }, params=p)
+
+    def get_exif_bitrate(self):
+        exif_data = self.get_exif_data()
+        bitrate = None
+        if "Composite:AvgBitrate" in exif_data:
+            bitrate = round(int(exif_data["Composite:AvgBitrate"]) / 1024 / 1024)
+        return bitrate
+
+    def get_exif_codec(self):
+        exif_data = self.get_exif_data()
+        codec = ""
+        if "QuickTime:CompressorID" in exif_data:
+            codec = exif_data["QuickTime:CompressorID"]
+        if codec == "avc1":
+            codec = "h264"
+        elif codec == "hev1":
+            codec = "h265"
+        return codec
+
+    def get_exif_fps(self):
+        exif_data = self.get_exif_data()
+        fps = None
+        if "QuickTime:VideoFrameRate" in exif_data:
+            fps = round(float(exif_data["QuickTime:VideoFrameRate"]))
+        return fps
+
     # ----------------- Private methods ------------------------------------------
 
     def __get_number_of_audio_tracks(self):