Skip to content

Commit 1c869fb

Browse files
committed
feat: add tqdm
1 parent 2b1cb8f commit 1c869fb

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

python_tiny_projects/remove_duplicates/remove_duplicates.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,14 +4,15 @@
44
import ipyplot
55
import numpy as np
66
from PIL import Image
7+
from tqdm import tqdm
78

89

910
path = './data_test'
1011

1112
file_list = os.walk(path)
1213

1314
file_hashes = dict()
14-
for root,folders,files in file_list:
15+
for root,folders,files in tqdm(file_list):
1516
for file in files:
1617
path = Path(os.path.join(root,file))
1718
fhash = hashlib.md5(open(path,'rb').read()).hexdigest()
@@ -24,7 +25,7 @@
2425
duplicate_file_paths = [x[1] for x in file_hashes.items() if len(x[1])>1]
2526

2627
get_len = lambda x: len(x.stem)
27-
for duplicates in duplicate_file_paths:
28+
for duplicates in tqdm(duplicate_file_paths):
2829
keep_index = np.argmin(list(map(get_len, duplicates)))
2930
for i, duplicate in enumerate(duplicates):
3031
if keep_index != i:

0 commit comments

Comments
 (0)