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
| import os
import re
import uuid
post_dir = "source/_posts/"
image_dir = "source/images/"
def get_ref_filename(root_dir):
referenced_files = set()
for dirpath, _, filenames in os.walk(root_dir):
for filename in filenames:
if filename.endswith(".md"):
file_path = os.path.join(dirpath, filename)
with open(file_path, "r") as file:
contents = file.read()
matches = re.findall(r"images\/(.+\..+)\)", contents)
referenced_files.update(matches)
return referenced_files
def get_source_filenames(directory):
picture_files = []
valid_extensions = ['.jpg', '.jpeg', '.png', '.gif', '.bmp']
for filename in os.listdir(directory):
file_extension = os.path.splitext(filename)[1].lower()
if file_extension in valid_extensions:
picture_files.append(filename)
return set(picture_files)
def del_unref_files():
referenced_files = get_ref_filename(post_dir)
pic_files = get_source_filenames(image_dir)
unref_file = list(pic_files - referenced_files)
print(unref_file)
for e in unref_file:
try:
os.remove(image_dir + e)
print(f"Deleted file: {e}")
except OSError as e:
print(f"Error deleting file: {e} - {e}")
def rename_ref_in_post(oldname, newname):
for dirpath, _, filenames in os.walk(post_dir):
for filename in filenames:
if filename.endswith(".md"):
file_path = os.path.join(dirpath, filename)
with open(file_path, "r+") as file:
contents = file.read()
contents = contents.replace(oldname, newname)
file.seek(0)
file.truncate()
file.write(contents)
def random_filename(post_dir, image_dir):
fms = get_ref_filename(post_dir)
for fm in fms:
suffix = fm.split(".")[-1]
uid = str(uuid.uuid4())[:6]
os.rename(f"{image_dir}{fm}", f"{image_dir}{uid}.{suffix}")
rename_ref_in_post(fm, f"{uid}.{suffix}")
if __name__ == '__main__':
random_filename(post_dir, image_dir)
|