import sys
import os
import datetime
from PyQt5.QtWidgets import *
from PyQt5.QtCore import *
from PyQt5.QtGui import *

class FileListGenerator(QWidget):
    def __init__(self):
        super().__init__()
        self.current_directory = ""
        self.init_ui()
        self.apply_styles()
        
    def init_ui(self):
        self.setWindowTitle('文件列表生成器 - 專業(yè)版')
        self.setWindowIcon(self.style().standardIcon(QStyle.SP_DirIcon))
        self.setGeometry(350, 250, 900, 650)
        
        main_layout = QVBoxLayout()
        main_layout.setSpacing(12)
        main_layout.setContentsMargins(20, 20, 20, 20)
        
        header = QLabel('?? 智能文件列表生成器')
        header.setAlignment(Qt.AlignCenter)
        header_font = QFont()
        header_font.setPointSize(18)
        header_font.setBold(True)
        header.setFont(header_font)
        main_layout.addWidget(header)
        
        description = QLabel('選擇目錄后自動(dòng)掃描所有文件，支持遞歸遍歷和多種導(dǎo)出格式')
        description.setAlignment(Qt.AlignCenter)
        description.setWordWrap(True)
        main_layout.addWidget(description)
        
        self.setup_directory_section(main_layout)
        self.setup_control_panel(main_layout)
        self.setup_file_display(main_layout)
        self.setup_export_section(main_layout)
        self.setup_status_bar(main_layout)
        
        self.setLayout(main_layout)
        
    def apply_styles(self):
        self.setStyleSheet("""
            QWidget {
                background-color: #f5f7fa;
                font-family: 'Segoe UI', 'Microsoft YaHei', sans-serif;
            }
            QLabel {
                color: #2c3e50;
            }
            QPushButton {
                background-color: #3498db;
                color: white;
                border: none;
                padding: 8px 16px;
                border-radius: 4px;
                font-weight: 500;
                min-height: 32px;
            }
            QPushButton:hover {
                background-color: #2980b9;
            }
            QPushButton:pressed {
                background-color: #1c6ea4;
            }
            QPushButton:disabled {
                background-color: #bdc3c7;
            }
            QPushButton#successButton {
                background-color: #27ae60;
            }
            QPushButton#successButton:hover {
                background-color: #219653;
            }
            QLineEdit, QTextEdit {
                border: 1px solid #dce4ec;
                border-radius: 4px;
                padding: 8px;
                background-color: white;
                selection-background-color: #3498db;
            }
            QLineEdit:focus, QTextEdit:focus {
                border: 1px solid #3498db;
                outline: none;
            }
            QGroupBox {
                border: 2px solid #ecf0f1;
                border-radius: 6px;
                margin-top: 10px;
                padding-top: 15px;
                font-weight: bold;
                color: #34495e;
                background-color: white;
            }
            QGroupBox::title {
                subcontrol-origin: margin;
                left: 10px;
                padding: 0 8px 0 8px;
            }
            QProgressBar {
                border: 1px solid #bdc3c7;
                border-radius: 4px;
                text-align: center;
                background-color: #ecf0f1;
            }
            QProgressBar::chunk {
                background-color: #2ecc71;
                border-radius: 4px;
            }
            QStatusBar {
                background-color: #34495e;
                color: #ecf0f1;
            }
            QCheckBox {
                spacing: 8px;
            }
            QCheckBox::indicator {
                width: 18px;
                height: 18px;
            }
        """)
        
    def setup_directory_section(self, parent_layout):
        group = QGroupBox("目錄選擇")
        layout = QVBoxLayout()
        
        path_layout = QHBoxLayout()
        self.path_label = QLabel("尚未選擇目錄")
        self.path_label.setWordWrap(True)
        self.path_label.setMinimumHeight(40)
        self.path_label.setAlignment(Qt.AlignCenter)
        
        browse_btn = QPushButton("選擇目錄")
        browse_btn.setIcon(self.style().standardIcon(QStyle.SP_DirOpenIcon))
        browse_btn.clicked.connect(self.select_directory)
        
        path_layout.addWidget(self.path_label, 4)
        path_layout.addWidget(browse_btn, 1)
        layout.addLayout(path_layout)
        
        group.setLayout(layout)
        parent_layout.addWidget(group)
        
    def setup_control_panel(self, parent_layout):
        group = QGroupBox("掃描選項(xiàng)")
        layout = QHBoxLayout()
        
        self.recursive_check = QCheckBox("遞歸掃描子目錄")
        self.recursive_check.setChecked(True)
        
        self.include_hidden_check = QCheckBox("包含隱藏文件")
        
        self.sort_check = QCheckBox("按名稱排序")
        self.sort_check.setChecked(True)
        
        layout.addWidget(self.recursive_check)
        layout.addWidget(self.include_hidden_check)
        layout.addWidget(self.sort_check)
        layout.addStretch()
        
        self.scan_btn = QPushButton("開始掃描")
        self.scan_btn.setEnabled(False)
        self.scan_btn.clicked.connect(self.scan_files)
        layout.addWidget(self.scan_btn)
        
        group.setLayout(layout)
        parent_layout.addWidget(group)
        
    def setup_file_display(self, parent_layout):
        group = QGroupBox("文件列表預(yù)覽")
        layout = QVBoxLayout()
        
        stats_layout = QHBoxLayout()
        self.file_count_label = QLabel("文件總數(shù): 0")
        self.total_size_label = QLabel("總大小: 0 B")
        
        stats_layout.addWidget(self.file_count_label)
        stats_layout.addStretch()
        stats_layout.addWidget(self.total_size_label)
        layout.addLayout(stats_layout)
        
        self.progress_bar = QProgressBar()
        self.progress_bar.setVisible(False)
        layout.addWidget(self.progress_bar)
        
        self.file_list_display = QTextEdit()
        self.file_list_display.setReadOnly(True)
        self.file_list_display.setMinimumHeight(300)
        font = QFont("Consolas", 10)
        self.file_list_display.setFont(font)
        layout.addWidget(self.file_list_display)
        
        group.setLayout(layout)
        parent_layout.addWidget(group)
        
    def setup_export_section(self, parent_layout):
        group = QGroupBox("導(dǎo)出設(shè)置")
        layout = QVBoxLayout()
        
        filename_layout = QHBoxLayout()
        filename_layout.addWidget(QLabel("文件名:"))
        
        self.filename_input = QLineEdit()
        self.filename_input.setText(f"file_list_{datetime.datetime.now().strftime('%Y%m%d')}.txt")
        filename_layout.addWidget(self.filename_input, 1)
        
        format_layout = QHBoxLayout()
        format_layout.addWidget(QLabel("格式:"))
        
        self.txt_radio = QRadioButton("TXT")
        self.txt_radio.setChecked(True)
        self.csv_radio = QRadioButton("CSV")
        
        format_layout.addWidget(self.txt_radio)
        format_layout.addWidget(self.csv_radio)
        format_layout.addStretch()
        
        layout.addLayout(filename_layout)
        layout.addLayout(format_layout)
        
        self.export_btn = QPushButton("導(dǎo)出文件")
        self.export_btn.setObjectName("successButton")
        self.export_btn.setEnabled(False)
        self.export_btn.clicked.connect(self.export_list)
        layout.addWidget(self.export_btn, 0, Qt.AlignRight)
        
        group.setLayout(layout)
        parent_layout.addWidget(group)
        
    def setup_status_bar(self, parent_layout):
        self.status_bar = QStatusBar()
        self.status_bar.showMessage("就緒")
        parent_layout.addWidget(self.status_bar)
        
    def select_directory(self):
        directory = QFileDialog.getExistingDirectory(
            self,
            "選擇目錄",
            QDir.homePath(),
            QFileDialog.ShowDirsOnly | QFileDialog.DontResolveSymlinks
        )
        
        if directory:
            self.current_directory = directory
            self.path_label.setText(directory)
            self.path_label.setToolTip(directory)
            self.scan_btn.setEnabled(True)
            self.status_bar.showMessage(f"已選擇目錄: {directory}")
            
    def scan_files(self):
        if not self.current_directory:
            return
            
        self.file_list_display.clear()
        self.progress_bar.setVisible(True)
        self.progress_bar.setValue(0)
        QApplication.processEvents()
        
        try:
            all_files = []
            total_size = 0
            
            if self.recursive_check.isChecked():
                walker = os.walk(self.current_directory)
                for root, dirs, files in walker:
                    if not self.include_hidden_check.isChecked():
                        dirs[:] = [d for d in dirs if not d.startswith('.')]
                        files = [f for f in files if not f.startswith('.')]
                    
                    for file in files:
                        full_path = os.path.join(root, file)
                        rel_path = os.path.relpath(full_path, self.current_directory)
                        all_files.append(rel_path)
                        
                        try:
                            total_size += os.path.getsize(full_path)
                        except:
                            pass
            else:
                for item in os.listdir(self.current_directory):
                    full_path = os.path.join(self.current_directory, item)
                    if os.path.isfile(full_path):
                        if not self.include_hidden_check.isChecked() and item.startswith('.'):
                            continue
                        all_files.append(item)
                        
                        try:
                            total_size += os.path.getsize(full_path)
                        except:
                            pass
            
            if self.sort_check.isChecked():
                all_files.sort(key=lambda x: x.lower())
            
            self.progress_bar.setValue(50)
            QApplication.processEvents()
            
            self.display_file_list(all_files, total_size)
            self.progress_bar.setValue(100)
            
            self.export_btn.setEnabled(True)
            self.status_bar.showMessage(f"掃描完成，找到 {len(all_files)} 個(gè)文件")
            
            QTimer.singleShot(1000, lambda: self.progress_bar.setVisible(False))
            
        except Exception as e:
            QMessageBox.critical(self, "掃描錯(cuò)誤", f"掃描過(guò)程中發(fā)生錯(cuò)誤:\n{str(e)}")
            self.status_bar.showMessage("掃描失敗")
            self.progress_bar.setVisible(False)
            
    def display_file_list(self, files, total_size):
        self.file_list_display.clear()
        
        header = f"目錄掃描報(bào)告\n"
        header += f"生成時(shí)間: {datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')}\n"
        header += f"掃描目錄: {self.current_directory}\n"
        header += f"文件數(shù)量: {len(files)}\n"
        header += f"總文件大小: {self.format_size(total_size)}\n"
        header += "=" * 60 + "\n\n"
        
        self.file_list_display.setPlainText(header)
        
        for i, file_path in enumerate(files, 1):
            self.file_list_display.append(f"{i:4d}. {file_path}")
            
            if i % 100 == 0:
                QApplication.processEvents()
        
        self.file_count_label.setText(f"文件總數(shù): {len(files)}")
        self.total_size_label.setText(f"總大小: {self.format_size(total_size)}")
        
    def format_size(self, size_bytes):
        for unit in ['B', 'KB', 'MB', 'GB', 'TB']:
            if size_bytes < 1024.0:
                return f"{size_bytes:.2f} {unit}"
            size_bytes /= 1024.0
        return f"{size_bytes:.2f} PB"
        
    def export_list(self):
        if not self.file_list_display.toPlainText().strip():
            QMessageBox.warning(self, "警告", "沒(méi)有可導(dǎo)出的文件列表")
            return
            
        filename = self.filename_input.text().strip()
        if not filename:
            QMessageBox.warning(self, "警告", "請(qǐng)輸入文件名")
            return
            
        default_ext = ".csv" if self.csv_radio.isChecked() else ".txt"
        if not filename.lower().endswith(('.txt', '.csv')):
            filename += default_ext
            
        save_path, _ = QFileDialog.getSaveFileName(
            self,
            "保存文件",
            os.path.join(QDir.homePath(), filename),
            "文本文件 (*.txt);;CSV文件 (*.csv);;所有文件 (*.*)"
        )
        
        if save_path:
            try:
                content = self.file_list_display.toPlainText()
                
                if self.csv_radio.isChecked() and save_path.endswith('.csv'):
                    lines = content.split('\n')
                    csv_content = "序號(hào),文件路徑\n"
                    for i, line in enumerate(lines[8:], 1):
                        if line.strip() and not line.startswith('='):
                            if '. ' in line:
                                csv_content += f"{i},{line.split('. ', 1)[1]}\n"
                else:
                    csv_content = content
                
                with open(save_path, 'w', encoding='utf-8') as f:
                    f.write(csv_content)
                    
                QMessageBox.information(
                    self,
                    "導(dǎo)出成功",
                    f"文件已成功導(dǎo)出到:\n{save_path}\n\n"
                    f"文件大小: {os.path.getsize(save_path):,} 字節(jié)"
                )
                
                self.status_bar.showMessage(f"文件已導(dǎo)出: {os.path.basename(save_path)}")
                
            except Exception as e:
                QMessageBox.critical(self, "導(dǎo)出失敗", f"導(dǎo)出文件時(shí)出錯(cuò):\n{str(e)}")
                self.status_bar.showMessage("導(dǎo)出失敗")

def main():
    app = QApplication(sys.argv)
    
    app.setApplicationName("文件列表生成器")
    app.setApplicationDisplayName("文件列表生成器專業(yè)版")
    
    if hasattr(Qt, 'AA_EnableHighDpiScaling'):
        app.setAttribute(Qt.AA_EnableHighDpiScaling, True)
    if hasattr(Qt, 'AA_UseHighDpiPixmaps'):
        app.setAttribute(Qt.AA_UseHighDpiPixmaps, True)
    
    window = FileListGenerator()
    window.show()
    
    sys.exit(app.exec_())

if __name__ == '__main__':
    main()