2小时完成的第一个副业单子：Python修正excel表格数据

import openpyxl
import random
from  openpyxl.utils.cell import get_column_letter,column_index_from_string

def num_to_string(n):
    column = n - 1
    real_colum = ''
    while column >= 26:
        value = column // 26
        column = column % 26
        real_colum += chr(value + 64)
    real_colum += chr(column + 65)
    return real_colum

#创建一个全局变量MATERIAL_MESSAGE
MATERIAL_MESSAGE = []
#获取小宛中数据
def get_construct_message():
    #读取数据原表.xlsx表
    wb = openpyxl.load_workbook('数据原表.xlsx')
    #获取获取当前活动工作表，默认为第一个表
    ws = wb.active
    #获取在生产记录更新.xlsx中的可以用到的数据存入MATERIAL_MESSAGE中
    for row in range(1,ws.max_row + 1):
        if ws[f'A{row}'].value == '日期':
            for i in range(ord('B'),ord('I')):
                material = []
                #将日期转换为与生产记录更新中相对应写法的形式
                date = ws[f'B{row}'].value.translate(str.maketrans('年月','--'))
                date = date.replace('号','')
                material.append(date)
                i = chr(i)
                material.append(ws[f'{i}{row + 1}'].value)
                material.append(ws[f'{i}{row + 3}'].value)
                MATERIAL_MESSAGE.append(material)
    print(MATERIAL_MESSAGE)

# 将客户要求的对应数据存入字典中
TABLES = {"水泥":"水泥1","粉煤灰":"煤灰","河沙":"沙1","粗骨料1":"石1","粗骨料2":"石2","减水剂":"外加剂","水":"水"}
#存入生产记录更新表的日期和其所在的行数
DATE = []
#存入生产记录更新表的材料和其所在的列数
MATERIAL = []
wb = openpyxl.load_workbook('生产记录更新.xlsx')
ws = wb.active
#获取日期
def get_date():
    for row in range(3,ws.max_row + 1):
        #将日期进行分割合并转换成与MATERIAL_MESSAGE中对应日期的格式
        date = str(ws[f'B{row}'].value).split()[0]
        date_list = date.split('-')
        if '0' in date_list[1]:
            date_list[1] = date_list[1].replace('0','')
        if '0' in date_list[2]:
            date_list[2] = date_list[2].replace('0','')
        date = '-'.join(date_list)
        DATE.append([date,row])
    pprint.pprint(DATE)

#获取材料
def get_material():
    n = 0
    for column in ws[1]:
        n += 1
        #判断该单元格的内容是否在TABLES中
        if column.value in TABLES.values():
        #运用行列转换的方法进行转换
            MATERIAL.append([column.value,get_column_letter(n - 1)])
    pprint.pprint(MATERIAL)

#两个参数分别为材料的类型和所在的列数
def update_default(matrial,letter):
    for row in range(3,ws.max_row + 1):
        #判断单元格是否为空
        if ws[f'{letter}{row}'].value != None:
            for i in range(len(MATERIAL_MESSAGE)):
                #需求条件，要将连个表中的日期和材料类型进行一一对应
                if DATE[row-3][0] == MATERIAL_MESSAGE[i][0] and matrial == TABLES[MATERIAL_MESSAGE[i][1]]:
                    #修改形参letter对应列的每个符号要求的单元格对应的设定值
                    ws[f'{letter}{DATE[row - 3][1]}'].value = MATERIAL_MESSAGE[i][2]
                    #分别获取该列列号的下两列对应的列号
                    letter_random = get_column_letter(column_index_from_string(letter) + 2)
                    letter_real = get_column_letter(column_index_from_string(letter) + 1)
                    #判断该列是否在这三列中，若在其中则得到-2到2之间的随机数，否则-1到1之间的随机数，并更新误差值
                    if letter in ('AV','BB','BE'):
                        ws[f'{letter_random}{DATE[row - 3][1]}'].value = round(random.uniform(-2, 2), 1)
                    else:
                        ws[f'{letter_random}{DATE[row - 3][1]}'].value = round(random.uniform(-1, 1), 1)
                    #运用设定值和误差值更新中间的实际值
                    ws[f'{letter_real}{DATE[row - 3][1]}'].value = round(ws[f'{letter}{DATE[row - 3][1]}'].value * (1 + ws[f'{letter_random}{DATE[row - 3][1]}'].value * 0.01),2)

def main():

    get_construct_message()
    get_date()
    get_material()
    for i in MATERIAL:
        update_default(i[0],i[1])
    wb.save('生产记录更新.xlsx')


if __name__ == "__main__":
    main()