import sys import os import hashlib import shutil import threading import time from datetime import datetime, timedelta from PyQt5.QtWidgets import ( QApplication, QMainWindow, QPushButton, QVBoxLayout, QWidget, QFileDialog, QProgressBar, QLabel, QTextEdit, QHBoxLayout ) from PyQt5.QtCore import Qt, pyqtSignal, QObject, QTimer LOG_FILE = 'backup_log.txt' HISTORY_FILE = 'backup_history.txt' MAX_BACKUPS = 5 # 자동 정리: 최대 백업본 개수 def file_hash(path): """파일의 SHA256 해시값 계산 (중복 체크용)""" h = hashlib.sha256() with open(path, 'rb') as f: while chunk := f.read(8192): h.update(chunk) return h.hexdigest() class BackupWorker(QObject): progress = pyqtSignal(int) file_copied = pyqtSignal(str) status = pyqtSignal(str) finished = pyqtSignal() stopped = pyqtSignal() def __init__(self, src, dst, pause_event, stop_event): super().__init__() self.src = src self.dst = dst self.pause_event = pause_event self.stop_event = stop_event def run(self): try: self.status.emit("파일 리스트 수집 중...") file_list = [] for root, dirs, files in os.walk(self.src): for name in files: full = os.path.join(root, name) rel = os.path.relpath(full, self.src) file_list.append(rel) total = len(file_list) copied = 0 # 히스토리 불러오기 (중복, 증분 체크) old_hashes = self._load_backup_history() # 새 백업본 폴더 (타임스탬프) backup_name = datetime.now().strftime('%Y%m%d_%H%M%S') backup_path = os.path.join(self.dst, backup_name) os.makedirs(backup_path, exist_ok=True) self.status.emit(f"백업 폴더 생성: {backup_path}") new_hashes = {} for idx, rel_path in enumerate(file_list): src_file = os.path.join(self.src, rel_path) dst_file = os.path.join(backup_path, rel_path) if self.stop_event.is_set(): self.status.emit("백업 중지됨") self.stopped.emit() return self.pause_event.wait() # 일시정지 시 block # 파일 해시로 변경 여부 확인 (증분/중복) hash_val = file_hash(src_file) new_hashes[rel_path] = hash_val if rel_path in old_hashes and old_hashes[rel_path] == hash_val: self.status.emit(f"[SKIP] {rel_path} (변경 없음)") else: os.makedirs(os.path.dirname(dst_file), exist_ok=True) shutil.copy2(src_file, dst_file) self.file_copied.emit(rel_path) self.status.emit(f"[COPY] {rel_path}") copied += 1 progress = int(copied / total * 100) self.progress.emit(progress) self._save_backup_history(new_hashes) self.status.emit("백업 완료!") self._auto_cleanup() self.finished.emit() except Exception as e: self.status.emit(f"오류 발생: {e}") self.finished.emit() def _load_backup_history(self): if not os.path.exists(HISTORY_FILE): return {} try: with open(HISTORY_FILE, 'r', encoding='utf-8') as f: lines = f.readlines() return dict(line.strip().split('\t') for line in lines if '\t' in line) except: return {} def _save_backup_history(self, hashes): with open(HISTORY_FILE, 'w', encoding='utf-8') as f: for rel, hashv in hashes.items(): f.write(f"{rel}\t{hashv}\n") def _auto_cleanup(self): """MAX_BACKUPS 초과시 가장 오래된 백업 삭제""" try: backups = sorted([ d for d in os.listdir(self.dst) if os.path.isdir(os.path.join(self.dst, d)) ]) if len(backups) > MAX_BACKUPS: for b in backups[:-MAX_BACKUPS]: b_path = os.path.join(self.dst, b) shutil.rmtree(b_path) self.status.emit(f"오래된 백업 삭제: {b_path}") except Exception as e: self.status.emit(f"[정리오류] {e}") class MainWindow(QMainWindow): def __init__(self): super().__init__() self.setWindowTitle("증분 백업 툴") self.setFixedSize(600, 400) self.src_dir = "" self.dst_dir = "" self.backup_thread = None self.worker = None self.pause_event = threading.Event() self.pause_event.set() self.stop_event = threading.Event() self.timer = QTimer() self.timer.timeout.connect(self.scheduled_backup) # UI 구성 layout = QVBoxLayout() btn_layout = QHBoxLayout() self.btn_src = QPushButton("원본 폴더 선택") self.btn_dst = QPushButton("백업 폴더 선택") self.btn_start = QPushButton("시작") self.btn_pause = QPushButton("일시정지") self.btn_stop = QPushButton("중지") self.btn_schedule = QPushButton("5분마다 자동 백업") self.progress = QProgressBar() self.label_file = QLabel("파일명:") self.text_status = QTextEdit() self.text_status.setReadOnly(True) layout.addWidget(self.btn_src) layout.addWidget(self.btn_dst) btn_layout.addWidget(self.btn_start) btn_layout.addWidget(self.btn_pause) btn_layout.addWidget(self.btn_stop) btn_layout.addWidget(self.btn_schedule) layout.addLayout(btn_layout) layout.addWidget(self.progress) layout.addWidget(self.label_file) layout.addWidget(self.text_status) central = QWidget() central.setLayout(layout) self.setCentralWidget(central) # 시그널 연결 self.btn_src.clicked.connect(self.select_src) self.btn_dst.clicked.connect(self.select_dst) self.btn_start.clicked.connect(self.start_backup) self.btn_pause.clicked.connect(self.pause_resume) self.btn_stop.clicked.connect(self.stop_backup) self.btn_schedule.clicked.connect(self.toggle_schedule) self.scheduled = False def select_src(self): d = QFileDialog.getExistingDirectory(self, "원본 폴더 선택") if d: self.src_dir = d self.log_status(f"원본: {self.src_dir}") def select_dst(self): d = QFileDialog.getExistingDirectory(self, "백업 폴더 선택") if d: self.dst_dir = d self.log_status(f"백업: {self.dst_dir}") def start_backup(self): if not self.src_dir or not self.dst_dir: self.log_status("원본/백업 폴더를 선택하세요.") return self.log_status("백업 시작!") self.progress.setValue(0) self.stop_event.clear() self.pause_event.set() self.worker = BackupWorker(self.src_dir, self.dst_dir, self.pause_event, self.stop_event) self.worker.progress.connect(self.progress.setValue) self.worker.file_copied.connect(self.label_file.setText) self.worker.status.connect(self.log_status) self.worker.finished.connect(self.backup_done) self.worker.stopped.connect(self.backup_stopped) self.backup_thread = threading.Thread(target=self.worker.run, daemon=True) self.backup_thread.start() def pause_resume(self): if not self.worker: return if self.pause_event.is_set(): self.pause_event.clear() self.btn_pause.setText("재개") self.log_status("일시정지 중...") else: self.pause_event.set() self.btn_pause.setText("일시정지") self.log_status("재개") def stop_backup(self): if self.worker: self.stop_event.set() self.pause_event.set() self.log_status("중지 요청됨.") def backup_done(self): self.log_status("백업이 정상적으로 종료되었습니다.") self.worker = None def backup_stopped(self): self.log_status("백업이 중지되었습니다.") self.worker = None def log_status(self, msg): now = datetime.now().strftime("%Y-%m-%d %H:%M:%S") line = f"[{now}] {msg}" self.text_status.append(line) with open(LOG_FILE, 'a', encoding='utf-8') as f: f.write(line + '\n') def toggle_schedule(self): if self.scheduled: self.timer.stop() self.scheduled = False self.btn_schedule.setText("5분마다 자동 백업") self.log_status("스케줄링 중지") else: self.timer.start(5 * 60 * 1000) # 5분 self.scheduled = True self.btn_schedule.setText("스케줄 중지") self.log_status("5분마다 자동 백업 시작") def scheduled_backup(self): if not self.worker: self.start_backup() if __name__ == "__main__": app = QApplication(sys.argv) win = MainWindow() win.show() sys.exit(app.exec_())