-
Notifications
You must be signed in to change notification settings - Fork 0
/
change_fn.py
82 lines (61 loc) · 2.23 KB
/
change_fn.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
import os
import time
import shutil
import yaml
from PIL import Image
from PIL.ExifTags import TAGS
def file_info(file_path):
results = {}
i = Image.open(file_path)
info = i._getexif()
for tag, value in info.items():
decoded = TAGS.get(tag, tag)
results[decoded] = value
date = "".join(results["DateTimeOriginal"].split(" ")[0][2:].split(":")[::-1])
return date
def check_number_delete(path, max_num=100):
edit_img_list = os.listdir(path)
if len(edit_img_list) > max_num:
n = int(len(edit_img_list)/max_num)
del edit_img_list[::n]
[os.remove(os.path.join(path, f)) for f in edit_img_list]
print("reduced to: ", len(edit_img_list))
def rename_edit(path):
edit_img_list = os.listdir(path)
for f in edit_img_list:
old_path = os.path.join(path, f)
# date_changed = time.strftime("%d%m%y",time.localtime(os.path.getmtime(old_path)))
date_created = file_info(old_path)
new_path = os.path.join(path, date_created+"_"+f)
os.rename(old_path, new_path)
print("--------------------------")
print(old_path)
print("--->")
print(new_path)
print("--------------------------")
def work_dir(src_path, single_tree=False):
jpg_img_list = [os.path.join(src_path, f) for f in os.listdir(src_path) if f.endswith(".JPG")]
if single_tree:
denom = 4
else:
denom = 20
n = max(int(len(jpg_img_list)/denom), 2)
reduced_list = jpg_img_list[::n]
print("Reduced List to:")
print("copying {0} files".format(len(reduced_list)))
# [print(os.path.join(dst_path, r.split("\\")[-1])) for r in reduced_list]
[shutil.copyfile(r, os.path.join(dst_path, r.split("\\")[-1])) for r in reduced_list]
with open("paths.yaml") as f:
yaml_file = yaml.safe_load(f)
src_path = yaml_file["src_path"]
dst_path = yaml_file["dst_path"]
folder_items = os.listdir(src_path)
if folder_items[0] == "Tree_1":
for sub_dir in folder_items:
print(sub_dir)
path = os.path.join(src_path, sub_dir)
work_dir(path)
elif folder_items[0].endswith(".JPG") or folder_items[0].endswith(".DNG"):
work_dir(src_path)
check_number_delete(dst_path)
rename_edit(dst_path)