chatgraph 0.1.3__py3-none-any.whl → 0.1.5__py3-none-any.whl

chatgraph/__init__.py

@@ -3,7 +3,7 @@ from .bot.chatbot_model import ChatbotApp
 from .bot.chatbot_router import ChatbotRouter
 from .messages.rabbitMQ_message_consumer import RabbitMessageConsumer
 from .types.message_types import Message
-from .types.output_state import ChatbotResponse, RedirectResponse
+from .types.output_state import ChatbotResponse, RedirectResponse, EndChatResponse, TransferToHuman
 from .types.route import Route
 from .types.user_state import SimpleUserState
 
@@ -17,4 +17,6 @@ __all__ = [
     'RedirectResponse',
     'RabbitMessageConsumer',
     'Route',
+    'EndChatResponse',
+    'TransferToHuman',
 ]

chatgraph/bot/chatbot_model.py

@@ -6,7 +6,7 @@ from logging import debug
 from ..error.chatbot_error import ChatbotError, ChatbotMessageError
 from ..messages.base_message_consumer import MessageConsumer
 from ..types.message_types import Message
-from ..types.output_state import ChatbotResponse, RedirectResponse
+from ..types.output_state import ChatbotResponse, RedirectResponse, EndChatResponse, TransferToHuman
 from ..types.route import Route
 from ..types.user_state import UserState
 from .chatbot_router import ChatbotRouter
@@ -143,11 +143,17 @@ class ChatbotApp(ABC):
         elif type(message_response) == ChatbotResponse:
             route = self.__adjust_route(message_response.route, menu)
             self.__user_state.set_menu(customer_id, route)
-            return message_response.message
+            return message_response.json()
         elif type(message_response) == RedirectResponse:
             route = self.__adjust_route(message_response.route, menu)
             self.__user_state.set_menu(customer_id, route)
             return self.process_message(message)
+        elif type(message_response) == EndChatResponse:
+            self.__user_state.delete_menu(customer_id)
+            return message_response.json()
+        elif type(message_response) == TransferToHuman:
+            self.__user_state.delete_menu(customer_id)
+            return message_response.json()
         else:
             raise ChatbotError('Tipo de retorno inválido!')
 

chatgraph/messages/test_message_consumer.py

@@ -0,0 +1,17 @@
+from typing import Any, Callable
+from .base_message_consumer import MessageConsumer
+
+
+class TestMessageConsumer(MessageConsumer):
+    """
+    Classe de consumidor de mensagens de teste para uso em testes unitários.
+    """
+    def __init__(self) -> None:
+        super().__init__()
+        
+    @classmethod
+    def load_dotenv(cls) -> 'TestMessageConsumer':
+        return cls()
+    
+    def start_consume(self, process_message: Callable) -> Any:
+        return super().start_consume(process_message)

chatgraph/types/output_state.py

@@ -1,4 +1,5 @@
 from typing import Union
+import json
 
 messageTypes = Union[str, float, int, None]
 
@@ -22,6 +23,15 @@ class ChatbotResponse:
         """
         self.message = message
         self.route = route
+    
+    def json(self):
+        '''
+        Retorna o objeto em formato json.
+        '''
+        return json.dumps({
+            'type': 'message',
+            'message': self.message,
+        })
 
 
 class RedirectResponse:
@@ -40,3 +50,50 @@ class RedirectResponse:
             route (str): A rota para a qual o chatbot deve redirecionar.
         """
         self.route = route
+
+class EndChatResponse:
+    """
+    Representa uma resposta que indica o fim do chatbot.
+
+    Atributos:
+        tabulation_id (str): O ID da tabulação do chatbot.
+        observations (str): As observações finais do chatbot.
+    """
+
+    def __init__(self, tabulation_id: str, observations:str) -> None:
+        '''
+        Finzaliza e tabula as informações do chatbot.
+        '''
+        self.tabulation_id = tabulation_id
+        self.observations = observations
+        
+    def json(self):
+        '''
+        Retorna o objeto em formato json.
+        '''
+        return json.dumps({
+            'type': 'tabulate',
+            'tabulation_id': self.tabulation_id,
+            'observations': self.observations,
+        })
+        
+class TransferToHuman:
+    """
+    Representa uma transferencia para um atendente humano.
+    """
+    def __init__(self, campaign_id: str, observations:str) -> None:
+        '''
+        Finzaliza e tabula as informações do chatbot.
+        '''
+        self.campaign_id = campaign_id
+        self.observations = observations
+        
+    def json(self):
+        '''
+        Retorna o objeto em formato json.
+        '''
+        return json.dumps({
+            'type': 'transfer',
+            'campaign_id': self.campaign_id,
+            'observations': self.observations,
+        })

chatgraph/types/user_state.py

@@ -31,6 +31,16 @@ class UserState(ABC):
             menu (str): O menu a ser definido para o cliente.
         """
         pass
+    
+    @abstractmethod
+    def delete_menu(self, customer_id: str) -> None:
+        """
+        Deleta o menu atual para o ID de cliente fornecido.
+
+        Args:
+            customer_id (str): O ID do cliente.
+        """
+        pass
 
 
 class SimpleUserState(UserState):
@@ -72,3 +82,13 @@ class SimpleUserState(UserState):
         """
         if menu:
             self.states[customer_id] = menu.lower()
+
+    def delete_menu(self, customer_id: str) -> None:
+        """
+        Deleta o menu atual para o ID de cliente fornecido.
+
+        Args:
+            customer_id (str): O ID do cliente.
+        """
+        if customer_id in self.states:
+            self.states.pop(customer_id)

{chatgraph-0.1.3.dist-info → chatgraph-0.1.5.dist-info}/METADATA

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

{chatgraph-0.1.3.dist-info → chatgraph-0.1.5.dist-info}/RECORD

@@ -1,16 +1,17 @@
-chatgraph/__init__.py,sha256=qPA9Vw2SZivqVUlB19D-YrRuequMThR6kgO2XsZcwDc,602
+chatgraph/__init__.py,sha256=oG_aiCUfJfhiNPuJMuDZv0yFwfTxUMWmzvwWJ6AEhWg,684
 chatgraph/auth/credentials.py,sha256=xsMEpLQnc66novPjL6upocMcnUnvFJ7yxINzUenkxmc,2388
-chatgraph/bot/chatbot_model.py,sha256=pyxLKv4PACv6ovAMllS6wd1oqILdOvlRMQ4hR_pjwXQ,6296
+chatgraph/bot/chatbot_model.py,sha256=6tPoplS9nPHMYK0gSKFGq6cQr3dA2jOiBJTmQVuOb0I,6643
 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=OSdTT4dHIzawLDOCZ-4hZ06T8UBxoJIosqvXl7gxpM0,1099
 chatgraph/messages/rabbitMQ_message_consumer.py,sha256=mRcJ1Hbd5wPFxC3v6Tj3pYvjd9yAc398Ys6cUox-CTY,7005
+chatgraph/messages/test_message_consumer.py,sha256=9XIkbCHd1S6S8pINRT-SLEvUT0TQWBCsdPhYN6PpZ2s,518
 chatgraph/types/message_types.py,sha256=NBuebmpO2e9ilspBoFgs9O3rY_VQ2AP_Z8Q_nC70nyM,965
-chatgraph/types/output_state.py,sha256=ZditFyycd6CEvemrW9c7mThQDP1l6tI_8p19Jgj8pbs,1471
+chatgraph/types/output_state.py,sha256=fpJ_hGO4aZ3wr6DV4i9ctnkHytWhK6ZDpKpJ3vSwRyY,3064
 chatgraph/types/route.py,sha256=nKTqzwGl7d_Bu8G6Sr0EmmhuuiZWKEoSozITRrdyi1g,2587
-chatgraph/types/user_state.py,sha256=3YkZQ_FfAVrWZqFWLFaNdVOGKYThp-Lip-BbldM35Gc,2321
-chatgraph-0.1.3.dist-info/LICENSE,sha256=rVJozpRzDlplOpvI8A1GvmfVS0ReYdZvMWc1j2jV0v8,1090
-chatgraph-0.1.3.dist-info/METADATA,sha256=1fDFbBxSMJlEdhb6aPNJpeOHri7QS-SAek-UoJsViy8,5829
-chatgraph-0.1.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-chatgraph-0.1.3.dist-info/RECORD,,
+chatgraph/types/user_state.py,sha256=B17bN300RW_9-WUbI7pWK_BUCvMbdpm6gLQHPqdGly4,2858
+chatgraph-0.1.5.dist-info/LICENSE,sha256=rVJozpRzDlplOpvI8A1GvmfVS0ReYdZvMWc1j2jV0v8,1090
+chatgraph-0.1.5.dist-info/METADATA,sha256=ZoNfC0TwLhUOtvAhs4TDdQCWjY5Wo9U6oJDPb-xzkYw,5829
+chatgraph-0.1.5.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+chatgraph-0.1.5.dist-info/RECORD,,