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