Pythonで簡単に環境変数を設定するスクリプトの作成方法

import os
import winreg
import ctypes

MB_YESNO = 4
MB_ICONQUESTION = 32
IDYES = 6

def show_message_box(text, title, flags):
    return ctypes.windll.user32.MessageBoxW(0, text, title, flags)

def add_to_system_path(new_path):
    with winreg.OpenKey(winreg.HKEY_CURRENT_USER, r"Environment", 0, winreg.KEY_READ | winreg.KEY_WRITE) as key:
        try:
            current_path = winreg.QueryValueEx(key, "Path")[0]
        except FileNotFoundError:
            current_path = ""

        if new_path not in current_path:
            updated_path = current_path + os.pathsep + new_path if current_path else new_path
            winreg.SetValueEx(key, "Path", 0, winreg.REG_EXPAND_SZ, updated_path)
            show_message_box(f"PATHに{new_path}を追加しました。", "成功", 0)
            return True
        else:
            show_message_box(f"{new_path}はすでにPATHに存在します。", "情報", 0)
            return False

def restart_prompt():
    response = show_message_box(
        "PCを再起動する必要があります。\n\n再起動しますか？", "再起動の確認", MB_YESNO | MB_ICONQUESTION
    )
    if response == IDYES:
        os.system("shutdown /r /t 0")  # 即時再起動
    else:
        show_message_box("再起動は後で行ってください。", "確認", 0)

# 使用例
new_path = r"C:\example\path"
if add_to_system_path(new_path):
    restart_prompt()

    with winreg.OpenKey(winreg.HKEY_LOCAL_MACHINE, r"SYSTEM\\CurrentControlSet\\Control\\Session Manager\\Environment", 0, winreg.KEY_READ | winreg.KEY_WRITE) as key: