36 lines
1.0 KiB
Python
36 lines
1.0 KiB
Python
# filter.py
|
|
import shutil
|
|
from pathlib import Path
|
|
|
|
# 경로 설정
|
|
DATA_ORIGIN_DIR = Path("data_origin")
|
|
DATA_DIR = Path("data")
|
|
DATA_JSON_DIR = Path("data_json")
|
|
|
|
|
|
def main():
|
|
# data_json 폴더 없으면 생성
|
|
DATA_JSON_DIR.mkdir(parents=True, exist_ok=True)
|
|
|
|
# data_origin의 파일명(확장자 제외) 집합 생성
|
|
origin_names = {f.stem for f in DATA_ORIGIN_DIR.iterdir() if f.is_file()}
|
|
print(f"[INFO] data_origin 파일명(확장자 제외) 개수: {len(origin_names)}")
|
|
|
|
moved_count = 0
|
|
|
|
# data 폴더 순회
|
|
for file_path in DATA_DIR.iterdir():
|
|
if not file_path.is_file():
|
|
continue
|
|
if file_path.stem in origin_names:
|
|
target_path = DATA_JSON_DIR / file_path.name
|
|
shutil.move(str(file_path), target_path)
|
|
moved_count += 1
|
|
print(f"[MOVE] {file_path.name} -> {target_path}")
|
|
|
|
print(f"[DONE] 이동된 파일 수: {moved_count}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|