Вставка кода Python в документы Microsoft Word

Расширенные примеры вставки кода Python в Word

Пример 1. Вставка многострочного кода с нумерацией строк и подсветкой через python-docx и pygments

Используем pygments для генерации HTML-разметки с подсветкой, затем конвертируем её в элементы Word через python-docx. Это даёт полноценную подсветку синтаксиса прямо в документе.

from docx import Document
from docx.shared import Pt, Inches
from pygments import highlight
from pygments.lexers import PythonLexer
from pygments.formatters import HtmlFormatter
import re

def add_code_block(doc, code, language='python'):
    lexer = PythonLexer()
    formatter = HtmlFormatter(nowrap=True, style='colorful')
    highlighted = highlight(code, lexer, formatter)
    # Преобразуем HTML в run'ы с соответствующим форматированием
    # Это упрощённый парсинг - для продакшена лучше использовать lxml
    p = doc.add_paragraph()
    parts = re.split(r'()', highlighted)
    for part in parts:
        if part.startswith(']+>', '', part)
            run = p.add_run(text)
            run.font.color.rgb = RGBColor.from_string(color)
            run.font.name = 'Consolas'
            run.font.size = Pt(9)
        else:
            # Обычный текст (может быть пробелами или переводом строки)
            if part:
                run = p.add_run(part)
                run.font.name = 'Consolas'
                run.font.size = Pt(9)
    return p

doc = Document()
code_example = '''def factorial(n):
    if n == 0:
        return 1
    else:
        return n * factorial(n-1)

print(factorial(5))'''
add_code_block(doc, code_example)
doc.save('highlighted_code.docx')
print('Документ с подсветкой синтаксиса создан.')

Документ содержит код с цветами ключевых слов, строк, чисел, как в IDE.

Пример 2. Вставка кода как изображения (для абсолютной точности отображения)

Иногда проще вставить скриншот кода, особенно если используются редкие шрифты. Python может сгенерировать изображение кода с помощью Pillow и вставить его в Word.

from docx import Document
from docx.shared import Inches
from PIL import Image, ImageDraw, ImageFont
import io

def code_to_image(code, font_path='C:/Windows/Fonts/Consolas.ttf', font_size=14):
    font = ImageFont.truetype(font_path, font_size)
    lines = code.split('\n')
    # Определяем размер изображения
    line_heights = [font.getbbox(line)[3] - font.getbbox(line)[1] for line in lines]
    max_height = max(line_heights)
    line_spacing = 4
    total_height = len(lines) * (max_height + line_spacing) + 20
    # Определяем максимальную ширину строки
    widths = [font.getbbox(line)[2] - font.getbbox(line)[0] for line in lines]
    max_width = max(widths) + 20
    img = Image.new('RGB', (max_width, total_height), 'white')
    draw = ImageDraw.Draw(img)
    y = 10
    for line in lines:
        draw.text((10, y), line, font=font, fill='black')
        y += max_height + line_spacing
    return img

doc = Document()
code = 'for i in range(10):\n    print(i)'
img = code_to_image(code)
# Сохраняем в байтовый поток
stream = io.BytesIO()
img.save(stream, format='PNG')
# Вставляем изображение в документ
doc.add_picture(stream, width=Inches(4))
doc.save('code_as_image.docx')
print('Готово.')

Документ содержит изображение кода, которое нельзя редактировать, но внешний вид гарантирован.

Пример 3. Вставка кода с использованием стилей Word (через python-docx)

Создайте собственный стиль «Код Python» в документе, задайте шрифт, размер, фон абзаца и применяйте его к абзацам с кодом.

from docx import Document
from docx.shared import Pt, RGBColor, Inches
from docx.enum.style import WD_STYLE_TYPE

doc = Document()
# Создаём стиль
style = doc.styles.add_style('PythonCode', WD_STYLE_TYPE.PARAGRAPH)
font = style.font
font.name = 'Consolas'
font.size = Pt(10)
font.color.rgb = RGBColor(0, 0, 0)
# Фон абзаца (заливка всего абзаца) - через XML
pf = style.paragraph_format
pf.space_before = Pt(6)
pf.space_after = Pt(6)
# Добавляем shading через элемент w:shd (не все версии python-docx поддерживают напрямую)

# Вставляем код с этим стилем
p = doc.add_paragraph('import sys\nprint(sys.version)', style='PythonCode')
doc.save('styled_code.docx')

Стиль применяется к абзацу, но фон может не отобразиться, если не прописать shading в XML.

Пример 4. Массовая вставка нескольких файлов кода Python в один документ Word

Обходите папку с .py файлами и вставляйте каждый файл как отдельный блок с заголовком-именем файла.

import os
from docx import Document

doc = Document()
source_dir = 'C:/myproject'
for filename in sorted(os.listdir(source_dir)):
    if filename.endswith('.py'):
        # Заголовок
        doc.add_heading(filename, level=2)
        with open(os.path.join(source_dir, filename), 'r', encoding='utf-8') as f:
            code = f.read()
        # Вставка кода (предыдущие методы)
        p = doc.add_paragraph()
        run = p.add_run(code)
        run.font.name = 'Consolas'
        run.font.size = Pt(10)
        doc.add_paragraph()  # разделитель
doc.save('all_codes.docx')
print('Создан документ с', len(source_dir), 'файлами.')

Документ содержит все файлы проекта, каждый с заголовком и моноширинным текстом.