att
This commit is contained in:
parent
cc3fea3629
commit
acc3f50eee
19
ruptura.py
19
ruptura.py
@ -288,16 +288,17 @@ def get_db_connection():
|
|||||||
# 7) Baixar arquivo e inserir no banco
|
# 7) Baixar arquivo e inserir no banco
|
||||||
# ==============================
|
# ==============================
|
||||||
|
|
||||||
def download_and_insert_to_db(token, request_id, file_name, business_unit):
|
def download_and_insert_to_db(token, request_id, file_name, business_unit, cycle):
|
||||||
"""
|
"""
|
||||||
Faz o download do arquivo exportado e insere os dados no banco de dados.
|
Faz o download do arquivo exportado e insere os dados no banco de dados.
|
||||||
Remove dados existentes para a mesma data e marca antes de inserir.
|
Remove dados existentes para a mesma data, marca e ciclo antes de inserir.
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
token: Token de autenticação
|
token: Token de autenticação
|
||||||
request_id: ID da requisição de exportação
|
request_id: ID da requisição de exportação
|
||||||
file_name: Nome do arquivo (usado apenas para referência)
|
file_name: Nome do arquivo (usado apenas para referência)
|
||||||
business_unit: Unidade de negócio (BOT, OUI, QDB)
|
business_unit: Unidade de negócio (BOT, OUI, QDB)
|
||||||
|
cycle: Ciclo formatado (ex: 202516)
|
||||||
|
|
||||||
Returns:
|
Returns:
|
||||||
True se o download e inserção foram bem-sucedidos, False caso contrário
|
True se o download e inserção foram bem-sucedidos, False caso contrário
|
||||||
@ -399,20 +400,20 @@ def download_and_insert_to_db(token, request_id, file_name, business_unit):
|
|||||||
dt_ruptura = (datetime.now() - timedelta(days=1)).date()
|
dt_ruptura = (datetime.now() - timedelta(days=1)).date()
|
||||||
print(f"Data de ruptura: {dt_ruptura}")
|
print(f"Data de ruptura: {dt_ruptura}")
|
||||||
|
|
||||||
# Deletar dados existentes para a mesma data e marca
|
# Deletar dados existentes para a mesma data, marca e ciclo
|
||||||
print(f"\nVerificando dados existentes para {business_unit} em {dt_ruptura}...")
|
print(f"\nVerificando dados existentes para {business_unit} - Ciclo {cycle} em {dt_ruptura}...")
|
||||||
delete_query = """
|
delete_query = """
|
||||||
DELETE FROM RupturaG
|
DELETE FROM RupturaG
|
||||||
WHERE dt_ruptura = ? AND Marca = ?
|
WHERE dt_ruptura = ? AND Marca = ? AND Ciclo = ?
|
||||||
"""
|
"""
|
||||||
cursor.execute(delete_query, dt_ruptura, business_unit)
|
cursor.execute(delete_query, dt_ruptura, business_unit, cycle)
|
||||||
rows_deleted = cursor.rowcount
|
rows_deleted = cursor.rowcount
|
||||||
conn.commit()
|
conn.commit()
|
||||||
|
|
||||||
if rows_deleted > 0:
|
if rows_deleted > 0:
|
||||||
print(f"✓ {rows_deleted} registros anteriores removidos para {business_unit} em {dt_ruptura}")
|
print(f"✓ {rows_deleted} registros anteriores removidos para {business_unit} - Ciclo {cycle} em {dt_ruptura}")
|
||||||
else:
|
else:
|
||||||
print(f"✓ Nenhum registro anterior encontrado para {business_unit} em {dt_ruptura}")
|
print(f"✓ Nenhum registro anterior encontrado para {business_unit} - Ciclo {cycle} em {dt_ruptura}")
|
||||||
|
|
||||||
print(f"\nInserindo dados na tabela RupturaG...")
|
print(f"\nInserindo dados na tabela RupturaG...")
|
||||||
|
|
||||||
@ -529,7 +530,7 @@ def processar_marca(token, business_unit, cycle):
|
|||||||
return False
|
return False
|
||||||
|
|
||||||
# 3. Fazer o download do arquivo e inserir no banco
|
# 3. Fazer o download do arquivo e inserir no banco
|
||||||
download_success = download_and_insert_to_db(token, request_id, file_name, business_unit)
|
download_success = download_and_insert_to_db(token, request_id, file_name, business_unit, cycle)
|
||||||
|
|
||||||
if download_success:
|
if download_success:
|
||||||
print(f"\n{'='*60}")
|
print(f"\n{'='*60}")
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user