chatgraph 0.3.0__tar.gz → 0.3.1__tar.gz

{chatgraph-0.3.0 → chatgraph-0.3.1}/PKG-INFO

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

{chatgraph-0.3.0 → chatgraph-0.3.1}/chatgraph/bot/chatbot_model.py

@@ -17,13 +17,16 @@ class ChatbotApp:
     Classe principal para a aplicação do chatbot, gerencia as rotas e a lógica de processamento de mensagens.
     """
 
-    def __init__(self, message_consumer: MessageConsumer):
+    def __init__(self, message_consumer: MessageConsumer = None):
         """
         Inicializa a classe ChatbotApp com um estado de usuário e um consumidor de mensagens.
 
         Args:
             message_consumer (MessageConsumer): O consumidor de mensagens que lida com a entrada de mensagens no sistema.
         """
+        if not message_consumer:
+            message_consumer = MessageConsumer.load_dotenv()
+            
         self.__message_consumer = message_consumer
         self.__routes = {}
     
@@ -125,6 +128,7 @@ class ChatbotApp:
             raise ChatbotMessageError(
                 customer_id, f'Rota não encontrada para {route}!'
             )
+            
         func = handler['function']
         userCall_name = handler['params'].get(UserCall, None)
         route_state_name = handler['params'].get(Route, None)
@@ -137,41 +141,24 @@ class ChatbotApp:
 
         userCall_response = func(**kwargs)
 
-        if type(userCall_response) in (str, float, int):
-            response = ChatbotResponse(userCall_response)
-            response = response.json()
-            response['user_state'] = {
-                'customer_id': customer_id,
-                'route': route,
-                'menu': menu,
-                'obs': obs,
-            }
-            return json.dumps(response)
-        elif type(userCall_response) == ChatbotResponse:
+        if isinstance(userCall_response, (str, float, int)):
+            return self.__create_response(userCall_response, customer_id, route, menu, obs)
+
+        elif isinstance(userCall_response, ChatbotResponse):
             adjusted_route = self.__adjust_route(userCall_response.route, route)
-            response = userCall_response.json()
-            response['user_state'] = {
-                'customer_id': customer_id,
-                'route': adjusted_route,
-                'menu': menu,
-                'obs': obs,
-            }
-            return json.dumps(response)            
-        elif type(userCall_response) in (EndChatResponse, TransferToHuman):
-            response = userCall_response.json()
-            response['user_state'] = {
-                'customer_id': customer_id,
-                'menu': None,
-                'route': None,
-                'obs': None,
-            }
-            return json.dumps(response)
-        
-        elif type(userCall_response) == RedirectResponse:
+            return self.__create_response(userCall_response.json(), customer_id, adjusted_route, menu, obs)
+
+        elif isinstance(userCall_response, (EndChatResponse, TransferToHuman)):
+            return self.__end_chat_response(userCall_response, customer_id)
+
+        elif isinstance(userCall_response, RedirectResponse):
             route = self.__adjust_route(userCall_response.route, route)
             userCall.route = route
             return self.process_message(userCall)
-        
+
+        elif not userCall_response:
+            return self.__create_response('', customer_id, None, None, None)
+
         else:
             error('Tipo de retorno inválido!')
             return None
@@ -194,4 +181,30 @@ class ChatbotApp:
         if 'start' not in route:
             route = absolute_route + route
 
-        return route
+        return route
+    
+    def __create_response(self, response, customer_id, route, menu, obs):
+        """
+        Cria uma resposta padronizada em formato JSON com o estado do usuário.
+        """
+        response = ChatbotResponse(response).json() if not isinstance(response, dict) else response
+        response['user_state'] = {
+            'customer_id': customer_id,
+            'route': route,
+            'menu': menu,
+            'obs': obs,
+        }
+        return json.dumps(response)
+
+    def __end_chat_response(self, response, customer_id):
+        """
+        Gera a resposta de finalização ou transferência de chat.
+        """
+        response = response.json()
+        response['user_state'] = {
+            'customer_id': customer_id,
+            'menu': None,
+            'route': None,
+            'obs': None,
+        }
+        return json.dumps(response)

{chatgraph-0.3.0 → chatgraph-0.3.1}/chatgraph/bot/chatbot_router.py

@@ -1,3 +1,4 @@
+
 import inspect
 from functools import wraps
 from logging import debug
@@ -37,6 +38,7 @@ class ChatbotRouter:
             signature = inspect.signature(func)
             output_param = signature.return_annotation
 
+            # Itera sobre os parâmetros da função e extrai seus tipos
             for name, param in signature.parameters.items():
                 param_type = (
                     param.annotation
@@ -46,6 +48,7 @@ class ChatbotRouter:
                 params[param_type] = name
                 debug(f'Parameter: {name}, Type: {param_type}')
 
+            # Adiciona a função e seus parâmetros à rota especificada
             self.routes[route_name.strip().lower()] = {
                 'function': func,
                 'params': params,
@@ -74,6 +77,7 @@ class ChatbotRouter:
         if 'start' not in router.routes.keys():
             raise ChatbotError('Erro ao incluir rota, start não encontrado!')
 
+        # Adiciona prefixo às rotas do roteador incluído
         prefixed_routes = {
             (
                 f'{prefix.lower()}'

{chatgraph-0.3.0 → chatgraph-0.3.1}/chatgraph/types/message_types.py

@@ -72,7 +72,9 @@ class UserCall:
         self.__wpp_server_client = WhatsappServiceClient(self.grpc_uri)
         self.__user_state_client = UserStateServiceClient(self.grpc_uri)
 
-    def send_text(self, text:str) -> None:
+    def send_text(self, text:str, abs_text:bool=False) -> None:
+        if not abs_text:
+            text = text.replace('\t', '')
         
         response = self.__wpp_server_client.send_text(
             {

{chatgraph-0.3.0 → chatgraph-0.3.1}/chatgraph/types/output_state.py

@@ -13,7 +13,12 @@ class ChatbotResponse:
         route (str, opcional): A rota para a qual o chatbot deve direcionar após esta mensagem. Padrão é None.
     """
 
-    def __init__(self, message: messageTypes = None, route: str = None) -> None:
+    def __init__(
+            self, 
+            message: messageTypes = None, 
+            route: str = None, 
+            abs_text:bool=False
+        ) -> None:
         """
         Inicializa a resposta do chatbot com uma mensagem e uma rota opcional.
 
@@ -21,6 +26,12 @@ class ChatbotResponse:
             message (messageTypes, opcional): A mensagem a ser enviada ao usuário. Pode ser uma string, um número, ou None.
             route (str, opcional): A rota para a qual o chatbot deve direcionar após esta mensagem. Padrão é None.
         """
+        if not message:
+            message = ''
+            
+        if not abs_text:
+            message = message.replace('\t', '')
+            
         self.message = message
         self.route = route
     

{chatgraph-0.3.0 → chatgraph-0.3.1}/chatgraph/types/route.py

@@ -23,6 +23,13 @@ class Route:
         self.routes = routes
         self.separator = separator
 
+    @property
+    def previous(self)->str:
+        """
+        Retorna a rota anterior a atual do usuário
+        """
+        return self.get_previous()
+    
     def get_previous(self) -> str:
         """
         Retorna o caminho anterior ao caminho atual.

{chatgraph-0.3.0 → chatgraph-0.3.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "chatgraph"
-version = "0.3.0"
+version = "0.3.1"
 description = "A user-friendly chatbot library"
 authors = ["Irisson N. Lima <irisson.lima@verdecard.com.br>"]
 readme = "README.md"