|
from openpyxl import load_workbook
|
import os
|
import shutil
|
import Base as base
|
|
|
def save_excel(model_name):
|
dst = base.model_dir + model_name +"\\jyl.xlsx"
|
shutil.copyfile(base.prefix +"雨量站输入模板.xlsx",dst)
|
|
#读取excel模板数据
|
def read_excel(model_name):
|
paths = base.model_dir + model_name +"\\jyl.xlsx"
|
data=[]
|
if not os.path.exists(paths):
|
return data
|
wb = load_workbook(filename = paths)
|
ws = wb[wb.sheetnames[1]]
|
|
for row in ws.iter_rows():
|
tmp =[]
|
for cell in row:
|
tmp.append(cell.value)
|
data.append(tmp)
|
wb.close()
|
return data
|
|
|
#读取降雨量的计算文件
|
def read_jyl_excel(model_name):
|
paths = base.model_dir + model_name +"\\jyl.xlsx"
|
data=[]
|
if not os.path.exists(paths):
|
return data
|
wb = load_workbook(filename = paths,data_only=True)
|
ws = wb[wb.sheetnames[2]]
|
|
for row in ws.iter_rows():
|
tmp =[]
|
for cell in row:
|
tmp.append(cell.value)
|
data.append(tmp)
|
wb.close()
|
return data
|
|
|
def write_excel(model_name,data):
|
paths = base.model_dir + model_name +"\\jyl.xlsx"
|
if not os.path.exists(paths):
|
save_excel(model_name)
|
wb = load_workbook(filename = paths)
|
ws = wb[wb.sheetnames[1]]
|
for i in range(len(data)):
|
for j in range(len(data[i])):
|
ws.cell(row=i+1, column=j+1).value = data[i][j]
|
wb.save(paths)
|
wb.close()
|
|
|
|
