import requests
import datas as d
from datetime import datetime
import PEF_VD as pef
import PEF_Loja as pef_loja
from csv import DictReader
from datetime import datetime, timedelta
from token_GI import main

dir = r'database'


hoje_coluna = datetime.today() # Pegando a Data e Hora de hoje
hoje_formatado = hoje_coluna.strftime('%Y-%m-%d')
mes_atual = hoje_coluna.strftime('%Y-%B')
current = datetime.now()
# Processo principal
def download():
    session = requests.Session()  # Iniciar uma sessão persistente
    with open(r"token.csv", 'r') as f:
        try:
            dict_reader = DictReader(f)
            lista = list(dict_reader)
            data =  lista[0]['data'] 
            data = datetime.strptime(data, "%Y-%m-%d %H:%M:%S") 
            delta = current - data - timedelta(hours=3)
            print(delta)
            if delta > timedelta(minutes=30):
                print('token venceu, reinicie')
                token =[]
                main()

            else:
                
                print('Token não venceu')
                tokens = lista[0]['token']
        except ValueError as e:
            print(e)
    
    # Carrega

    if len(tokens) > 0:
        Access =  tokens
              #print("ID Token:", tokens["id_token"])
              #print("Refresh Token:", tokens["refresh_token"])      

        # PEF_VD

        url = pef.PEF_VD(Access)
        baixarPEF = requests.get(url)
        arquivo = fr"database/PEFVD_{mes_atual}.xlsx"
        
        if baixarPEF.status_code==200:
            with open(arquivo,"wb") as file:
                file.write(baixarPEF.content)

                print(f"Arquivo salvo com sucesso como {arquivo}.")
           # x=pd.read_excel(arquivo)
           # x['data'] = mes_atual
           # x.to_excel(arquivo)
        else:
            print(f"Erro ao acessar o arquivo. Código de status: {baixarPEF.status_code}")

        #PEC_VD
        url = pef_loja.PEF_LOJA(Access)
        baixarPEF_Loja = requests.get(url)
        arquivoPEC = fr"database/PEFLOJA_{mes_atual}.xlsx"
        if baixarPEF_Loja.status_code==200:
            with open(arquivoPEC,"wb") as file:
                file.write(baixarPEF_Loja.content)

                print(f"Arquivo salvo com sucesso como {arquivoPEC}.")
        else:
            print(f"Erro ao acessar o arquivo. Código de status: {baixarPEF_Loja.status_code}")    
    else:
             print("Falha ao carregar o token.")


if __name__ == "__main__":
    download()