如何在Python/Django中程序化生成自定义颜色的SVG与ICO头像？

不需要依赖第三方头像生成库，完全可以通过代码程序化生成 SVG 和 ICO 格式的头像，以下是具体实现步骤：

核心思路 #

1. 直接构造 SVG 字符串生成基础头像（SVG 结构简单，适合程序化拼接）
2. 基于 Tailwind CSS 配色预设背景色，通过亮度计算自动匹配黑白文字
3. 使用 Pillow 将 SVG 转换为多尺寸 ICO 文件

步骤1：定义配色与辅助工具函数 #

先预设一组 Tailwind CSS 纯色 palette，再实现颜色亮度计算工具，用于判断文字用黑色还是白色：

import re

def hex_to_rgb(hex_color):
    # 处理带#和不带#的十六进制颜色
    hex_color = re.sub(r'^#', '', hex_color)
    return tuple(int(hex_color[i:i+2], 16) for i in (0, 2, 4))

def get_text_color(bg_hex):
    # 根据背景色亮度计算文字颜色（亮度>127用黑色，否则白色）
    r, g, b = hex_to_rgb(bg_hex)
    brightness = (0.299 * r) + (0.587 * g) + (0.114 * b)
    return "#000000" if brightness > 127 else "#ffffff"

# 预定义Tailwind CSS适配的背景色列表
TAILWIND_COLORS = [
    "#3b82f6", "#ef4444", "#10b981", "#f59e0b", "#8b5cf6",
    "#ec4899", "#6366f1", "#14b8a6", "#f97316", "#d946ef",
    "#06b6d4", "#84cc16", "#f43f5e", "#0ea5e9", "#a855f7"
]

步骤2：生成 SVG 格式头像 #

直接拼接 SVG 字符串返回响应，无需额外库：

from django.http import HttpResponse

def generate_avatar_svg(request, initial):
    initial = initial.upper()
    # 校验输入，仅保留A-Z单个字母
    if not initial.isalpha() or len(initial) != 1:
        initial = "A"
    
    # 根据字母索引匹配固定背景色（A对应0，B对应1，循环取色）
    color_index = ord(initial) - ord('A')
    bg_color = TAILWIND_COLORS[color_index % len(TAILWIND_COLORS)]
    text_color = get_text_color(bg_color)
    
    # 构造SVG内容
    svg_content = f'''
    <svg xmlns="http://www.w3.org/2000/svg" width="200" height="200" viewBox="0 0 200 200">
        <rect width="200" height="200" fill="{bg_color}" />
        <text x="100" y="120" font-family="Arial, sans-serif" font-size="100" font-weight="bold" text-anchor="middle" fill="{text_color}">
            {initial}
        </text>
    </svg>
    '''.strip()
    
    response = HttpResponse(svg_content, content_type='image/svg+xml')
    response['Content-Disposition'] = f'inline; filename="{initial}_avatar.svg"'
    return response

步骤3：生成 ICO 格式头像 #

通过 Pillow 将 SVG 转换为多尺寸 ICO（ICO 需要包含16x16、32x32等多种尺寸）：

1. 先安装依赖：pip install pillow
2. 编写转换函数：

from PIL import Image
import io

def generate_avatar_ico(request, initial):
    initial = initial.upper()
    if not initial.isalpha() or len(initial) != 1:
        initial = "A"
    
    color_index = ord(initial) - ord('A')
    bg_color = TAILWIND_COLORS[color_index % len(TAILWIND_COLORS)]
    text_color = get_text_color(bg_color)
    
    # 生成基础SVG内容
    svg_content = f'''
    <svg xmlns="http://www.w3.org/2000/svg" width="200" height="200" viewBox="0 0 200 200">
        <rect width="200" height="200" fill="{bg_color}" />
        <text x="100" y="120" font-family="Arial, sans-serif" font-size="100" font-weight="bold" text-anchor="middle" fill="{text_color}">
            {initial}
        </text>
    </svg>
    '''.strip()
    
    # 将SVG加载为PIL图像
    svg_bytes = svg_content.encode('utf-8')
    img = Image.open(io.BytesIO(svg_bytes))
    
    # 生成ICO所需的多尺寸图像
    sizes = [(16,16), (32,32), (64,64), (128,128)]
    ico_images = []
    for size in sizes:
        resized_img = img.resize(size, Image.Resampling.LANCZOS)
        ico_images.append(resized_img)
    
    # 保存为ICO格式
    ico_bytes = io.BytesIO()
    ico_images[0].save(ico_bytes, format='ICO', sizes=sizes)
    ico_bytes.seek(0)
    
    response = HttpResponse(ico_bytes, content_type='image/x-icon')
    response['Content-Disposition'] = f'inline; filename="{initial}_avatar.ico"'
    return response

步骤4：配置 Django 路由 #

在项目的 urls.py 中添加头像接口路由：

from django.urls import path
from .views import generate_avatar_svg, generate_avatar_ico

urlpatterns = [
    path('avatar/svg/<str:initial>/', generate_avatar_svg, name='avatar_svg'),
    path('avatar/ico/<str:initial>/', generate_avatar_ico, name='avatar_ico'),
]

优化建议 #

• 缓存优化：用 Django 缓存框架缓存生成的头像内容，避免重复计算，缓存键可设为 avatar_svg_{initial}、avatar_ico_{initial}
• 字体适配：SVG 中可替换为系统支持的无衬线字体，或引入自定义字体确保显示一致
• 位置微调：可根据字体大小调整 SVG 文本的 y 坐标，让字母居中更精准

内容的提问来源于stack exchange，提问作者Kim Stacks