在Windows中自动将彩色文本写入控制台

import ctypes

# 定义控制台颜色常量
FOREGROUND_BLUE = 0x0001
FOREGROUND_GREEN = 0x0002
FOREGROUND_RED = 0x0004
FOREGROUND_INTENSITY = 0x0008
BACKGROUND_BLUE = 0x0010
BACKGROUND_GREEN = 0x0020
BACKGROUND_RED = 0x0040
BACKGROUND_INTENSITY = 0x0080

# 获取标准输出的句柄
std_out_handle = ctypes.windll.kernel32.GetStdHandle(-11)

def set_console_color(color):
    ctypes.windll.kernel32.SetConsoleTextAttribute(std_out_handle, color)

def reset_console_color():
    set_console_color(FOREGROUND_RED | FOREGROUND_GREEN | FOREGROUND_BLUE)

def write_colored_text(text, color):
    set_console_color(color)
    print(text)
    reset_console_color()

# 示例用法
write_colored_text("这是红色文本", FOREGROUND_RED)
write_colored_text("这是绿色文本", FOREGROUND_GREEN)
write_colored_text("这是蓝色文本", FOREGROUND_BLUE)