111 lines
3.4 KiB
Python
111 lines
3.4 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
"""
|
||
Windows 控制台编码修复模块
|
||
用于解决 Windows 下 Unicode 字符显示问题
|
||
"""
|
||
|
||
import sys
|
||
import os
|
||
|
||
def setup_console_encoding():
|
||
"""设置控制台编码为 UTF-8"""
|
||
if sys.platform.startswith('win'):
|
||
try:
|
||
# 方法1: 设置环境变量
|
||
os.environ['PYTHONIOENCODING'] = 'utf-8'
|
||
|
||
# 方法2: 重新配置标准流
|
||
import codecs
|
||
import io
|
||
|
||
# 检查是否已经是正确的编码
|
||
if hasattr(sys.stdout, 'encoding') and sys.stdout.encoding.lower() not in ['utf-8', 'utf8']:
|
||
# 只有在需要时才重新配置
|
||
if hasattr(sys.stdout, 'detach'):
|
||
sys.stdout = io.TextIOWrapper(
|
||
sys.stdout.detach(),
|
||
encoding='utf-8',
|
||
errors='replace'
|
||
)
|
||
sys.stderr = io.TextIOWrapper(
|
||
sys.stderr.detach(),
|
||
encoding='utf-8',
|
||
errors='replace'
|
||
)
|
||
if hasattr(sys.stdin, 'detach'):
|
||
sys.stdin = io.TextIOWrapper(
|
||
sys.stdin.detach(),
|
||
encoding='utf-8',
|
||
errors='replace'
|
||
)
|
||
|
||
return True
|
||
|
||
except Exception as e:
|
||
print(f"Warning: Could not set UTF-8 encoding: {e}")
|
||
return False
|
||
return True
|
||
|
||
def safe_print(text, fallback_text=None):
|
||
"""安全打印，如果遇到编码错误则使用备用文本"""
|
||
try:
|
||
print(text)
|
||
except UnicodeEncodeError:
|
||
if fallback_text:
|
||
try:
|
||
print(fallback_text)
|
||
except UnicodeEncodeError:
|
||
print("Message with special characters (encoding issue)")
|
||
else:
|
||
# 移除非ASCII字符的简单版本
|
||
ascii_text = text.encode('ascii', errors='ignore').decode('ascii')
|
||
print(ascii_text if ascii_text.strip() else "Message (encoding issue)")
|
||
|
||
def get_safe_emoji_text(emoji_text, fallback_text):
|
||
"""获取安全的emoji文本，如果不支持则返回备用文本"""
|
||
try:
|
||
# 测试是否能编码
|
||
emoji_text.encode(sys.stdout.encoding or 'utf-8')
|
||
return emoji_text
|
||
except (UnicodeEncodeError, LookupError):
|
||
return fallback_text
|
||
|
||
# 预定义的安全文本映射
|
||
EMOJI_FALLBACKS = {
|
||
"🚀": "[START]",
|
||
"📍": "[ADDR]",
|
||
"⏰": "[TIME]",
|
||
"✅": "[OK]",
|
||
"🔗": "[CONN]",
|
||
"📨": "[MSG]",
|
||
"❌": "[ERROR]",
|
||
"🔌": "[DISC]",
|
||
"📋": "[LIST]",
|
||
"🖥️": "[SERVER]",
|
||
"💻": "[CLIENT]",
|
||
"📁": "[FILE]",
|
||
"🎯": "[RESULT]",
|
||
"⚠️": "[WARN]",
|
||
"📊": "[REPORT]",
|
||
"🧪": "[TEST]",
|
||
"🎮": "[INTERACTIVE]",
|
||
"📤": "[SEND]",
|
||
"📥": "[RECV]",
|
||
"👉": "[INPUT]",
|
||
"👋": "[EXIT]",
|
||
"🎉": "[SUCCESS]",
|
||
"🔧": "[FIX]",
|
||
"⛔": "[STOP]",
|
||
"💡": "[TIP]"
|
||
}
|
||
|
||
def safe_format_message(message):
|
||
"""将消息中的emoji替换为安全的文本"""
|
||
for emoji, fallback in EMOJI_FALLBACKS.items():
|
||
message = message.replace(emoji, fallback)
|
||
return message
|
||
|
||
# 自动设置编码
|
||
if __name__ != "__main__":
|
||
setup_console_encoding() |