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, QGridLayout, QFrame ) from PyQt5.QtCore import Qt, pyqtSignal, QObject, QTimer from PyQt5.QtGui import QFont, QPalette, QColor 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 ModernButton(QPushButton): def __init__(self, text, parent=None): super().__init__(text, parent) self.setFixedHeight(40) self.setFont(QFont("Arial", 10, QFont.Bold)) class MainWindow(QMainWindow): def __init__(self): super().__init__() self.setWindowTitle("Modern Backup Tool") self.setFixedSize(700, 500) 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 구성 main_layout = QVBoxLayout() main_layout.setContentsMargins(20, 20, 20, 20) main_layout.setSpacing(15) # 상단: 폴더 선택 버튼 folder_layout = QGridLayout() folder_layout.setSpacing(10) self.btn_src = ModernButton("Select Source Folder") self.btn_dst = ModernButton("Select Backup Folder") self.label_src = QLabel("No source selected") self.label_src.setStyleSheet("color: #A0A0A0; font-size: 12px;") self.label_dst = QLabel("No backup destination selected") self.label_dst.setStyleSheet("color: #A0A0A0; font-size: 12px;") folder_layout.addWidget(self.btn_src, 0, 0) folder_layout.addWidget(self.label_src, 0, 1) folder_layout.addWidget(self.btn_dst, 1, 0) folder_layout.addWidget(self.label_dst, 1, 1) # 중단: 제어 버튼 control_layout = QHBoxLayout() control_layout.setSpacing(10) self.btn_start = ModernButton("Start Backup") self.btn_pause = ModernButton("Pause") self.btn_stop = ModernButton("Stop") self.btn_schedule = ModernButton("Auto Backup (5 min)") control_layout.addWidget(self.btn_start) control_layout.addWidget(self.btn_pause) control_layout.addWidget(self.btn_stop) control_layout.addWidget(self.btn_schedule) # 하단: 진행 상태 self.progress = QProgressBar() self.progress.setTextVisible(True) self.progress.setAlignment(Qt.AlignCenter) self.progress.setFixedHeight(25) self.label_file = QLabel("Current file: None") self.label_file.setStyleSheet("font-size: 12px; color: #A0A0A0;") self.text_status = QTextEdit() self.text_status.setReadOnly(True) self.text_status.setStyleSheet("border: 1px solid #333; background: #2A2A2A; color: #E0E0E0; padding: 10px;") self.text_status.setFont(QFont("Consolas", 10)) # 레이아웃 조립 main_layout.addLayout(folder_layout) main_layout.addWidget(self.create_separator()) main_layout.addLayout(control_layout) main_layout.addWidget(self.progress) main_layout.addWidget(self.label_file) main_layout.addWidget(self.text_status) central = QWidget() central.setLayout(main_layout) self.setCentralWidget(central) # 스타일시트 적용 self.apply_stylesheet() # 시그널 연결 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 create_separator(self): separator = QFrame() separator.setFrameShape(QFrame.HLine) separator.setFrameShadow(QFrame.Sunken) separator.setStyleSheet("border: 1px solid #333;") return separator def apply_stylesheet(self): stylesheet = """ QMainWindow { background-color: #1E1E1E; } ModernButton { background-color: #3A3A3A; color: #FFFFFF; border-radius: 8px; padding: 10px; font-weight: bold; } ModernButton:hover { background-color: #4A4A4A; } ModernButton:pressed { background-color: #2A2A2A; } QProgressBar { border: 1px solid #333; border-radius: 5px; background: #2A2A2A; text-align: center; color: #FFFFFF; } QProgressBar::chunk { background-color: #4CAF50; border-radius: 3px; } QLabel { color: #E0E0E0; } """ self.setStyleSheet(stylesheet) def select_src(self): d = QFileDialog.getExistingDirectory(self, "Select Source Folder") if d: self.src_dir = d self.label_src.setText(f"Source: {os.path.basename(self.src_dir)}") self.log_status(f"Source: {self.src_dir}") def select_dst(self): d = QFileDialog.getExistingDirectory(self, "Select Backup Folder") if d: self.dst_dir = d self.label_dst.setText(f"Backup: {os.path.basename(self.dst_dir)}") self.log_status(f"Backup: {self.dst_dir}") def start_backup(self): if not self.src_dir or not self.dst_dir: self.log_status("Please select source and backup folders.") return self.log_status("Backup started!") 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(lambda x: self.label_file.setText(f"Current file: {x}")) 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("Resume") self.log_status("Paused...") else: self.pause_event.set() self.btn_pause.setText("Pause") self.log_status("Resumed") def stop_backup(self): if self.worker: self.stop_event.set() self.pause_event.set() self.log_status("Stop requested.") def backup_done(self): self.log_status("Backup completed successfully.") self.worker = None def backup_stopped(self): self.log_status("Backup stopped.") 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 scheduled_backup(self): """예약된 백업 실행""" if self.src_dir and self.dst_dir: self.log_status("Scheduled backup started") self.start_backup() def toggle_schedule(self): if self.scheduled: self.timer.stop() self.scheduled = False self.btn_schedule.setText("Auto Backup (5 min)") self.log_status("Scheduling stopped") else: self.timer.start(5 * 60 * 1000) # 5분 self.scheduled = True self.btn_schedule.setText("Stop Auto Backup") self.log_status("Auto backup every 5 minutes started") if __name__ == "__main__": app = QApplication(sys.argv) app.setStyle("Fusion") # 기본 스타일 개선 palette = QPalette() palette.setColor(QPalette.Window, QColor(30, 30, 30)) palette.setColor(QPalette.WindowText, QColor(224, 224, 224)) palette.setColor(QPalette.Base, QColor(42, 42, 42)) palette.setColor(QPalette.AlternateBase, QColor(50, 50, 50)) palette.setColor(QPalette.Text, QColor(224, 224, 224)) palette.setColor(QPalette.Button, QColor(58, 58, 58)) palette.setColor(QPalette.ButtonText, QColor(255, 255, 255)) palette.setColor(QPalette.Highlight, QColor(76, 175, 80)) palette.setColor(QPalette.HighlightedText, QColor(255, 255, 255)) app.setPalette(palette) win = MainWindow() win.show() sys.exit(app.exec_())