src/misc.py

   1 import datetime
   2 import logging
   3 import mimetypes
   4 import os
   5 import PIL.Image
   6 import re
   7 import shutil
   8 import subprocess
   9
  10 def walk_media_files(dir_path):
  11     for root, dirs, files in os.walk(dir_path):
  12         for f in files:
  13             file_path = os.path.join(root, f)
  14             if is_media_file(file_path):
  15                 yield (f, file_path)
  16
  17 def extract_meta_time(file_path):
  18     if is_media_file(file_path, types=['image']):
  19         return _extract_image_timestamp(file_path)
  20     elif is_media_file(file_path, types=['video']):
  21         return _extract_video_timestamp(file_path)
  22
  23 def find_file(dir_path, file_name, file_size, exclude_dir):
  24     for root, dirs, files in os.walk(dir_path):
  25         if root.startswith(exclude_dir):
  26             continue
  27         for f in files:
  28             if f == file_name:
  29                 full_path = os.path.join(root, f)
  30                 if os.path.getsize(full_path) == file_size:
  31                     return root
  32     return None
  33
  34 def import_file(src_file_path, dst_file_path):
  35     shutil.copyfile(src_file_path, dst_file_path)
  36     src_stat = os.stat(src_file_path)
  37     dst_stat = os.stat(dst_file_path)
  38     os.utime( dst_file_path, ns=(dst_stat.st_atime_ns, src_stat.st_mtime_ns) )
  39
  40 def is_media_file(file_path, types=['image', 'video']):
  41     if not os.path.isfile(file_path):
  42         return False
  43     mime_type = mimetypes.guess_type(file_path)[0]
  44     if not mime_type:
  45         return False
  46     if not mime_type.split('/')[0] in types:
  47         return False
  48     return True
  49
  50 def _extract_image_timestamp(file_path):
  51     time = None
  52     try:
  53         with PIL.Image.open(file_path) as image:
  54             exif = image._getexif()
  55             if exif and 36867 in exif:
  56                 time = datetime.datetime\
  57                                .strptime(exif[36867], '%Y:%m:%d %H:%M:%S')\
  58                                .timestamp()
  59     except Exception as e:
  60         logging.warn("Error extracting exif for %s: %s", file_path, str(e))
  61     return time
  62
  63 def _extract_video_timestamp(file_path):
  64     p = subprocess.run(['ffmpeg', '-i', file_path],
  65                        capture_output=True, encoding='UTF-8')
  66     for line in p.stderr.splitlines():
  67         m = re.search('^.*creation_time.*: ([^ ]+)$', line)
  68         if m:
  69             return datetime.datetime\
  70                            .strptime(m.group(1), '%Y-%m-%dT%H:%M:%S.%fZ')\
  71                            .timestamp()
  72     return None