import pandas as pd

def extract_data(file):
    consultoras = pd.read_excel(f"{file}", sheet_name="PDV")
    consultoras.drop(index=1, inplace=True)
    consultoras.columns = consultoras.iloc[0]
    consultoras.drop(index=0, inplace=True)
    consultoras= consultoras.iloc[:,[0,2,3,8,9]]
    consultoras.columns =  ['PDV','GMV ATUAL', 'VAR GMV', 'BOLETO MEDIO', 'VAR BOL']
    return consultoras 

def busca_realizado(codigo_pesquisar):
    dimensao = pd.read_csv('Dimensao.csv',encoding='utf-8')
    df = dimensao
    codigos_correspondentes = df[(df['cpf_gr'] == codigo_pesquisar) | (df['cpf_sup'] == codigo_pesquisar)]['codigo'].tolist()
    dados = extract_data("consultoras.xlsx")
    pdv_filtrados = dados[dados['PDV'].isin(codigos_correspondentes)]
    receita_gmv = pdv_filtrados["GMV ATUAL"].sum()
    return int(receita_gmv)