cybertron-spark 0.1.5__tar.gz → 0.1.7__tar.gz

{cybertron_spark-0.1.5 → cybertron_spark-0.1.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cybertron-spark
-Version: 0.1.5
+Version: 0.1.7
 Summary: Automação do Zendesk com Selenium, Zenpy e integração Cybertron
 Author: Gustavo Sartorio
 Author-email: strov3rl@gmail.com

{cybertron_spark-0.1.5 → cybertron_spark-0.1.7}/cybertron_spark/heartbeat.py

@@ -4,6 +4,8 @@ import requests
 import traceback
 import urllib3
 from dotenv import load_dotenv
+import pandas as pd
+import numpy as np
 
 # Desabilita avisos de certificados inseguros (opcional, conforme seu código original)
 urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
@@ -100,7 +102,7 @@ class Heartbeat():
             'traceback': codigo_quebrou
         }
         
-    def registrar_ticket(self, ticket_id, pedido=None, protocolo=None, output_data=None):
+    def registrar_ticket(self, ticket_id=None, pedido=None, protocolo=None, output_data=None):
         endpoint = '/transformers/bots-tickets/'
 
         payload = {
@@ -113,6 +115,33 @@ class Heartbeat():
     
         return self._post(endpoint, payload)
 
+    def registrar_lote(self, lista_ou_df):
+        endpoint = '/transformers/bots-tickets/'
+        payload_lote = []
+
+        if hasattr(lista_ou_df, 'to_dict'):
+            dados_raw = lista_ou_df.to_dict(orient='records')
+
+        elif isinstance(lista_ou_df, list):
+            dados_raw = lista_ou_df
+        else:
+            print("❌ Erro: registrar_lote espera uma Lista ou DataFrame.")
+            return None
+
+        for item in dados_raw:
+            registro = {
+                'bot_id': self.bot_id, 
+                'ticket_id': str(item.get('ticket_id') or item.get('service_protocol') or item.get('pedido') or item.get('order_code') or ''),
+                'pedido': str(item.get('pedido') or item.get('order_code') or ''),
+                'protocolo': str(item.get('protocolo') or item.get('service_protocol') or ''),
+                'output_data': item 
+            }
+            payload_lote.append(registro)
+
+        if payload_lote:
+            print(f"📤 Enviando lote de {len(payload_lote)} tickets...")
+            return self._post(endpoint, payload_lote)
+
     def atualizar_fila(self, quantidade):
         endpoint = '/transformers/bots-queue/'
 
@@ -175,4 +204,36 @@ class Heartbeat():
         'bot_realizado': True,
         }
     
-        return self._patch(endpoint, payload) 
+        return self._patch(endpoint, payload) 
+
+    def salvar_bugiganga(self, dataframe):
+        endpoint = '/transformers/bots-bugigangasave/'
+        
+        df_save = dataframe.copy()
+        colunas_data = ['abertura', 'dt_status', 'dt_cancelamento', 'ult_msg', 'dt_entrega', 'dt_prevista', 'created_at']
+        
+        for col in colunas_data:
+            if col in df_save.columns:
+                df_save[col] = pd.to_datetime(df_save[col], errors='coerce').dt.strftime('%Y-%m-%dT%H:%M:%S')
+
+        df_save = df_save.replace({np.nan: None})
+        df_save = df_save.where(pd.notnull(df_save), None)
+
+        payload_total = df_save.to_dict(orient='records')
+        
+        batch_size = 500
+        total_enviado = 0
+        
+        print(f"📦 Iniciando envio de {len(payload_total)} registros para o banco...")
+
+        for i in range(0, len(payload_total), batch_size):
+            lote = payload_total[i:i + batch_size]
+            
+            response = self._post(endpoint, lote)
+            if response and response.status_code in [200, 201]:
+                total_enviado += len(lote)
+            else:
+                print(f"⚠️ Falha no lote {i} a {i+batch_size}")
+
+        print(f"🏁 Finalizado. Total salvo no banco: {total_enviado}")
+        return total_enviado

{cybertron_spark-0.1.5 → cybertron_spark-0.1.7}/cybertron_spark.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cybertron-spark
-Version: 0.1.5
+Version: 0.1.7
 Summary: Automação do Zendesk com Selenium, Zenpy e integração Cybertron
 Author: Gustavo Sartorio
 Author-email: strov3rl@gmail.com

{cybertron_spark-0.1.5 → cybertron_spark-0.1.7}/setup.py

@@ -5,7 +5,7 @@ with open("README.md", "r", encoding="utf-8") as arq:
 
 setup(
     name='cybertron-spark',
-    version='0.1.5',
+    version='0.1.7',
     license='MIT',
     author='Gustavo Sartorio',
     author_email='strov3rl@gmail.com',