Blog/remove_metadata.py at main · mhoff12358/Blog · GitHub

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
from PIL import Image

import os
import argparse


def remove_image_metadata(images_path: str):
    for root, dirs, files in os.walk(images_path):
        path = root.split(os.sep)
        for file in files:
            end = file[-4:].lower()
            if end == ".jpg" or end == ".png":
                remove_data(path, file)


def remove_data(path, file):
    file_name = os.path.join(*path, file)
    image = Image.open(file_name)

    data = list(image.getdata())
    image_without_exif = Image.new(image.mode, image.size)
    image_without_exif.putdata(data)

    image_without_exif.save(file_name)

    image_without_exif.close()
    print("Fixed file", file_name)


if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument('-p', '--path')
    args = parser.parse_args()

    remove_image_metadata(os.path.join("root/assets/images", args.path))