chatgraph 0.2.3__tar.gz → 0.2.6__tar.gz

{chatgraph-0.2.3 → chatgraph-0.2.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: chatgraph
-Version: 0.2.3
+Version: 0.2.6
 Summary: A user-friendly chatbot library
 Home-page: https://github.com/irissonnlima/chatgraph
 License: MIT
@@ -12,6 +12,7 @@ Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: pika (>=1.3.2,<2.0.0)
+Requires-Dist: rich (>=13.8.1,<14.0.0)
 Project-URL: Repository, https://github.com/irissonnlima/chatgraph
 Description-Content-Type: text/markdown
 

{chatgraph-0.2.3 → chatgraph-0.2.6}/chatgraph/bot/chatbot_model.py

@@ -98,8 +98,9 @@ class ChatbotApp(ABC):
         """
         Inicia o consumo de mensagens pelo chatbot, processando cada mensagem recebida.
         """
+        self.__message_consumer.reprer()
         self.__message_consumer.start_consume(self.process_message)
-
+    
     def process_message(self, message: Message):
         """
         Processa uma mensagem recebida, identificando a rota correspondente e executando a função associada.
@@ -138,7 +139,13 @@ class ChatbotApp(ABC):
 
         if type(message_response) in (str, float, int):
             response = ChatbotResponse(message_response)
-            return response.json()
+            response = message_response.json()
+            response['user_state'] = {
+                'customer_id': customer_id,
+                'menu': menu,
+                'obs': user_state.obs,
+            }
+            return json.dumps(response)
         elif type(message_response) == ChatbotResponse:
             route = self.__adjust_route(message_response.route, menu)
             response = message_response.json()
@@ -148,8 +155,14 @@ class ChatbotApp(ABC):
                 'obs': user_state.obs,
             }
             return json.dumps(response)            
-        elif type(message_response) in (ChatbotResponse, EndChatResponse, TransferToHuman):
-            return json.dumps(message_response.json())
+        elif type(message_response) in (EndChatResponse, TransferToHuman):
+            response = message_response.json()
+            response['user_state'] = {
+                'customer_id': customer_id,
+                'menu': None,
+                'obs': None,
+            }
+            return json.dumps(response)
         elif type(message_response) == RedirectResponse:
             route = self.__adjust_route(message_response.route, menu)
             message.user_state.menu = route
@@ -174,4 +187,4 @@ class ChatbotApp(ABC):
         if 'start' not in route:
             route = absolute_route + route
 
-        return route
+        return route

{chatgraph-0.2.3 → chatgraph-0.2.6}/chatgraph/messages/base_message_consumer.py

@@ -31,3 +31,10 @@ class MessageConsumer(ABC):
             MessageConsumer: A instância do consumidor de mensagens configurado.
         """
         pass
+    
+    @abstractmethod
+    def reprer(self) -> str:
+        '''
+        Retorna uma representação textual do consumidor de mensagens.
+        '''
+        pass

chatgraph-0.2.6/chatgraph/messages/base_message_sender.py

@@ -0,0 +1,23 @@
+from abc import ABC, abstractmethod
+
+class MessageSender(ABC):
+    
+    @abstractmethod
+    def load_dotenv(self) -> 'MessageSender':
+        """
+        Carrega variáveis de ambiente para configurar o disparador de mensagens.
+        
+        Returns:
+            MessageSender: A instância do disparador de mensagens configurado.
+        """
+        pass
+    
+    @abstractmethod
+    def send_message(self) -> None:
+        """
+        Envia uma mensagem para o destino especificado.
+        
+        Args:
+            message (str): A mensagem a ser enviada.
+        """
+        pass

{chatgraph-0.2.3 → chatgraph-0.2.6}/chatgraph/messages/rabbitMQ_message_consumer.py

@@ -6,7 +6,10 @@ from typing import Callable
 from ..auth.credentials import Credential
 from ..types.message_types import Message, UserState
 from .base_message_consumer import MessageConsumer
-
+from rich.console import Console
+from rich.table import Table
+from rich.text import Text
+from rich.panel import Panel
 
 class RabbitMessageConsumer(MessageConsumer):
     """
@@ -161,17 +164,46 @@ class RabbitMessageConsumer(MessageConsumer):
         Returns:
             Message: Uma instância da classe Message com os dados extraídos do dicionário.
         """
+        
+        user_state = message.get('user_state', {})
         return Message(
             type=message.get('type', ''),
             text=message.get('text', ''),
             user_state=UserState(
-                customer_id=message.get('customer_id', ''),
-                menu=message.get('menu', ''),
-                lst_update=message.get('lst_update', ''),
-                obs=message.get('obs', {}),
+                customer_id=user_state.get('customer_id', ''),
+                menu=user_state.get('menu', ''),
+                lst_update=user_state.get('lst_update', ''),
+                obs=user_state.get('obs', {}),
             ),
             channel=message.get('channel', ''),
             customer_phone=message.get('customer_phone', ''),
             company_phone=message.get('company_phone', ''),
             status=message.get('status'),
         )
+
+    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")
+
+        table.add_row("Virtual Host", self.__virtual_host)
+        table.add_row("Prefetch Count", str(self.__prefetch_count))
+        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
+
+        # 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-0.2.6/chatgraph/messages/rabbitMQ_message_sender.py

@@ -0,0 +1,79 @@
+from .base_message_sender import MessageSender
+from ..auth.credentials import Credential
+import pika
+import os
+
+class RabbitMessageSender(MessageSender):
+    def __init__(
+        self,
+        credential: Credential,
+        amqp_url: str,
+        send_queue: 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.__send_queue = send_queue
+        self.__amqp_url = amqp_url
+        self.__credentials = pika.PlainCredentials(
+            credential.username, credential.password
+        )
+    
+    @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',
+    ) -> 'RabbitMessageSender':
+        """
+        Carrega variáveis de ambiente para configurar o disparador de mensagens.
+        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 as variáveis de ambiente não forem configuradas corret
+        Returns:
+            RabbitMessageSender: A instância do disparador de mensagens configurado.
+        """
+        username = os.getenv(user_env)
+        password = os.getenv(pass_env)
+        url = os.getenv(uri_env)
+        queue = os.getenv(queue_env)
+        vhost = os.getenv(vhost_env, '/')
+
+        if not username or not password or not url or not queue:
+            raise ValueError('Corrija as variáveis de ambiente!')
+
+        return cls(
+            credential=Credential(username=username, password=password),
+            amqp_url=url,
+            send_queue=queue,
+            virtual_host=vhost,
+        )
+    
+    
+    def send_message(self) -> None:
+        """
+        Envia uma mensagem para o destino especificado.
+        
+        Args:
+            message (str): A mensagem a ser enviada.
+        """
+        pass

{chatgraph-0.2.3 → chatgraph-0.2.6}/chatgraph/types/output_state.py

@@ -30,7 +30,7 @@ class ChatbotResponse:
         '''
         return {
             'type': 'message',
-            'message': self.message,
+            'text': self.message,
         }
 
 

{chatgraph-0.2.3 → chatgraph-0.2.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "chatgraph"
-version = "0.2.3"
+version = "0.2.6"
 description = "A user-friendly chatbot library"
 authors = ["Irisson N. Lima <irisson.lima@verdecard.com.br>"]
 readme = "README.md"
@@ -12,6 +12,7 @@ license = "MIT"
 [tool.poetry.dependencies]
 python = "^3.12"
 pika = "^1.3.2"
+rich = "^13.8.1"
 
 [tool.poetry.group.dev.dependencies]
 ruff = "^0.5.1"