]> git.treefish.org Git - photosort.git/blob - src/misc.py
added missing import
[photosort.git] / 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