set_default_app(".txt", "C:\\Windows\\notepad.exe")
if check_default_app(".txt", "C:\\Windows\\notepad.exe"):
    print("設(shè)置成功")

import winreg
import subprocess
import os
import time
import hashlib

def set_default_app_cmd(file_extension, app_path):
    """
    使用 Windows 命令行工具設(shè)置默認(rèn)打開程序，兼容 Windows 10/11
    """
    try:
        ext = file_extension if file_extension.startswith('.') else '.' + file_extension
        prog_id = ext[1:].upper() + "File"
        subprocess.run(f'assoc {ext}={prog_id}', shell=True, check=True)
        subprocess.run(f'ftype {prog_id}="{app_path}" "%1"', shell=True, check=True)
        return True, ""
    except Exception as e:
        return False, str(e)

def set_default_app(file_extension, app_path, icon_path=None):
    """
    設(shè)置文件擴(kuò)展名的默認(rèn)打開程序
    參數(shù):
        file_extension: 文件擴(kuò)展名（如 ".txt" 或 "txt"）
        app_path: 應(yīng)用程序完整路徑
        icon_path: 可選圖標(biāo)路徑
    返回: 無
    """
    # 先用命令行設(shè)置
    ok, msg = set_default_app_cmd(file_extension, app_path)
    if not ok:
        print(f"命令行設(shè)置失敗: {msg}")
    
    try:
        # 確保擴(kuò)展名格式正確
        ext = file_extension if file_extension.startswith('.') else '.' + file_extension
        
        # 創(chuàng)建或打開擴(kuò)展名對應(yīng)的注冊表鍵
        with winreg.CreateKey(winreg.HKEY_CLASSES_ROOT, ext) as key:
            prog_id = winreg.QueryValue(key, None)
            if not prog_id:
                prog_id = ext[1:].upper() + "File"
                winreg.SetValue(key, None, winreg.REG_SZ, prog_id)
            
            # 設(shè)置打開命令
            with winreg.CreateKey(winreg.HKEY_CLASSES_ROOT, f"{prog_id}\\shell\\open\\command") as cmd_key:
                winreg.SetValue(cmd_key, None, winreg.REG_SZ, f"\"{app_path}\" \"%1\"")
        
        # 處理用戶選擇（Windows 10/11 需要）
        user_choice_path = f"Software\\Microsoft\\Windows\\CurrentVersion\\Explorer\\FileExts\\{ext}\\"
        try:
            with winreg.OpenKey(winreg.HKEY_CURRENT_USER, user_choice_path, 0, winreg.KEY_ALL_ACCESS) as key:
                winreg.DeleteKey(key, "UserChoice")
        except WindowsError:
            pass
        
        # 生成用戶選擇哈希（Windows 10/11 驗(yàn)證機(jī)制）
        try:
            user_sid = os.getlogin()
        except Exception:
            user_sid = "unknown"
        
        timestamp = int(time.time())
        hash_input = f"{prog_id}{user_sid}{timestamp}".encode('utf-16le')
        hash_value = hashlib.sha256(hash_input).hexdigest()
        
        with winreg.CreateKey(winreg.HKEY_CURRENT_USER, user_choice_path + "UserChoice") as key:
            winreg.SetValueEx(key, "ProgId", 0, winreg.REG_SZ, prog_id)
            winreg.SetValueEx(key, "Hash", 0, winreg.REG_SZ, hash_value)
        
        print(f"成功設(shè)置 {ext} 的默認(rèn)打開程序?yàn)? {app_path}")
        
    except Exception as e:
        print(f"設(shè)置默認(rèn)程序失敗: {e}")

def check_default_app(file_extension, app_path):
    """
    檢查當(dāng)前擴(kuò)展名的默認(rèn)打開程序是否為指定程序
    參數(shù):
        file_extension: 文件擴(kuò)展名
        app_path: 應(yīng)用程序路徑
    返回: True/False
    """
    ext = file_extension if file_extension.startswith('.') else '.' + file_extension
    try:
        output = subprocess.check_output(f"assoc {ext}", shell=True, encoding="gbk", errors="ignore")
        if "=" not in output:
            return False
        prog_id = output.strip().split("=")[-1]
        output2 = subprocess.check_output(f"ftype {prog_id}", shell=True, encoding="gbk", errors="ignore")
        if app_path.lower() in output2.lower():
            return True
    except Exception:
        pass
    return False

# 使用示例
if __name__ == "__main__":
    # 示例1：設(shè)置.txt文件用記事本打開
    set_default_app(".txt", "C:\\Windows\\notepad.exe")
    
    # 示例2：設(shè)置.jpg文件用照片查看器打開
    set_default_app("jpg", "C:\\Windows\\System32\\rundll32.exe", "C:\\Windows\\System32\\photo_viewer.dll")
    
    # 檢查設(shè)置是否成功
    if check_default_app(".txt", "C:\\Windows\\notepad.exe"):
        print("設(shè)置成功！")
    else:
        print("設(shè)置失?。?)