使用Python实现一键批量生成员工名片

pip install pandas reportlab matplotlib

# 导入pandas库，用于数据处理和Excel文件读取
import pandas as pd
# 从reportlab库中导入A4页面尺寸
from reportlab.lib.pagesizes import A4
# 从reportlab库中导入canvas模块，用于创建PDF文档
from reportlab.pdfgen import canvas
# 从reportlab库中导入pdfmetrics模块，用于字体度量
from reportlab.pdfbase import pdfmetrics
# 从reportlab库中导入TTFont类，用于处理TrueType字体
from reportlab.pdfbase.ttfonts import TTFont
# 从reportlab库中导入HexColor类，用于使用十六进制颜色代码
from reportlab.lib.colors import HexColor
# 导入os模块，用于文件系统操作
import os
# 导入platform模块，用于获取系统信息
import platform
# 导入matplotlib的字体管理器，用于查找系统字体
import matplotlib.font_manager as fm

# 定义函数，获取系统中的中文字体
def get_chinese_font():
    # 获取当前操作系统名称
    system = platform.system()
    # 初始化字体列表
    fonts = []
    
    # 定义常见中文字体路径列表
    common_chinese_fonts = [
        # Windows 常用中文字体
        'C:/Windows/Fonts/simhei.ttf',  # 黑体
        'C:/Windows/Fonts/simsun.ttc',  # 宋体
        'C:/Windows/Fonts/simkai.ttf',  # 楷体
        'C:/Windows/Fonts/msyh.ttf',    # 微软雅黑
        
        # macOS 常用中文字体
        '/System/Library/Fonts/PingFang.ttc',  # 苹方
        '/Library/Fonts/Arial Unicode.ttf',
        
        # Linux 常用中文字体
        '/usr/share/fonts/truetype/arphic/uming.ttc',
        '/usr/share/fonts/truetype/droid/DroidSansFallbackFull.ttf',
    ]
    
    # 遍历常见中文字体路径列表
    for font_path in common_chinese_fonts:
        # 检查字体文件是否存在
        if os.path.exists(font_path):
            # 如果存在，返回字体路径
            return font_path
    
    # 如果上面的字体都不存在，尝试从matplotlib获取系统字体
    try:
        # 获取系统中所有存在的字体路径
        fonts = [f for f in fm.findSystemFonts() if os.path.exists(f)]
        # 遍历所有字体
        for font in fonts:
            try:
                # 检查字体名称中是否包含非ASCII字符（即中文字符）
                if any([True for c in fm.FontProperties(fname=font).get_name() if ord(c) > 127]):
                    # 如果找到中文字体，返回字体路径
                    return font
            except:
                # 如果处理出错，继续检查下一个字体
                continue
    except:
        # 如果获取系统字体出错，忽略错误
        pass
    
    # 如果找不到中文字体，返回None
    return None

# 定义函数，读取Excel数据
def read_excel(file_path):
    try:
        # 使用pandas读取Excel文件
        df = pd.read_excel(file_path)
        # 返回数据框
        return df
    except Exception as e:
        # 如果读取出错，打印错误信息
        print(f"读取Excel文件时出错: {e}")
        # 返回None
        return None

# 定义函数，创建PDF名片
def create_business_cards(df, output_file):
    # 获取中文字体路径
    chinese_font_path = get_chinese_font()
    
    # 检查是否找到中文字体
    if chinese_font_path:
        # 设置字体名称
        font_name = 'ChineseFont'
        # 注册中文字体
        pdfmetrics.registerFont(TTFont(font_name, chinese_font_path))
        # 打印使用的字体路径
        print(f"使用中文字体: {chinese_font_path}")
    else:
        # 如果找不到中文字体，使用默认字体
        font_name = 'Helvetica'  # 默认字体
        # 打印警告信息
        print("警告: 找不到中文字体，中文可能显示为乱码")
    
    # 创建PDF文档，使用A4页面大小
    c = canvas.Canvas(output_file, pagesize=A4)
    # 获取A4页面的宽度和高度
    width, height = A4
    
    # 设置名片尺寸和布局参数
    card_width = 250  # 名片宽度
    card_height = 150  # 名片高度
    margin = 20  # 边距
    cards_per_row = 2  # 每行名片数量
    cards_per_column = 4  # 每列名片数量
    
    # 遍历数据框中的每一行
    for i, record in df.iterrows():
        # 计算当前名片在页面中的位置
        row_num = i % (cards_per_row * cards_per_column)
        col = row_num % cards_per_row  # 计算列位置
        row = row_num // cards_per_row  # 计算行位置
        
        # 计算名片在页面上的x坐标
        x = margin + col * (card_width + margin)
        # 计算名片在页面上的y坐标
        y = height - margin - (row + 1) * card_height - row * margin
        
        # 检查是否需要添加新页面
        if row_num == 0 and i > 0:
            # 添加新页面
            c.showPage()
        
        # 绘制名片背景
        c.setFillColor(HexColor('#f8f8f8'))  # 设置填充颜色
        c.roundRect(x, y, card_width, card_height, 5, fill=1, stroke=0)  # 绘制圆角矩形
        
        # 绘制名片边框
        c.setStrokeColor(HexColor('#cccccc'))  # 设置边框颜色
        c.setLineWidth(0.5)  # 设置边框宽度
        c.roundRect(x, y, card_width, card_height, 5, fill=0, stroke=1)  # 绘制圆角矩形边框
        
        # 绘制公司名称
        c.setFillColor(HexColor('#2c3e50'))  # 设置文字颜色
        c.setFont(font_name, 14)  # 设置字体和大小
        company = str(record['公司'])  # 获取公司名称
        c.drawString(x + 15, y + card_height - 25, company)  # 在名片上绘制公司名称
        
        # 绘制分隔线
        c.setStrokeColor(HexColor('#3498db'))  # 设置线条颜色
        c.setLineWidth(1)  # 设置线条宽度
        c.line(x + 15, y + card_height - 35, x + card_width - 15, y + card_height - 35)  # 绘制分隔线
        
        # 绘制姓名（较大字体）
        c.setFillColor(HexColor('#2c3e50'))  # 设置文字颜色
        c.setFont(font_name, 12)  # 设置字体和大小
        name = str(record['姓名'])  # 获取姓名
        c.drawString(x + 15, y + card_height - 60, f"姓名：{name}")  # 在名片上绘制姓名
        
        # 绘制其他信息
        c.setFont(font_name, 10)  # 设置字体和大小
        c.setFillColor(HexColor('#555555'))  # 设置文字颜色
        
        department = str(record['部门'])  # 获取部门
        c.drawString(x + 15, y + card_height - 80, f"部门：{department}")  # 在名片上绘制部门
        
        position = str(record['职位'])  # 获取职位
        c.drawString(x + 15, y + card_height - 100, f"职位：{position}")  # 在名片上绘制职位
        
        phone = str(record['电话'])  # 获取电话
        c.drawString(x + 15, y + card_height - 120, f"电话：{phone}")  # 在名片上绘制电话
    
    # 保存PDF文件
    c.save()
    # 打印成功信息
    print(f"名片已生成到文件: {output_file}")

# 主程序入口
if __name__ == "__main__":
    # 指定Excel文件路径
    excel_file = "employee_data.xlsx"  # 请根据实际情况修改
    
    # 指定输出PDF文件路径
    pdf_file = "员工名片1.pdf"
    
    # 读取Excel数据
    data = read_excel(excel_file)
    
    # 检查是否成功读取数据
    if data is not None:
        # 创建名片
        create_business_cards(data, pdf_file)