Корректировка вывода исключений Python при неверной кодировке

Расширенные примеры работы с кодировкой traceback

Пример 1: Перехват исключения и вывод в ASCII с заменой

import sys
import traceback
import io

# Искусственно имитируем поток, который не поддерживает UTF-8
fake_stderr = io.TextIOWrapper(io.BytesIO(), encoding='ascii', errors='strict')
original_stderr = sys.stderr
sys.stderr = fake_stderr

try:
    raise ValueError("Значение: üñîçødë")
except:
    try:
        # Это вызовет UnicodeEncodeError
        traceback.print_exc()
    except UnicodeEncodeError as e:
        # Обработка: заменяем все не-ASCII символы на '?'
        exc_info = sys.exc_info()
        formatted = traceback.format_exception(*exc_info)
        safe = ''.join(formatted).encode('ascii', errors='replace').decode('ascii')
        original_stderr.write(safe)
    finally:
        sys.stderr = original_stderr

# Покажем, что записалось в fake_stderr
fake_stderr.seek(0)
print("Содержимое fake_stderr (байты):", fake_stderr.buffer.getvalue())

Traceback (most recent call last):
  File "", line 2, in 
ValueError: Значение: ???????
Содержимое fake_stderr (байты): b''

В этом примере показано, как перехватить ошибку кодировки при записи traceback в поток, который не поддерживает UTF-8. Вместо исключения программа выводит заменённый вариант.

Пример 2: Явное форматирование с указанием кодировки (Python 3.10+)

import traceback
import sys
import io

# Создаём исключение
try:
    raise TypeError("Недопустимый тип: float вместо int")
except TypeError:
    # Форматируем с кодировкой cp1251, чтобы вписаться в старый Windows-терминал
    tb_lines = traceback.format_exception(*sys.exc_info(), encoding='cp1251', errors='replace')
    result = ''.join(tb_lines)
    print("Результат в cp1251 (декодировано обратно для отображения):")
    print(result.encode('cp1251').decode('cp1251'))

Результат в cp1251 (декодировано обратно для отображения):
Traceback (most recent call last):
  File "", line 2, in 
TypeError: Недопустимый тип: float вместо int

Параметр encoding гарантирует, что символы, не входящие в целевую кодировку, будут заменены (errors='replace').

Пример 3: Сравнение вывода с PYTHONIOENCODING и без него

# script.py
# Создадим исключение с не-ASCII символами
raise SyntaxError("Неверный синтаксис: выражение '2 + 2 = 5'")

# Запуск без установки кодировки (в Windows может быть cp1251)
# python script.py 2>err.txt
# Потом посмотрим err.txt в UTF-8 – возможно, искажения.

# Запуск с PYTHONIOENCODING
# set PYTHONIOENCODING=utf-8 && python script.py 2>err_utf.txt
# Содержимое err_utf.txt будет в UTF-8.

# Пример содержимого err_utf.txt (открыт в UTF-8):
  File "script.py", line 2
    raise SyntaxError("Неверный синтаксис: выражение '2 + 2 = 5'")
SyntaxError: Неверный синтаксис: выражение '2 + 2 = 5'

Без установки кодировки сообщение могло бы выглядеть как ... Неверный синтаксис: ... с кракозябрами или вызвать UnicodeEncodeError при перенаправлении.

Пример 4: Использование locale для адаптации под систему

import sys
import locale

# Устанавливаем русскую локаль (работает в Linux)
try:
    locale.setlocale(locale.LC_ALL, 'ru_RU.UTF-8')
except locale.Error:
    # Если локаль не поддерживается, используем системную
    locale.setlocale(locale.LC_ALL, '')

# Получаем предпочтительную кодировку
enc = locale.getpreferredencoding()
sys.stderr.reconfigure(encoding=enc, errors='replace')

# Проверка
raise RuntimeError("Тест: €, ®, °")

Traceback (most recent call last):
  File "", line 1, in 
RuntimeError: Тест: , , °

На Linux этот код подстроит кодировку под системную (например, UTF-8), на Windows – под cp1251/cp866. Если в системе нет поддержки символа (например, €), он будет заменён.

Пример 5: Обработка ошибки в многопоточном приложении с записью в лог

import sys
import threading
import logging

# Настраиваем логгер с UTF-8
handler = logging.FileHandler('app.log', encoding='utf-8', mode='w')
formatter = logging.Formatter('%(asctime)s - %(threadName)s - %(message)s')
handler.setFormatter(formatter)
logger = logging.getLogger('multithread_app')
logger.addHandler(handler)
logger.setLevel(logging.ERROR)

def my_excepthook(args):
    # Обработчик для threading.excepthook
    logger.error('Exception in thread %s: %s', args.thread.name, ''.join(traceback.format_exception(*args.exc_info)))

threading.excepthook = my_excepthook

def worker():
    raise ValueError("Потоковый сбой: 100% загрузка CPU")

t = threading.Thread(target=worker, name='Worker-1')
t.start()
t.join()

# Читаем лог
with open('app.log', 'r', encoding='utf-8') as f:
    print(f.read())

2025-03-23 10:15:30,123 - Worker-1 - Exception in thread Worker-1: Traceback (most recent call last):
  File "", line 1, in worker
    raise ValueError("Потоковый сбой: 100% загрузка CPU")
ValueError: Потоковый сбой: 100% загрузка CPU

Здесь логгер обеспечивает корректное сохранение traceback в файл, независимо от кодировки терминала.