from openpyxl import load_workbook, Workbook
from bcompiler.utils import project_data_from_master

current_Q_dict = project_data_from_master('C:\\Users\\Standalone\\Will\\masters folder\\master_4_2017.xlsx')

def filter_dictionary(dictionary, category):
    f_dict = {}
    for x in dictionary:
        if dictionary[x][category] != None:
            f_dict[x] = dictionary[x]
    return f_dict

def get_project_names(data):
    project_name_list = []
    for x in data:
        project_name_list.append(x)
    return project_name_list

current_Q_list = get_project_names(current_Q_dict)

output_wb = Workbook()
ws = output_wb.active

for x in range(0, len(current_Q_list)):
    ws.cell(row=x+2, column=1, value=current_Q_list[x])

output_wb.save('C:\\Users\\Standalone\\Will\\masters folder\\Financial\\throw_away_list.xlsx')

wb = load_workbook('C:\\Users\\Standalone\\Will\\masters folder\\Financial\\throw_away_list.xlsx')
ws = wb.active

for row_num in range(2, ws.max_row + 1):
    project_name = ws.cell(row=row_num, column=1).value
    print(project_name)
    if project_name in current_Q_dict:
        ws.cell(row=row_num, column=2).value = current_Q_dict[project_name]['Source of Finance']
        ws.cell(row=row_num, column=4).value = current_Q_dict[project_name]['Other Finance type Description']

ws.cell(row=1, column=1, value='Project')
ws.cell(row=1, column=2, value='Source of finance')
ws.cell(row=1, column=3, value='source of finance amended')
ws.cell(row=1, column=4, value='notes')


#ws.cell(row=1, column=4, value='Cost to closure')



wb.save('C:\\Users\\Standalone\\Will\\masters folder\\Financial\\source_finance_dash.xlsx')