81 lines
2.4 KiB
Python
81 lines
2.4 KiB
Python
|
|
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() |