python用openpyxl操作excel-单元格样式操作

python用openpyxl操作excel的单元格样式操作

python 复制代码
import logging
import com.pro001.log.pro_log_config as pro_log_config
from openpyxl import Workbook, load_workbook
from openpyxl.styles import  Font, Alignment, Side, PatternFill, Border
import os
import datetime
import random
import pandas as pd


def create_sn2(prefix, output_len=4):
    '''生成含有前缀带日期和尾长度格式的序列号,
    格式:prefix_YYYYMMDD_HHMMSS_0{output_len}d'''
    if output_len < 1:
        output_len = 1
    if not prefix:
        prefix = ''
    sn = 0
    while True:
        sn += 1
        if len(str(sn)) > output_len:
            sn = 0
        dt = datetime.datetime.now().strftime('%Y%m%d_%H%M%S')
        tmp = f'{prefix}{dt}'
        tmp = tmp + '{:0{}}'.format(sn, output_len)
        yield tmp


def create_sn(prefix, output_len=4):
    '''生成含有前缀并至少带指定长度格式的序列号,
    格式:prefix_d0{output_len}1'''
    if output_len < 1:
        output_len = 1
    sn = 0
    while True:
        sn += 1
        if len(str(sn)) > output_len:
            # 当大于指定位数时直接加上位数
            yield f'{prefix}' + str(sn)
        else:
            yield f'{prefix}' + '{:0{}}'.format(sn, output_len)



def sheet_cell_style_operate(base_file_path):
    """ 工作表单元格样式操作 """
    if not base_file_path:
        os.path.makedirs(base_file_path)
        logger.error(f'已创建文件路径:{base_file_path}')

    try:
        # 创建 Workbook 对象并指定第一个sheet为活动sheet
        wb = Workbook()
        ws = wb.worksheets[0]
        wb.active = ws
        sh_name = create_sn3('',4)
        # 指定sheet名称
        ws.title = next(sh_name)

        # 创建表头样式
        header_style_font = Font(name='Arial', size=10, bold=True, color='060606')
        body_style_font = Font(name='Arial', size=9, bold=False, color='006600')
        cell_alignment = Alignment(horizontal='center', vertical='center')

        # 添加列名,即表头
        header_list = ['报案号','姓名', '年龄', '邮箱','报案时间',
                       '受理时间','受理状态','受理人','是否结案','结案时间']
        # 序号从1开始,枚举 header_list列表,建立表头
        for i, header in enumerate(header_list, start=1):
            ws.cell(row=1, column=i, value=header)

        # 获取表头的各个单元格对象
        head_cell = ws[1]
        # 设置表头样式
        for head_cell in head_cell:
            head_cell.font = header_style_font
            head_cell.alignment = cell_alignment

        # 设置表头的行高
        ws.row_dimensions[1].height = 18
        # 动态设置各列的宽度
        max_cell_num = ws.max_column
        for i in range(1, max_cell_num+1):
            # 动态获取每列的坐标字母
            col_letter = ws.cell(row=1, column=i).column_letter
            # 根据列的字母坐标设定每列的宽度
            ws.column_dimensions[col_letter].width = 16

        # 添加表体数据
        claim_no = create_sn2('CN', 5)
        claim_status = ['未受理','已受理']
        case_status = ['理赔中', '已结案','已撤案']
        '''['报案号','姓名','年龄','邮箱','报案时间','受理时间','受理状态','受理人','是否结案','结案时间']'''
        for i in range(1, 31):
            new_max_num = ws.max_row + 1
            ws.cell(row=new_max_num, column=1, value=next(claim_no))
            ws.cell(row=new_max_num, column=2, value=f'姓名{random.randint(10, 200)}')
            ws.cell(row=new_max_num, column=3, value=random.randint(18, 100))
            ws.cell(row=new_max_num, column=4, value=f'email{random.randint(100, 300)}@xxx.com')
            dt = datetime.datetime.now()
            dt = dt - datetime.timedelta(days=random.randint(3, 180))
            dt_str = dt.strftime('%Y-%m-%d %H:%M:%S')
            ws.cell(row=new_max_num, column=5, value=f'{dt_str}')
            dt = dt + datetime.timedelta(days=random.randint(2, 7))
            dt_str = dt.strftime('%Y-%m-%d %H:%M:%S')
            ws.cell(row=new_max_num, column=6, value=f'{dt_str}')
            ws.cell(row=new_max_num, column=7, value=f'{random.sample(claim_status, 1)[0]}')
            ws.cell(row=new_max_num, column=8, value=f'受理人{random.randint(10, 40)}')
            ws.cell(row=new_max_num, column=9, value=f'{random.sample(case_status, 1)[0]}')
            dt = dt + datetime.timedelta(days=random.randint(7, 30))
            dt_str = dt.strftime('%Y-%m-%d %H:%M:%S')
            ws.cell(row=new_max_num, column=10, value=f'{dt_str}')

        # 遍历表体(表头除外)所有单元格并应用样式
        max_cell_num = ws.max_column
        max_row_num = ws.max_row
        next5 = 1
        next5_row_fill = PatternFill(start_color='09E1F2', end_color='09E1F2', fill_type='solid')
        calim_done_fill = PatternFill(start_color='278b06', end_color='278b06', fill_type='solid')
        calim_done_font = Font(name='Arial', size=8, bold=True, color='faea37')
        calim_ing_fill = PatternFill(start_color='FDE9D9', end_color='FDE9D9', fill_type='solid')
        calim_ing_side = Side(border_style='medium', color='e46c0a') # 定义颜色边框
        claim_ing_border = Border(top=calim_ing_side, right=calim_ing_side,
                                  bottom=calim_ing_side, left=calim_ing_side)
        blue_side = Side(border_style='medium', color='e46c0a')  # 定义蓝色边框
        top_border = Border(top=blue_side)
        for row in ws.iter_rows(min_row=2, max_row=max_row_num, min_col=1, max_col=max_cell_num):
            next5 = next5 + 1
            if next5 % 5 == 0:
                ws.row_dimensions[next5].height = 18
            for cell in row:
                cell.alignment = cell_alignment
                # 每隔5行设置单元格样式浅蓝色上边框深蓝色
                if next5 % 5 == 0:
                    cell.fill = next5_row_fill
                    cell.border = top_border
                val = cell.value
                # 特定单元格应用不一样的样式
                if val == '未受理':
                    cell.fill = calim_ing_fill
                    cell.border = claim_ing_border
                if val == '已结案':
                    cell.font = calim_done_font
                    cell.fill = calim_done_fill
                else:
                    cell.font = body_style_font

        # 保存wb对象
        suffix = '.xlsx'
        file_path_name = ''
        if not base_file_path.lower().endswith(suffix):
            file_path_name = 'ELX' + datetime.datetime.now().strftime('%Y%m%d_%H%M%S') + suffix
            file_path_name = os.path.join(base_file_path, file_path_name)
        else:
            file_path_name = base_file_path
        logger.info(f'参加文件:{file_path_name}')
        wb.save(file_path_name)
    except Exception as e:
        logger.error(f'文件{base_file_path}操作失败,info:\n{e}')
        return None


def main():
    """主函数"""
    sheet_cell_style_operate(r'F:\appData')


if __name__ == '__main__':
    print('-' * 60)
    main()

运行后效果如下:

相关推荐
岁月宁静2 小时前
多模态 Agent 技术全景解析 — 从模型能力、Agent 架构到工程化与商业落地
python·agent
试着2 小时前
【VSCode+AI+测试】连接ai大模型
ide·人工智能·vscode·python·学习·编辑器·ai-test
零小陈上(shouhou6668889)3 小时前
YOLOv8+PyQt5海洋船只检测(可以重新训练,yolov8模型,从图像、视频和摄像头三种路径识别检测)
开发语言·python·yolo
znhy_233 小时前
day36打卡
python
gf13211113 小时前
python_字幕文本、音频、视频一键组合
python·音视频·swift
我的xiaodoujiao3 小时前
使用 Python 语言 从 0 到 1 搭建完整 Web UI自动化测试学习系列 32--开源电商商城系统项目实战--如何区分登录状态
python·学习·测试工具·pytest
ZAz_3 小时前
DAY 38 模型可视化与推理
python
艾上编程3 小时前
第二章——数据分析场景之用Python进行CSV/Excel数据清洗:为数据分析筑牢根基
python·数据分析·excel