# 必需庫(kù)安裝
pip install pyqt5 pywin32

def emoji_icon(self, emoji):
    """使用QPainter繪制Emoji圖標(biāo)"""
    pixmap = QPixmap(32, 32)
    pixmap.fill(Qt.transparent)
    
    painter = QPainter(pixmap)
    font = painter.font()
    font.setPointSize(20)  # 調(diào)整字號(hào)控制Emoji大小
    painter.setFont(font)
    painter.drawText(pixmap.rect(), Qt.AlignCenter, emoji)
    painter.end()
    
    return QIcon(pixmap)

def nuclear_cleanup(self, server_ip):
    """組合使用多種方式確保徹底清理"""
    # 1. 標(biāo)準(zhǔn)net命令清理
    self.run_cmd("net use * /delete /y")
    
    # 2. Windows憑據(jù)管理器清理
    self.run_cmd(f"cmdkey /delete:\\\\{server_ip}")
    
    # 3. Win32 API強(qiáng)制斷開(kāi)
    windll.mpr.WNetCancelConnection2W(
        create_unicode_buffer(f"\\\\{server_ip}"), 
        0, 
        True
    )
    
    # 4. 重啟網(wǎng)絡(luò)服務(wù)
    self.run_cmd("net stop workstation /y")
    time.sleep(2)
    self.run_cmd("net start workstation")

def map_drive(self):
    # 參數(shù)驗(yàn)證
    if not all([server_ip, share, drive, user, pwd]):
        QMessageBox.warning(self, "警告", "請(qǐng)?zhí)顚?xiě)所有必填字段")
        return
    
    # 構(gòu)造UNC路徑
    path = f"\\\\{server_ip}\\{share}"
    
    # 執(zhí)行映射命令
    result = self.run_cmd(
        f"net use {drive} {path} {pwd} /user:{user} "
        f"{'/persistent:yes' if self.persistent_check.isChecked() else ''}"
    )
    
    # 結(jié)果驗(yàn)證
    if result and "成功" in result:
        self._verify_drive_access(drive)

def run_cmd(self, command):
    try:
        result = subprocess.run(
            command,
            shell=True,
            check=True,
            stdout=subprocess.PIPE,
            stderr=subprocess.PIPE,
            encoding='gbk',  # 中文系統(tǒng)編碼處理
            text=True
        )
        return result.stdout.strip()
    except subprocess.CalledProcessError as e:
        self._handle_command_error(e)

# 使用QHBoxLayout和QVBoxLayout嵌套實(shí)現(xiàn)復(fù)雜布局
connection_layout = QVBoxLayout()
ip_layout = QHBoxLayout()
ip_layout.addWidget(QLabel("??? 服務(wù)器IP:"))
ip_layout.addWidget(self.ip_input)
connection_layout.addLayout(ip_layout)

# 添加彈性空間使按鈕居中
button_layout.addItem(QSpacerItem(
    20, 20, 
    QSizePolicy.Expanding, 
    QSizePolicy.Minimum
))

/* 按鈕懸停效果 */
QPushButton {
    background-color: #4CAF50;
    border-radius: 5px;
    padding: 8px;
}
QPushButton:hover {
    background-color: #45a049;
    box-shadow: 0 2px 5px rgba(0,0,0,0.2);
}

???????/* 狀態(tài)欄樣式 */
QLabel#status_bar {
    background-color: #f5f5f5;
    border-radius: 5px;
    padding: 8px;
    color: #666;
}

import subprocess
import sys
import os
import time
from ctypes import windll, create_unicode_buffer
import win32wnet
import win32netcon
from PyQt5.QtWidgets import (QApplication, QMainWindow, QWidget, QVBoxLayout, QHBoxLayout, 
                             QLabel, QLineEdit, QPushButton, QGroupBox, QCheckBox, 
                             QMessageBox, QComboBox, QSpacerItem, QSizePolicy)
from PyQt5.QtCore import Qt, QSize
from PyQt5.QtGui import QIcon, QFont, QPixmap, QPainter

class DriveMapperApp(QMainWindow):
    def __init__(self):
        super().__init__()
        self.setWindowTitle("網(wǎng)絡(luò)驅(qū)動(dòng)器映射工具")
        self.setWindowIcon(self.emoji_icon("??"))
        self.setFixedSize(500, 500)  # 稍微增大窗口尺寸
        
        # 主窗口部件
        self.main_widget = QWidget()
        self.setCentralWidget(self.main_widget)
        
        # 主布局
        self.main_layout = QVBoxLayout()
        self.main_layout.setContentsMargins(20, 20, 20, 20)  # 設(shè)置邊距
        self.main_layout.setSpacing(15)  # 設(shè)置控件間距
        self.main_widget.setLayout(self.main_layout)
        
        # 初始化UI
        self.init_ui()
        
    def emoji_icon(self, emoji):
        """創(chuàng)建emoji圖標(biāo)"""
        pixmap = QPixmap(32, 32)
        pixmap.fill(Qt.transparent)
        
        painter = QPainter(pixmap)
        font = painter.font()
        font.setPointSize(20)
        painter.setFont(font)
        painter.drawText(pixmap.rect(), Qt.AlignCenter, emoji)
        painter.end()
        
        return QIcon(pixmap)
    
    def init_ui(self):
        """初始化用戶(hù)界面"""
        # 標(biāo)題
        title = QLabel("網(wǎng)絡(luò)驅(qū)動(dòng)器映射工具")
        title.setFont(QFont("Microsoft YaHei", 16, QFont.Bold))
        title.setAlignment(Qt.AlignCenter)
        title.setStyleSheet("margin-bottom: 15px;")
        self.main_layout.addWidget(title)
        
        # 連接設(shè)置組
        connection_group = QGroupBox("? 連接設(shè)置")
        connection_group.setFont(QFont("Microsoft YaHei", 10))
        connection_layout = QVBoxLayout()
        connection_layout.setSpacing(12)  # 組內(nèi)控件間距
        connection_layout.setContentsMargins(15, 15, 15, 15)  # 組內(nèi)邊距
        
        # 服務(wù)器IP
        ip_layout = QHBoxLayout()
        ip_label = QLabel("??? 服務(wù)器IP:")
        ip_label.setFixedWidth(100)  # 固定標(biāo)簽寬度
        ip_layout.addWidget(ip_label)
        self.ip_input = QLineEdit("")
        self.ip_input.setPlaceholderText("例如: 192.168.1.100")
        self.ip_input.setStyleSheet("padding: 5px;")
        ip_layout.addWidget(self.ip_input)
        connection_layout.addLayout(ip_layout)
        
        # 共享文件夾
        share_layout = QHBoxLayout()
        share_label = QLabel("?? 共享文件夾:")
        share_label.setFixedWidth(100)
        share_layout.addWidget(share_label)
        self.share_input = QLineEdit("")
        self.share_input.setPlaceholderText("例如: SharedFolder")
        self.share_input.setStyleSheet("padding: 5px;")
        share_layout.addWidget(self.share_input)
        connection_layout.addLayout(share_layout)
        
        # 驅(qū)動(dòng)器盤(pán)符
        drive_layout = QHBoxLayout()
        drive_label = QLabel("?? 驅(qū)動(dòng)器盤(pán)符:")
        drive_label.setFixedWidth(100)
        drive_layout.addWidget(drive_label)
        self.drive_combo = QComboBox()
        self.drive_combo.addItems([f"{chr(i)}:" for i in range(90, 64, -1)])
        self.drive_combo.setCurrentText("")
        self.drive_combo.setStyleSheet("padding: 5px;")
        drive_layout.addWidget(self.drive_combo)
        connection_layout.addLayout(drive_layout)
        
        # 賬戶(hù)信息
        user_layout = QHBoxLayout()
        user_label = QLabel("?? 用戶(hù)名:")
        user_label.setFixedWidth(100)
        user_layout.addWidget(user_label)
        self.user_input = QLineEdit("")
        self.user_input.setPlaceholderText("例如: administrator")
        self.user_input.setStyleSheet("padding: 5px;")
        user_layout.addWidget(self.user_input)
        connection_layout.addLayout(user_layout)
        
        pwd_layout = QHBoxLayout()
        pwd_label = QLabel("?? 密碼:")
        pwd_label.setFixedWidth(100)
        pwd_layout.addWidget(pwd_label)
        self.pwd_input = QLineEdit("")
        self.pwd_input.setPlaceholderText("輸入密碼")
        self.pwd_input.setEchoMode(QLineEdit.Password)
        self.pwd_input.setStyleSheet("padding: 5px;")
        pwd_layout.addWidget(self.pwd_input)
        connection_layout.addLayout(pwd_layout)
        
        # 持久化選項(xiàng)
        self.persistent_check = QCheckBox("保持持久連接 (重啟后自動(dòng)重新連接)")
        self.persistent_check.setChecked(True)
        self.persistent_check.setStyleSheet("margin-top: 10px;")
        connection_layout.addWidget(self.persistent_check)
        
        connection_group.setLayout(connection_layout)
        self.main_layout.addWidget(connection_group)
        
        # 按鈕區(qū)域
        button_layout = QHBoxLayout()
        button_layout.setSpacing(20)  # 按鈕間距
        
        # 添加彈性空間使按鈕居中
        button_layout.addItem(QSpacerItem(20, 20, QSizePolicy.Expanding, QSizePolicy.Minimum))
        
        # 映射按鈕
        self.map_button = QPushButton(" 映射驅(qū)動(dòng)器")
        self.map_button.setIcon(self.emoji_icon("???"))
        self.map_button.setFixedSize(150, 40)  # 固定按鈕大小
        self.map_button.setStyleSheet("""
            QPushButton {
                background-color: #4CAF50;
                color: white;
                border-radius: 5px;
                padding: 8px;
                font-weight: bold;
            }
            QPushButton:hover {
                background-color: #45a049;
            }
        """)
        self.map_button.clicked.connect(self.map_drive)
        button_layout.addWidget(self.map_button)
        
        # 清理按鈕
        self.clean_button = QPushButton(" 清理連接")
        self.clean_button.setIcon(self.emoji_icon("??"))
        self.clean_button.setFixedSize(150, 40)
        self.clean_button.setStyleSheet("""
            QPushButton {
                background-color: #f44336;
                color: white;
                border-radius: 5px;
                padding: 8px;
                font-weight: bold;
            }
            QPushButton:hover {
                background-color: #d32f2f;
            }
        """)
        self.clean_button.clicked.connect(self.clean_connections)
        button_layout.addWidget(self.clean_button)
        
        button_layout.addItem(QSpacerItem(20, 20, QSizePolicy.Expanding, QSizePolicy.Minimum))
        
        self.main_layout.addLayout(button_layout)
        
        # 狀態(tài)欄
        self.status_bar = QLabel("?? 就緒")
        self.status_bar.setAlignment(Qt.AlignCenter)
        self.status_bar.setStyleSheet("""
            color: #666;
            margin-top: 10px;
            padding: 8px;
            background-color: #f5f5f5;
            border-radius: 5px;
        """)
        self.main_layout.addWidget(self.status_bar)
        
    def run_cmd(self, command):
        """執(zhí)行命令并返回輸出"""
        try:
            result = subprocess.run(
                command,
                shell=True,
                check=True,
                stdout=subprocess.PIPE,
                stderr=subprocess.PIPE,
                encoding='gbk',
                text=True
            )
            return result.stdout.strip()
        except subprocess.CalledProcessError as e:
            self.status_bar.setText(f"?? 錯(cuò)誤: {e.stderr}")
            return None
    
    def nuclear_cleanup(self, server_ip):
        """徹底清除所有可能的殘留連接"""
        self.status_bar.setText("?? 正在深度清理...")
        QApplication.processEvents()
        
        self.run_cmd("net use * /delete /y")
        self.run_cmd(f"net use \\\\{server_ip} /delete /y")
        
        creds = self.run_cmd("cmdkey /list")
        if creds and server_ip in creds:
            self.run_cmd(f"cmdkey /delete:\\\\{server_ip}")
            self.run_cmd(f"cmdkey /delete:WindowsLive:target=\\\\{server_ip}")
        
        try:
            windll.mpr.WNetCancelConnection2W(create_unicode_buffer(f"\\\\{server_ip}"), 0, True)
            win32wnet.WNetCancelConnection2(f"\\\\{server_ip}", 0, True)
        except Exception as e:
            self.status_bar.setText(f"?? API清理錯(cuò)誤: {e}")
        
        self.status_bar.setText("?? 正在重啟網(wǎng)絡(luò)服務(wù)...")
        QApplication.processEvents()
        self.run_cmd("net stop workstation /y")
        time.sleep(2)
        self.run_cmd("net start workstation")
        time.sleep(1)
        
        self.status_bar.setText("?? 清理完成")
    
    def clean_connections(self):
        """清理所有網(wǎng)絡(luò)連接"""
        server_ip = self.ip_input.text().strip()
        if not server_ip:
            QMessageBox.warning(self, "警告", "請(qǐng)輸入服務(wù)器IP地址")
            return
            
        reply = QMessageBox.question(
            self, '確認(rèn)',
            '確定要清理所有網(wǎng)絡(luò)連接嗎？這可能會(huì)斷開(kāi)現(xiàn)有的網(wǎng)絡(luò)驅(qū)動(dòng)器連接。',
            QMessageBox.Yes | QMessageBox.No, QMessageBox.No
        )
        
        if reply == QMessageBox.Yes:
            self.nuclear_cleanup(server_ip)
            QMessageBox.information(self, "完成", "網(wǎng)絡(luò)連接已清理完成")
    
    def map_drive(self):
        """映射網(wǎng)絡(luò)驅(qū)動(dòng)器"""
        server_ip = self.ip_input.text().strip()
        share = self.share_input.text().strip()
        drive = self.drive_combo.currentText()
        user = self.user_input.text().strip()
        pwd = self.pwd_input.text()
        
        if not all([server_ip, share, drive, user, pwd]):
            QMessageBox.warning(self, "警告", "請(qǐng)?zhí)顚?xiě)所有必填字段")
            return
            
        path = f"\\\\{server_ip}\\{share}"
        persistent = "/persistent:yes" if self.persistent_check.isChecked() else ""
        
        self.status_bar.setText("?? 正在準(zhǔn)備映射...")
        QApplication.processEvents()
        
        self.nuclear_cleanup(server_ip)
        
        self.status_bar.setText(f"?? 正在映射 {path} 到 {drive}...")
        QApplication.processEvents()
        
        result = self.run_cmd(f"net use {drive} {path} {pwd} /user:{user} {persistent}")
        
        if result:
            self.status_bar.setText(f"?? 成功映射 {path} 到 {drive}")
            QMessageBox.information(self, "成功", f"網(wǎng)絡(luò)驅(qū)動(dòng)器已成功映射到 {drive}")
            
            test_result = self.run_cmd(f"dir {drive}")
            if test_result:
                self.status_bar.setText(f"?? 訪問(wèn)測(cè)試成功: {drive} 驅(qū)動(dòng)器內(nèi)容可讀")
            else:
                self.status_bar.setText(f"?? 映射成功但訪問(wèn)測(cè)試失敗")
        else:
            self.status_bar.setText("?? 映射失敗")
            QMessageBox.critical(
                self, "錯(cuò)誤", 
                "驅(qū)動(dòng)器映射失?。n\n"
                "請(qǐng)嘗試以下解決方案：\n"
                "1. 手動(dòng)執(zhí)行清理操作\n"
                "2. 重啟計(jì)算機(jī)后重試\n"
                "3. 檢查服務(wù)器端的共享權(quán)限設(shè)置"
            )

if __name__ == "__main__":
    app = QApplication(sys.argv)
    app.setStyle('Fusion')  # 使用Fusion風(fēng)格使界面更現(xiàn)代
    window = DriveMapperApp()
    window.show()
    sys.exit(app.exec_())