import os
from pathlib import Path
import platform

def get_desktop_path():
    system = platform.system()
    if system == 'Windows':
        # 通過注冊(cè)表獲取路徑（首選方案）
        try:
            import winreg
            with winreg.OpenKey(winreg.HKEY_CURRENT_USER, 
                               r'Software\Microsoft\Windows\CurrentVersion\Explorer\User Shell Folders') as key:
                return Path(winreg.QueryValueEx(key, 'Desktop')[0]).resolve()
        except:
            # 回退到環(huán)境變量方案
            return Path.home() / 'Desktop'
    elif system == 'Darwin':  # macOS
        return Path.home() / 'Desktop'
    else:  # Linux
        xdg_path = os.getenv('XDG_DESKTOP_DIR')
        return Path(xdg_path) if xdg_path else Path.home() / 'Desktop'

print("Desktop path:", get_desktop_path())

def safe_get_desktop():
    desktop = get_desktop_path()
    if not desktop.exists():
        raise FileNotFoundError(f"Desktop path {desktop} does not exist")
    return desktop

# 處理非英語系統(tǒng)環(huán)境
localized_names = {
    'Windows': {'zh-CN': '桌面', 'de-DE': 'Desktop'},
    'macOS': {'zh-CN': '桌面'}
}

def get_localized_desktop():
    system = platform.system()
    lang = os.getenv('LANG', 'en_US').split('_')[0]
    base = get_desktop_path()
    
    if system == 'Windows' and lang in localized_names['Windows']:
        return base / localized_names['Windows'][lang]
    return base

# 現(xiàn)代寫法
desktop = Path.home() / "Desktop"

# 傳統(tǒng)寫法
desktop = os.path.join(os.path.expanduser('~'), 'Desktop')

# 檢查寫權(quán)限
if not os.access(str(desktop), os.W_OK):
    print("Warning: No write permission to desktop")

from functools import lru_cache

@lru_cache(maxsize=1)
def get_desktop_cached():
    return get_desktop_path()

# 自動(dòng)在桌面創(chuàng)建日志文件
desktop = get_desktop_path()
log_path = desktop / "app_log.txt"
with log_path.open('a', encoding='utf-8') as f:
    f.write(f"App started at {datetime.now()}\n")

# 在桌面創(chuàng)建快捷方式（Windows）
import winshell
desktop = get_desktop_path()
winshell.create_shortcut(
    path=desktop / "MyApp.lnk",
    target="C:\\Program Files\\MyApp\\app.exe",
    description="My Application"
)

import unittest

class TestDesktopPath(unittest.TestCase):
    def test_windows_path(self):
        self.assertTrue(get_desktop_path().endswith('Desktop'))
    
    def test_macos_path(self):
        self.assertEqual(get_desktop_path(), Path.home() / 'Desktop')