DeepMedAI / cleanup_duplicates.py
PBThuong's picture
Thiết lập lại thư viện y khoa sạch và cập nhật chroma_db
8eaa451
Raw
History Blame Contribute Delete
1.59 kB
import os
import re
DATA_DIRS = [
r"backend\data\thông tin thuốc nội bộ",
r"backend\data\cảnh giác dược"
]
# Patterns for duplicated files
# Pattern 1: ...__c70a15f1.md (8 random hex characters after __)
pattern1 = re.compile(r"__[a-f0-9]{8,}\.md$", re.IGNORECASE)
# Pattern 2: ... 2010bc1073c2815493bdccda01af5459.md (32 hex chars with a preceding space)
pattern2 = re.compile(r" [a-f0-9]{32}.*\.(md|csv)$", re.IGNORECASE)
def cleanup_folder(folder_path):
if not os.path.exists(folder_path):
print(f"Không tìm thấy thư mục: {folder_path}")
return
count = 0
size_saved = 0
for file_name in os.listdir(folder_path):
if pattern1.search(file_name) or pattern2.search(file_name):
file_path = os.path.join(folder_path, file_name)
try:
size = os.path.getsize(file_path)
os.remove(file_path)
print(f"Đã xoá: {file_name}")
count += 1
size_saved += size
except Exception as e:
print(f"Không thể xoá {file_name}: {e}")
print(f"--- Dọn dẹp xong {folder_path} ---")
print(f"Xoá tổng cộng: {count} tập tin rác.")
print(f"Tiết kiệm được: {size_saved / 1024 / 1024:.2f} MB\n")
if __name__ == "__main__":
print("BẮT ĐẦU DỌN DẸP...\n")
base_dir = os.path.dirname(os.path.abspath(__file__))
for d in DATA_DIRS:
folder_path = os.path.join(base_dir, d)
cleanup_folder(folder_path)
print("HOÀN TẤT DỌN DẸP!")