Consultme-GI/busca_pdv.py
2025-01-29 09:00:34 -03:00

22 lines
838 B
Python

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)