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

#創(chuàng)建一個全局變量MATERIAL_MESSAGE
MATERIAL_MESSAGE = []
#獲取小宛中數(shù)據(jù)
def get_construct_message():
? ? #讀取數(shù)據(jù)原表.xlsx表
? ? wb = openpyxl.load_workbook('數(shù)據(jù)原表.xlsx')
? ? #獲取獲取當前活動工作表，默認為第一個表
? ? ws = wb.active
? ? #獲取在生產(chǎn)記錄更新.xlsx中的可以用到的數(shù)據(jù)存入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 = []
? ? ? ? ? ? ? ? #將日期轉換為與生產(chǎn)記錄更新中相對應寫法的形式
? ? ? ? ? ? ? ? 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)

# 將客戶要求的對應數(shù)據(jù)存入字典中
TABLES = {"水泥":"水泥1","粉煤灰":"煤灰","河沙":"沙1","粗骨料1":"石1","粗骨料2":"石2","減水劑":"外加劑","水":"水"}
#存入生產(chǎn)記錄更新表的日期和其所在的行數(shù)
DATE = []
#存入生產(chǎn)記錄更新表的材料和其所在的列數(shù)
MATERIAL = []
wb = openpyxl.load_workbook('生產(chǎn)記錄更新.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
? ? ? ? #判斷該單元格的內(nèi)容是否在TABLES中
? ? ? ? if column.value in TABLES.values():
? ? ? ? #運用行列轉換的方法進行轉換
? ? ? ? ? ? MATERIAL.append([column.value,get_column_letter(n - 1)])
? ? pprint.pprint(MATERIAL)

#兩個參數(shù)分別為材料的類型和所在的列數(shù)
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之間的隨機數(shù)，否則-1到1之間的隨機數(shù)，并更新誤差值
? ? ? ? ? ? ? ? ? ? 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('生產(chǎn)記錄更新.xlsx')

if __name__ == "__main__":
? ? main()