chatgraph 0.4.2__py3-none-any.whl → 0.5.1__py3-none-any.whl

chatgraph/messages/message_consumer.py

@@ -1,27 +1,18 @@
 import json
+import asyncio
 from logging import debug, info
 import os
-import pika
+import aio_pika
 from typing import Callable
 from ..auth.credentials import Credential
-from ..types.request_types import UserCall, UserState
+from ..types.request_types import UserCall, UserState, ChatID
 from rich.console import Console
 from rich.table import Table
 from rich.text import Text
 from rich.panel import Panel
+from urllib.parse import quote
 
 class MessageConsumer:
-    """
-    Implementação de MessageConsumer para consumir mensagens de uma fila RabbitMQ.
-
-    Atributos:
-        __virtual_host (str): O host virtual usado para a conexão RabbitMQ.
-        __prefetch_count (int): O número de mensagens pré-carregadas que o consumidor pode processar.
-        __queue_consume (str): O nome da fila de consumo.
-        __amqp_url (str): A URL de conexão AMQP do RabbitMQ.
-        __credentials (pika.PlainCredentials): Credenciais do RabbitMQ para autenticação.
-    """
-
     def __init__(
         self,
         credential: Credential,
@@ -29,65 +20,36 @@ class MessageConsumer:
         grpc_uri: str,
         queue_consume: str,
         prefetch_count: int = 1,
-        virtual_host: str = '/',
+        virtual_host: str = "/",
     ) -> None:
-        """
-        Inicializa o consumidor de mensagens RabbitMQ com as configurações fornecidas.
-
-        Args:
-            credential (Credential): Credenciais de autenticação para o RabbitMQ.
-            amqp_url (str): A URL de conexão AMQP do RabbitMQ.
-            queue_consume (str): O nome da fila de consumo.
-            prefetch_count (int, opcional): O número de mensagens pré-carregadas. Padrão é 1.
-            virtual_host (str, opcional): O host virtual do RabbitMQ. Padrão é '/'.
-        """
         self.__virtual_host = virtual_host
         self.__prefetch_count = prefetch_count
         self.__queue_consume = queue_consume
         self.__amqp_url = amqp_url
         self.__grpc_uri = grpc_uri
-        self.__credentials = pika.PlainCredentials(
-            credential.username, credential.password
-        )
+        self.__credentials = credential
 
     @classmethod
     def load_dotenv(
         cls,
-        user_env: str = 'RABBIT_USER',
-        pass_env: str = 'RABBIT_PASS',
-        uri_env: str = 'RABBIT_URI',
-        queue_env: str = 'RABBIT_QUEUE',
-        prefetch_env: str = 'RABBIT_PREFETCH',
-        vhost_env: str = 'RABBIT_VHOST',
-        grpc_uri: str = 'GRPC_URI',
-    ) -> 'MessageConsumer':
-        """
-        Carrega as configurações do RabbitMQ a partir de variáveis de ambiente e retorna uma instância de RabbitMessageConsumer.
-
-        Args:
-            user_env (str): Nome da variável de ambiente para o usuário do RabbitMQ. Padrão é 'RABBIT_USER'.
-            pass_env (str): Nome da variável de ambiente para a senha do RabbitMQ. Padrão é 'RABBIT_PASS'.
-            uri_env (str): Nome da variável de ambiente para a URL do RabbitMQ. Padrão é 'RABBIT_URI'.
-            queue_env (str): Nome da variável de ambiente para a fila de consumo do RabbitMQ. Padrão é 'RABBIT_QUEUE'.
-            prefetch_env (str): Nome da variável de ambiente para o prefetch count. Padrão é 'RABBIT_PREFETCH'.
-            vhost_env (str): Nome da variável de ambiente para o host virtual do RabbitMQ. Padrão é 'RABBIT_VHOST'.
-
-        Raises:
-            ValueError: Se qualquer uma das variáveis de ambiente necessárias não estiver definida.
-
-        Returns:
-            RabbitMessageConsumer: Uma instância configurada do RabbitMessageConsumer.
-        """
+        user_env: str = "RABBIT_USER",
+        pass_env: str = "RABBIT_PASS",
+        uri_env: str = "RABBIT_URI",
+        queue_env: str = "RABBIT_QUEUE",
+        prefetch_env: str = "RABBIT_PREFETCH",
+        vhost_env: str = "RABBIT_VHOST",
+        grpc_uri: str = "GRPC_URI",
+    ) -> "MessageConsumer":
         username = os.getenv(user_env)
         password = os.getenv(pass_env)
         url = os.getenv(uri_env)
         queue = os.getenv(queue_env)
         prefetch = os.getenv(prefetch_env, 1)
-        vhost = os.getenv(vhost_env, '/')
+        vhost = os.getenv(vhost_env, "/")
         grpc = os.getenv(grpc_uri)
 
         if not username or not password or not url or not queue or not grpc:
-            raise ValueError('Corrija as variáveis de ambiente!')
+            raise ValueError("Corrija as variáveis de ambiente!")
 
         return cls(
             credential=Credential(username=username, password=password),
@@ -98,106 +60,80 @@ class MessageConsumer:
             grpc_uri=grpc,
         )
 
-    def start_consume(self, process_message: Callable) -> None:
-        """
-        Inicia o consumo de mensagens da fila RabbitMQ e processa cada mensagem usando a função fornecida.
-
-        Args:
-            process_message (Callable): Função de callback que processa cada mensagem recebida.
-
-        Raises:
-            pika.exceptions.StreamLostError: Se a conexão com o RabbitMQ for perdida, tentará reconectar automaticamente.
-        """
+    async def start_consume(self, process_message: Callable):
         try:
-            connection = pika.BlockingConnection(
-                pika.ConnectionParameters(
-                    host=self.__amqp_url,
-                    virtual_host=self.__virtual_host,
-                    credentials=self.__credentials,
-                )
-            )
-            channel = connection.channel()
-
-            channel.basic_qos(prefetch_count=self.__prefetch_count)
-            channel.basic_consume(
-                queue=self.__queue_consume,
-                on_message_callback=lambda c, m, p, b: self.on_request(
-                    c, m, p, b, process_message
-                ),
-            )
-
-            info('[x] Server inicializado! Aguardando solicitações RPC')
-            channel.start_consuming()
-        except pika.exceptions.StreamLostError as e:
-            debug(e)
-            self.start_consume(process_message)
-
-    def on_request(self, ch, method, props, body, process_message) -> None:
-        """
-        Processa uma mensagem recebida e publica a resposta de volta na fila especificada.
-
-        Args:
-            ch: Canal do RabbitMQ.
-            method: Método de entrega do RabbitMQ.
-            props: Propriedades da mensagem do RabbitMQ.
-            body: Corpo da mensagem recebida.
-            process_message (Callable): Função que processa a mensagem e retorna uma resposta.
-        """
-        message = body.decode()
-        message_json = json.loads(message)
-        pure_message = self.__transform_message(message_json)
-        process_message(pure_message)
-
-        ch.basic_ack(delivery_tag=method.delivery_tag)
+            user = quote(self.__credentials.username)
+            pwd = quote(self.__credentials.password)
+            vhost = quote(self.__virtual_host)
+            uri = self.__amqp_url
+            amqp_url = f"amqp://{user}:{pwd}@{uri}/{vhost}"
+            connection = await aio_pika.connect_robust(amqp_url)
+
+            async with connection:
+                channel = await connection.channel()
+                await channel.set_qos(prefetch_count=self.__prefetch_count)
+                
+                try:
+                    queue = await channel.get_queue(self.__queue_consume, ensure=True)
+                except aio_pika.exceptions.ChannelNotFoundEntity:
+                    arguments = {
+                        "x-dead-letter-exchange": "log_error",  # Dead Letter Exchange
+                        "x-expires": 86400000,                 # Expiração da fila (em milissegundos)
+                        "x-message-ttl": 300000                # Tempo de vida das mensagens (em milissegundos)
+                    }
+                    queue = await channel.declare_queue(self.__queue_consume, durable=True, arguments=arguments)
+                
+                info("[x] Server inicializado! Aguardando solicitações RPC")
+
+                async for message in queue:
+                    async with message.process():
+                        await self.on_request(message.body, process_message)
+        except Exception as e:
+            print(f"Erro durante o consumo de mensagens: {e}")
+            # Reiniciar a conexão em caso de falha
+            # await self.start_consume(process_message)
+
+    async def on_request(self, body: bytes, process_message: Callable):
+        try:
+            message = body.decode()
+            message_json = json.loads(message)
+            pure_message = self.__transform_message(message_json)
+            await process_message(pure_message)
+        except Exception as e:
+            print(f"Erro ao processar mensagem: {e}")
 
     def __transform_message(self, message: dict) -> UserCall:
-        """
-        Transforma o dicionário JSON recebido em uma instância de Message.
-
-        Args:
-            message (dict): Dicionário contendo os dados da mensagem.
+        user_state = message.get("user_state", {})
+        observation = user_state.get("observation", {})
+        if isinstance(observation, str):
+            observation = json.loads(observation)
 
-        Returns:
-            Message: Uma instância da classe Message com os dados extraídos do dicionário.
-        """
-        
-        user_state = message.get('user_state', {})
-        obs = user_state.get('obs', {})
-        if isinstance(obs, str):
-            obs = json.loads(obs)
-            
-        return UserCall(
-            type=message.get('type', ''),
-            text=message.get('text', ''),
+        usercall = UserCall(
             user_state=UserState(
-                customer_id=user_state.get('customer_id', ''),
-                menu=user_state.get('menu', ''),
-                route=user_state.get('route', ''),
-                lst_update=user_state.get('lst_update', ''),
-                direction_in=user_state.get('direction_in', ''),
-                voll_id=user_state.get('voll_id', ''),
-                platform=user_state.get('platform', ''),
-                obs=obs,
-                protocol=user_state.get('protocol', ''),
+                chatID=ChatID(
+                    user_id=user_state['chat_id'].get("user_id", ""),
+                    company_id=user_state['chat_id'].get("company_id", ""),
+                ),
+                menu=user_state.get("menu", ""),
+                route=user_state.get("route", ""),
+                protocol=user_state.get("protocol", ""),
+                observation=observation,
             ),
-            channel=message.get('channel', ''),
-            customer_phone=message.get('customer_phone', ''),
-            company_phone=message.get('company_phone', ''),
-            status=message.get('status'),
+            type_message=message.get("type_message", ""),
+            content_message=message.get("content_message", ""),
             grpc_uri=self.__grpc_uri,
         )
+        
+        return usercall
 
     def reprer(self):
         console = Console()
 
-        # Título "ChatGraph" destacado em vermelho e negrito dentro de um painel
         title_text = Text("ChatGraph", style="bold red", justify="center")
         title_panel = Panel.fit(title_text, title=" ", border_style="bold red", padding=(1, 4))
 
-        # Linha separadora com emojis
         separator = Text("🐇🐇🐇 RabbitMessageConsumer 📨📨📨", style="cyan", justify="center")
 
-        # Criação da tabela com os atributos
         table = Table(show_header=True, header_style="bold magenta", title="RabbitMQ Consumer")
         table.add_column("Atributo", justify="center", style="cyan", no_wrap=True)
         table.add_column("Valor", justify="center", style="magenta")
@@ -207,9 +143,8 @@ class MessageConsumer:
         table.add_row("Queue Consume", self.__queue_consume)
         table.add_row("AMQP URL", self.__amqp_url)
         table.add_row("Username", self.__credentials.username)
-        table.add_row("Password", "******")  # Oculta a senha
+        table.add_row("Password", "******")
 
-        # Imprime o título, separador e a tabela centralizada
         console.print(title_panel, justify="center")
         console.print(separator, justify="center")
         console.print(table, justify="center")

chatgraph/pb/router.proto

@@ -0,0 +1,127 @@
+syntax = "proto3";
+
+package chatbot;
+
+option go_package = "./chatbot";
+
+///// Serviços de Estado do Usuário /////
+service UserStateService {
+    rpc InsertUpdateUserState(UserState) returns (RequestStatus);
+    rpc DeleteUserState(ChatID) returns (RequestStatus);
+    rpc GetUserState(ChatID) returns (UserState);
+    rpc GetAllUserStates(Void) returns (UserStateList);
+}
+
+///// Serviços de Mensagens /////
+service SendMessage {
+    rpc SendMessage(Message) returns (RequestStatus);
+}
+
+///// Serviços de Transfer /////
+service Transfer {
+    rpc GetAllCampaigns(Void) returns (CampaignsList);
+    rpc GetCampaignID(CampaignName) returns (CampaignDetails);
+    rpc TransferToHuman(TransferToHumanRequest) returns (RequestStatus);
+    rpc TransferToMenu(TransferToMenuRequest) returns (RequestStatus);
+}
+
+///// Serviços de EndChat /////
+service EndChat {
+    rpc GetAllTabulations(Void) returns (TabulationsList);
+    rpc GetTabulationID(TabulationName) returns (TabulationDetails);
+    rpc EndChat(EndChatRequest) returns (RequestStatus);
+}
+
+///// Mensagens Compartilhadas /////
+message Void {}
+
+message RequestStatus {
+    bool status = 1;
+    string message = 2;
+}
+
+message ChatID {
+    string user_id = 1;
+    string company_id = 2;
+}
+///// Mensagens para Estado do Usuário /////
+message UserState {
+    ChatID chat_id = 1;
+    string menu = 2;
+    string route = 3;
+    string protocol = 4;
+    string observation = 5;
+}
+
+message UserStateList {
+    repeated UserState user_states = 1;
+}
+
+///// Mensagens para Serviços de Mensagens /////
+message TextMessage {
+    string type = 1;
+    string url = 2;
+    string filename = 3;
+    string title = 4;
+    string detail = 5;
+    string caption = 6;
+}
+
+message Button{
+    string type = 1;
+    string title = 2;
+    string detail = 3;
+}
+
+message Message {
+    ChatID chat_id = 1;
+    TextMessage message = 2;
+    repeated Button buttons = 3;
+    Button display_button = 4;
+}
+
+///// Mensagens para Serviços de Transfer /////
+message TransferToHumanRequest{
+    ChatID chat_id = 1;
+    string campaign_id = 2;
+}
+
+message TransferToMenuRequest{
+    ChatID chat_id = 1;
+    string menu = 2;
+    string user_message = 3;
+}
+
+message TabulationName{
+    string name = 1;
+}
+
+message TabulationDetails{
+    string id = 1;
+    string name = 2;
+    string last_update = 3;
+}
+
+message TabulationsList{
+    repeated TabulationDetails tabulations = 1;
+}
+
+///// Serviços de EndChat /////
+message EndChatRequest {
+    ChatID chat_id = 1;
+    string tabulation_id = 2;
+}
+
+message CampaignName {
+    string name = 1;
+}
+
+message CampaignDetails {
+    string id = 1;
+    string name = 2;
+    string last_update = 3;
+}
+
+message CampaignsList {
+    repeated CampaignDetails campaigns = 1;
+}

chatgraph/pb/router_pb2.py

@@ -0,0 +1,77 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: router.proto
+# Protobuf Python Version: 5.27.2
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    5,
+    27,
+    2,
+    '',
+    'router.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0crouter.proto\x12\x07\x63hatbot\"\x06\n\x04Void\"0\n\rRequestStatus\x12\x0e\n\x06status\x18\x01 \x01(\x08\x12\x0f\n\x07message\x18\x02 \x01(\t\"-\n\x06\x43hatID\x12\x0f\n\x07user_id\x18\x01 \x01(\t\x12\x12\n\ncompany_id\x18\x02 \x01(\t\"q\n\tUserState\x12 \n\x07\x63hat_id\x18\x01 \x01(\x0b\x32\x0f.chatbot.ChatID\x12\x0c\n\x04menu\x18\x02 \x01(\t\x12\r\n\x05route\x18\x03 \x01(\t\x12\x10\n\x08protocol\x18\x04 \x01(\t\x12\x13\n\x0bobservation\x18\x05 \x01(\t\"8\n\rUserStateList\x12\'\n\x0buser_states\x18\x01 \x03(\x0b\x32\x12.chatbot.UserState\"j\n\x0bTextMessage\x12\x0c\n\x04type\x18\x01 \x01(\t\x12\x0b\n\x03url\x18\x02 \x01(\t\x12\x10\n\x08\x66ilename\x18\x03 \x01(\t\x12\r\n\x05title\x18\x04 \x01(\t\x12\x0e\n\x06\x64\x65tail\x18\x05 \x01(\t\x12\x0f\n\x07\x63\x61ption\x18\x06 \x01(\t\"5\n\x06\x42utton\x12\x0c\n\x04type\x18\x01 \x01(\t\x12\r\n\x05title\x18\x02 \x01(\t\x12\x0e\n\x06\x64\x65tail\x18\x03 \x01(\t\"\x9d\x01\n\x07Message\x12 \n\x07\x63hat_id\x18\x01 \x01(\x0b\x32\x0f.chatbot.ChatID\x12%\n\x07message\x18\x02 \x01(\x0b\x32\x14.chatbot.TextMessage\x12 \n\x07\x62uttons\x18\x03 \x03(\x0b\x32\x0f.chatbot.Button\x12\'\n\x0e\x64isplay_button\x18\x04 \x01(\x0b\x32\x0f.chatbot.Button\"O\n\x16TransferToHumanRequest\x12 \n\x07\x63hat_id\x18\x01 \x01(\x0b\x32\x0f.chatbot.ChatID\x12\x13\n\x0b\x63\x61mpaign_id\x18\x02 \x01(\t\"]\n\x15TransferToMenuRequest\x12 \n\x07\x63hat_id\x18\x01 \x01(\x0b\x32\x0f.chatbot.ChatID\x12\x0c\n\x04menu\x18\x02 \x01(\t\x12\x14\n\x0cuser_message\x18\x03 \x01(\t\"\x1e\n\x0eTabulationName\x12\x0c\n\x04name\x18\x01 \x01(\t\"B\n\x11TabulationDetails\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x13\n\x0blast_update\x18\x03 \x01(\t\"B\n\x0fTabulationsList\x12/\n\x0btabulations\x18\x01 \x03(\x0b\x32\x1a.chatbot.TabulationDetails\"I\n\x0e\x45ndChatRequest\x12 \n\x07\x63hat_id\x18\x01 \x01(\x0b\x32\x0f.chatbot.ChatID\x12\x15\n\rtabulation_id\x18\x02 \x01(\t\"\x1c\n\x0c\x43\x61mpaignName\x12\x0c\n\x04name\x18\x01 \x01(\t\"@\n\x0f\x43\x61mpaignDetails\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0c\n\x04name\x18\x02 \x01(\t\x12\x13\n\x0blast_update\x18\x03 \x01(\t\"<\n\rCampaignsList\x12+\n\tcampaigns\x18\x01 \x03(\x0b\x32\x18.chatbot.CampaignDetails2\x83\x02\n\x10UserStateService\x12\x43\n\x15InsertUpdateUserState\x12\x12.chatbot.UserState\x1a\x16.chatbot.RequestStatus\x12:\n\x0f\x44\x65leteUserState\x12\x0f.chatbot.ChatID\x1a\x16.chatbot.RequestStatus\x12\x33\n\x0cGetUserState\x12\x0f.chatbot.ChatID\x1a\x12.chatbot.UserState\x12\x39\n\x10GetAllUserStates\x12\r.chatbot.Void\x1a\x16.chatbot.UserStateList2F\n\x0bSendMessage\x12\x37\n\x0bSendMessage\x12\x10.chatbot.Message\x1a\x16.chatbot.RequestStatus2\x9c\x02\n\x08Transfer\x12\x38\n\x0fGetAllCampaigns\x12\r.chatbot.Void\x1a\x16.chatbot.CampaignsList\x12@\n\rGetCampaignID\x12\x15.chatbot.CampaignName\x1a\x18.chatbot.CampaignDetails\x12J\n\x0fTransferToHuman\x12\x1f.chatbot.TransferToHumanRequest\x1a\x16.chatbot.RequestStatus\x12H\n\x0eTransferToMenu\x12\x1e.chatbot.TransferToMenuRequest\x1a\x16.chatbot.RequestStatus2\xcb\x01\n\x07\x45ndChat\x12<\n\x11GetAllTabulations\x12\r.chatbot.Void\x1a\x18.chatbot.TabulationsList\x12\x46\n\x0fGetTabulationID\x12\x17.chatbot.TabulationName\x1a\x1a.chatbot.TabulationDetails\x12:\n\x07\x45ndChat\x12\x17.chatbot.EndChatRequest\x1a\x16.chatbot.RequestStatusB\x0bZ\t./chatbotb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'router_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'Z\t./chatbot'
+  _globals['_VOID']._serialized_start=25
+  _globals['_VOID']._serialized_end=31
+  _globals['_REQUESTSTATUS']._serialized_start=33
+  _globals['_REQUESTSTATUS']._serialized_end=81
+  _globals['_CHATID']._serialized_start=83
+  _globals['_CHATID']._serialized_end=128
+  _globals['_USERSTATE']._serialized_start=130
+  _globals['_USERSTATE']._serialized_end=243
+  _globals['_USERSTATELIST']._serialized_start=245
+  _globals['_USERSTATELIST']._serialized_end=301
+  _globals['_TEXTMESSAGE']._serialized_start=303
+  _globals['_TEXTMESSAGE']._serialized_end=409
+  _globals['_BUTTON']._serialized_start=411
+  _globals['_BUTTON']._serialized_end=464
+  _globals['_MESSAGE']._serialized_start=467
+  _globals['_MESSAGE']._serialized_end=624
+  _globals['_TRANSFERTOHUMANREQUEST']._serialized_start=626
+  _globals['_TRANSFERTOHUMANREQUEST']._serialized_end=705
+  _globals['_TRANSFERTOMENUREQUEST']._serialized_start=707
+  _globals['_TRANSFERTOMENUREQUEST']._serialized_end=800
+  _globals['_TABULATIONNAME']._serialized_start=802
+  _globals['_TABULATIONNAME']._serialized_end=832
+  _globals['_TABULATIONDETAILS']._serialized_start=834
+  _globals['_TABULATIONDETAILS']._serialized_end=900
+  _globals['_TABULATIONSLIST']._serialized_start=902
+  _globals['_TABULATIONSLIST']._serialized_end=968
+  _globals['_ENDCHATREQUEST']._serialized_start=970
+  _globals['_ENDCHATREQUEST']._serialized_end=1043
+  _globals['_CAMPAIGNNAME']._serialized_start=1045
+  _globals['_CAMPAIGNNAME']._serialized_end=1073
+  _globals['_CAMPAIGNDETAILS']._serialized_start=1075
+  _globals['_CAMPAIGNDETAILS']._serialized_end=1139
+  _globals['_CAMPAIGNSLIST']._serialized_start=1141
+  _globals['_CAMPAIGNSLIST']._serialized_end=1201
+  _globals['_USERSTATESERVICE']._serialized_start=1204
+  _globals['_USERSTATESERVICE']._serialized_end=1463
+  _globals['_SENDMESSAGE']._serialized_start=1465
+  _globals['_SENDMESSAGE']._serialized_end=1535
+  _globals['_TRANSFER']._serialized_start=1538
+  _globals['_TRANSFER']._serialized_end=1822
+  _globals['_ENDCHAT']._serialized_start=1825
+  _globals['_ENDCHAT']._serialized_end=2028
+# @@protoc_insertion_point(module_scope)