# -*- coding:utf-8 -*- 
import pandas as pd 
from math import ceil
import os

def account(adress,weight):
    
    if adress == "湖南":
        if weight = 3:
            totel = 2.5
        elif  (weight >= 3) and (weight=5):
            totel = 3.5 + ceil((weight-3))*1
        else:
            totel = ceil(weight)*1
        return totel
            
    elif adress in ["河北","天津","山西","浙江","江蘇","安徽","福建","山東","江西","廣東","廣西","河南","湖北","陜西","四川","重慶","云南","貴州"]:
        if weight = 3:
            totel = 2.5
        elif  (weight >= 3) and (weight=5):
            totel = 3.5 + ceil((weight-3))*1
        else:
            totel = ceil(weight)*2 
        return totel
    
    elif adress in ["深圳","北京","上海"]:
        if weight = 3:
            totel = 3.3
        elif  (weight >= 3) and (weight=5):
            totel = 3.5 + ceil((weight-3))*1.5
        else:
            totel = ceil(weight)*2
        return totel
            
    elif adress in ["海南","遼寧","黑龍江","吉林"]:
        if weight = 3:
            totel = 2.5
        elif  (weight >= 3) and (weight=5):
            totel = 3.5 + ceil((weight-3))*2.5
        else:
            totel = ceil(weight)*3
        return totel
    elif adress in ["內蒙古","甘肅","寧夏","青海"]:
        if  weight = 1:
            totel = 9
        else:
            totel = 9 + ceil(weight-1)*6 
        return totel
    elif adress == "新疆":
        if  weight = 1:
            totel = 15
        else:
            totel = 15 + ceil(weight-1)*12 
        return totel
    elif adress == "西藏":
        if  weight = 1:
            totel = 16
        else:
            totel = 15 + ceil(weight-1)*18 
        return totel    
    else:
        print("你輸入的省份不合法?。。?)
    
file_path = input("請輸入文件路徑：")   
sheet_name = input("請輸入工作簿名稱：")
pf = pd.read_excel(file_path,sheet_name=sheet_name)
#獲取省份一列
pro = pf["省份"].values.tolist()
#獲取重量一列
wt = pf["重量"].values.tolist()
#核算列
totel = []
for p,w in zip(pro,wt):
    print(p,w)
    totel.append(account(p,w))

pf["最新核算結果"] = totel
file_name = os.path.basename(file_path)
pf.to_excel(os.path.join(os.path.dirname(file_path),os.path.basename(file_path).split(".")[0]+sheet_name+"最新核算結果"+".xlsx"))