chatgraph 0.2.4__py3-none-any.whl → 0.2.6__py3-none-any.whl

chatgraph/bot/chatbot_model.py

@@ -139,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()
@@ -149,9 +155,13 @@ class ChatbotApp(ABC):
                 'obs': user_state.obs,
             }
             return json.dumps(response)            
-        elif type(message_response) in (ChatbotResponse, EndChatResponse, TransferToHuman):
+        elif type(message_response) in (EndChatResponse, TransferToHuman):
             response = message_response.json()
-            response['customer_id'] = customer_id
+            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)

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/messages/rabbitMQ_message_consumer.py

@@ -164,14 +164,16 @@ 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', ''),
@@ -205,7 +207,3 @@ class RabbitMessageConsumer(MessageConsumer):
         console.print(title_panel, justify="center")
         console.print(separator, justify="center")
         console.print(table, justify="center")
-        
-c = Credential('user', 'pass')	
-consumer = RabbitMessageConsumer(c, "amqp://example.com", "queue_name", 10, "/virtual_host")
-consumer.reprer()

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/types/output_state.py

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

{chatgraph-0.2.4.dist-info → chatgraph-0.2.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: chatgraph
-Version: 0.2.4
+Version: 0.2.6
 Summary: A user-friendly chatbot library
 Home-page: https://github.com/irissonnlima/chatgraph
 License: MIT

{chatgraph-0.2.4.dist-info → chatgraph-0.2.6.dist-info}/RECORD

@@ -1,17 +1,19 @@
 chatgraph/__init__.py,sha256=6pDNMUOLR8iFK_3mWula1IJQcIoE8io-65mS_QheXXE,642
 chatgraph/auth/credentials.py,sha256=xsMEpLQnc66novPjL6upocMcnUnvFJ7yxINzUenkxmc,2388
-chatgraph/bot/chatbot_model.py,sha256=IBmFLGRkFh4eEwNInIMw3czQ5Kdry8NmQ4dCg81Yqro,6626
+chatgraph/bot/chatbot_model.py,sha256=4XTlAxFIpEeNp7hwxC3UjhnDeZMFD3QH3AD09GDxD78,6943
 chatgraph/bot/chatbot_router.py,sha256=kZ_4X5AhhE2mntx4tcHiapaSKC1T4K1jqIVjLp1Pjg0,2817
 chatgraph/error/chatbot_error.py,sha256=4sEcW8vz0eQk2QDmDygucVM4caHliZW5iH7XJvmLBuk,1897
 chatgraph/error/route_error.py,sha256=CY8m82ap7-Sr-DXPsolltRW50TqD__5RyXBmNNJCWr8,793
 chatgraph/messages/base_message_consumer.py,sha256=4qZj69eDtqeqmSDSfGsCRt6hyypSPTSry3pmRh8r8nM,1269
-chatgraph/messages/rabbitMQ_message_consumer.py,sha256=xdKrHeq1zV_0FDy1JtcLpWGW0RoUD-8q32ZWvnsdEOQ,8851
+chatgraph/messages/base_message_sender.py,sha256=0IgNW0WMwRXnXiKokqhMtX5imK3L6ZDKEYjFdSbQIq4,610
+chatgraph/messages/rabbitMQ_message_consumer.py,sha256=Y9Y0BuHHPBj3qJZNVBQ0QA9XI-T_AqUR-ULd5WSeY1A,8771
+chatgraph/messages/rabbitMQ_message_sender.py,sha256=TWegMfDasjUjKF5u8LVfr8ai32E3oBa_EELtGrTS7xk,3213
 chatgraph/messages/test_message_consumer.py,sha256=9XIkbCHd1S6S8pINRT-SLEvUT0TQWBCsdPhYN6PpZ2s,518
 chatgraph/types/message_types.py,sha256=7eB45hrlbYKV1Lp9vysR6V7OXNsySpeKYbGBc2Ux6xE,1363
-chatgraph/types/output_state.py,sha256=C8kNy3bqSKBW3jAF-37Jrph-1Kqw08cY5dS3_bo3czA,3028
+chatgraph/types/output_state.py,sha256=_pbCssNM4ls7tFG1Wbdg8BJN6ih0q3x2onXZdvZcGDo,3025
 chatgraph/types/route.py,sha256=nKTqzwGl7d_Bu8G6Sr0EmmhuuiZWKEoSozITRrdyi1g,2587
 chatgraph/types/user_state.py,sha256=nliV-kC_yOmYoFh8CQSfT92DbOv4BLF0vvn8kotnDog,2884
-chatgraph-0.2.4.dist-info/LICENSE,sha256=rVJozpRzDlplOpvI8A1GvmfVS0ReYdZvMWc1j2jV0v8,1090
-chatgraph-0.2.4.dist-info/METADATA,sha256=6JnfS0l3ArZorgyQNUX93L9S-3hA6NuA-IMgPyDXsy0,5868
-chatgraph-0.2.4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-chatgraph-0.2.4.dist-info/RECORD,,
+chatgraph-0.2.6.dist-info/LICENSE,sha256=rVJozpRzDlplOpvI8A1GvmfVS0ReYdZvMWc1j2jV0v8,1090
+chatgraph-0.2.6.dist-info/METADATA,sha256=lRvleQCJFEMf1tjGSNYaXQZ4n_tFzrVpSNp85sWX-1Y,5868
+chatgraph-0.2.6.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+chatgraph-0.2.6.dist-info/RECORD,,