Rafael Lopes
4af74ab458
All checks were successful
Deploy WiFi-ETL Prod / deploy (push) Successful in 47s
- Substitui uso da Wifeed pela Wifere como fonde de dados de coleta dos usuarios - Correção de cron - Melhoria de logging
43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
import time
|
|
from datetime import date, timedelta
|
|
import psycopg2
|
|
|
|
from app.extractor.wifire import extract_customers
|
|
from app.core.config import DB_HOST, DB_PORT, DB_NAME, DB_USER, DB_PASSWORD
|
|
from app.transform.merge_mac import transform_wifire
|
|
from app.load.load_database import load
|
|
|
|
START_DATE = date(2026, 4, 21)
|
|
END_DATE = date.today() - timedelta(days=1)
|
|
|
|
SLEEP_BETWEEN = 20 # 20s entre requisições = 3 req/min, abaixo do limite de 10 req/min
|
|
|
|
conn = psycopg2.connect(host=DB_HOST, port=DB_PORT, dbname=DB_NAME, user=DB_USER, password=DB_PASSWORD)
|
|
conn.autocommit = False
|
|
|
|
current = START_DATE
|
|
while current <= END_DATE:
|
|
print(f"Processando {current}...")
|
|
try:
|
|
raw, _ = extract_customers(watermark_date=current)
|
|
users = [transform_wifire(r) for r in raw]
|
|
if users:
|
|
load(conn, users, [])
|
|
conn.commit()
|
|
print(f" {len(users)} usuários inseridos")
|
|
else:
|
|
print(f" sem dados")
|
|
except Exception as e:
|
|
conn.rollback()
|
|
if "429" in str(e):
|
|
print(f" Rate limit — aguardando 90s...")
|
|
time.sleep(90)
|
|
else:
|
|
print(f" ERRO em {current}: {e}")
|
|
|
|
current += timedelta(days=1)
|
|
time.sleep(SLEEP_BETWEEN)
|
|
|
|
conn.close()
|
|
print("Backfill concluído!")
|