import os
import shutil
import sys

# =========================
# 폴더 위치: 작업물(A) / 대상(B) 폴더
# =========================

src_dir = r"Z:\Temp"
dst_dir = r"D:\Temp"

# 로그 파일
missing_log_file = "missing_files.txt"

# =========================
# 실행 전 확인
# =========================

print("\n========== 실행 확인 ==========")
print(f"A 폴더 (작업물): {src_dir}")
print(f"B 폴더 (대상): {dst_dir}")
print("================================")

answer = input("\n덮어쓰기를 진행하시겠습니까? (Y/N): ").strip().lower()

if answer != "y":
    print("\n작업이 취소되었습니다.")
    sys.exit()

print("\n작업을 시작합니다...\n")

# =========================
# 통계 변수
# =========================

overwrite_count = 0
skip_count = 0
not_found_count = 0

total_old_size = 0
total_new_size = 0

missing_files = []

# =========================
# 파일 처리
# =========================

for root, dirs, files in os.walk(src_dir):

    rel_path = os.path.relpath(root, src_dir)
    target_root = os.path.join(dst_dir, rel_path)

    for file in files:

        src_file = os.path.join(root, file)
        dst_file = os.path.join(target_root, file)

        # 대상 파일이 없는 경우
        if not os.path.exists(dst_file):

            not_found_count += 1

            relative_file = os.path.relpath(src_file, src_dir)
            missing_files.append(relative_file)

            continue

        src_size = os.path.getsize(src_file)
        dst_size = os.path.getsize(dst_file)

        # A가 더 작을 때만 덮어쓰기
        if src_size < dst_size:

            total_old_size += dst_size
            total_new_size += src_size

            os.remove(dst_file)
            shutil.move(src_file, dst_file)

            overwrite_count += 1

        else:
            skip_count += 1

# =========================
# 결과 요약
# =========================

saved_size = total_old_size - total_new_size

reduction_percent = (
    (saved_size / total_old_size) * 100
    if total_old_size > 0 else 0
)

def format_size(size):
    for unit in ['B', 'KB', 'MB', 'GB', 'TB']:
        if size < 1024:
            return f"{size:.2f} {unit}"
        size /= 1024

summary = f"""
========== 작업 완료 ==========
덮어쓰기 파일 수      : {overwrite_count}
건너뛴 파일 수        : {skip_count}
대상 파일 없음 수    : {not_found_count}

기존(B) 총 용량       : {format_size(total_old_size)}
새(A) 총 용량         : {format_size(total_new_size)}

절감된 용량           : {format_size(saved_size)}
총 감소율             : {reduction_percent:.2f}%
================================
"""

print(summary)

# 요약 로그 저장
with open("summary_log.txt", "w", encoding="utf-8") as f:
    f.write(summary)

# 대상 없는 파일 목록 저장
with open(missing_log_file, "w", encoding="utf-8") as f:
    for item in missing_files:
        f.write(item + "\n")

print(f"대상 없는 파일 목록 저장 완료: {missing_log_file}")

input("\n엔터를 누르면 종료됩니다...")