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()