67 lines
1.9 KiB
Python
67 lines
1.9 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""
|
||
编码辅助工具
|
||
提供UTF-8编码相关的辅助函数
|
||
"""
|
||
|
||
import sys
|
||
import io
|
||
import os
|
||
|
||
|
||
def setup_utf8_output():
|
||
"""设置标准输出为UTF-8编码（Windows系统）"""
|
||
if sys.platform == 'win32':
|
||
try:
|
||
# 设置标准输出编码
|
||
if hasattr(sys.stdout, 'buffer'):
|
||
sys.stdout = io.TextIOWrapper(
|
||
sys.stdout.buffer,
|
||
encoding='utf-8',
|
||
errors='replace',
|
||
line_buffering=True
|
||
)
|
||
if hasattr(sys.stderr, 'buffer'):
|
||
sys.stderr = io.TextIOWrapper(
|
||
sys.stderr.buffer,
|
||
encoding='utf-8',
|
||
errors='replace',
|
||
line_buffering=True
|
||
)
|
||
# 设置控制台代码页为UTF-8
|
||
os.system('chcp 65001 >nul 2>&1')
|
||
except Exception:
|
||
pass
|
||
|
||
|
||
def safe_print(*args, **kwargs):
|
||
"""安全的UTF-8打印函数"""
|
||
try:
|
||
print(*args, **kwargs)
|
||
except UnicodeEncodeError:
|
||
# 如果输出失败，尝试使用ASCII安全版本
|
||
safe_args = []
|
||
for arg in args:
|
||
if isinstance(arg, str):
|
||
try:
|
||
safe_args.append(arg.encode('ascii', 'replace').decode('ascii'))
|
||
except:
|
||
safe_args.append(repr(arg))
|
||
else:
|
||
safe_args.append(arg)
|
||
print(*safe_args, **kwargs)
|
||
|
||
|
||
def read_file_utf8(file_path: str) -> str:
|
||
"""读取UTF-8编码的文件"""
|
||
with open(file_path, 'r', encoding='utf-8') as f:
|
||
return f.read()
|
||
|
||
|
||
def write_file_utf8(file_path: str, content: str):
|
||
"""写入UTF-8编码的文件"""
|
||
os.makedirs(os.path.dirname(file_path) if os.path.dirname(file_path) else '.', exist_ok=True)
|
||
with open(file_path, 'w', encoding='utf-8', newline='\n') as f:
|
||
f.write(content)
|
||
|