chatgraph 0.6.1__py3-none-any.whl → 0.6.2__py3-none-any.whl

chatgraph/bot/chatbot_model.py

@@ -2,6 +2,7 @@ import inspect
 from functools import wraps
 from logging import debug, error
 import asyncio
+import re
 
 from ..error.chatbot_error import ChatbotMessageError
 from ..messages.message_consumer import MessageConsumer
@@ -16,6 +17,11 @@ from ..types.end_types import (
 from ..types.route import Route
 from .chatbot_router import ChatbotRouter
 from ..types.background_task import BackgroundTask
+from .default_functions import voltar
+
+DEFAULT_FUNCTION: dict[str, callable] = {
+    r"^\s*(voltar)\s*$": voltar,
+}
 
 
 class ChatbotApp:
@@ -23,16 +29,22 @@ 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 = None):
+    def __init__(
+        self,
+        message_consumer: MessageConsumer = None,
+        default_functions: dict[str, callable] = DEFAULT_FUNCTION,
+    ):
         """
         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.
+            default_functions (dict[str, callable]): Dicionário de funções padrão que podem ser usadas antes das rotas.
         """
         if not message_consumer:
             message_consumer = MessageConsumer.load_dotenv()
 
+        self.default_functions = default_functions
         self.__message_consumer = message_consumer
         self.__routes = {}
 
@@ -106,7 +118,20 @@ class ChatbotApp:
         route = userCall.route.lower()
         route_handler = route.split(".")[-1]
 
-        handler = self.__routes.get(route_handler, None)
+        matchDefault = False
+
+        for regex, func in self.default_functions.items():
+            if re.match(regex, userCall.content_message):
+                matchDefault = True
+                debug(f"Função padrão encontrada: {func.__name__} para a rota {route}")
+                handler = {
+                    "function": func,
+                    "params": {UserCall: "userCall", Route: "route"},
+                }
+                break
+
+        if not matchDefault:
+            handler = self.__routes.get(route_handler, None)
 
         if not handler:
             raise ChatbotMessageError(user_id, f"Rota não encontrada para {route}!")
@@ -127,6 +152,9 @@ class ChatbotApp:
             loop = asyncio.get_running_loop()
             userCall_response = await loop.run_in_executor(None, lambda: func(**kwargs))
 
+        if matchDefault:
+            userCall.content_message = ""
+
         if isinstance(userCall_response, (list, tuple)):
             for response in userCall_response:
                 await self.__process_func_response(response, userCall, route=route)
@@ -166,6 +194,7 @@ class ChatbotApp:
                 userCall.end_chat,
                 userCall_response.observations,
                 userCall_response.tabulation_id,
+                userCall_response.tabulation_name,
             )
             return
 
@@ -175,6 +204,7 @@ class ChatbotApp:
                 userCall.transfer_to_human,
                 userCall_response.observations,
                 userCall_response.campaign_id,
+                userCall_response.campaign_name,
             )
             return
 

chatgraph/bot/default_functions.py

@@ -0,0 +1,24 @@
+from ..types.request_types import UserCall
+from ..types.message_types import Message, Button
+from ..types.end_types import (
+    RedirectResponse,
+    EndChatResponse,
+    TransferToHuman,
+    TransferToMenu,
+)
+from ..types.route import Route
+
+
+async def voltar(route: Route, userCall: UserCall) -> tuple:
+    """
+    Função para voltar à rota anterior.
+    Args:
+        route (Route): A rota atual do chatbot.
+        usercall (UserCall): O objeto UserCall associado à chamada do usuário.
+    """
+
+    previous = route.get_previous()
+    userCall.console.print(
+        f"Voltando rota. ({route.current}) -> ({previous.current})", style="bold yellow"
+    )
+    return RedirectResponse(previous.current_node)

chatgraph/gRPC/gRPCCall.py

@@ -1,6 +1,7 @@
 import os
 import grpc
 import json
+from rich.console import Console
 
 import chatgraph.pb.router_pb2 as chatbot_pb2
 import chatgraph.pb.router_pb2_grpc as chatbot_pb2_grpc
@@ -22,15 +23,22 @@ class RouterServiceClient:
         self.transfer_stub = chatbot_pb2_grpc.TransferStub(self.channel)
         self.end_chat_stub = chatbot_pb2_grpc.EndChatStub(self.channel)
 
+        self.console = Console()
+
     def insert_update_user_state(self, user_state_data):
         request = chatbot_pb2.UserState(**user_state_data)
         try:
             response = self.user_state_stub.InsertUpdateUserState(request)
             if not response.status:
-                print(f"Erro ao chamar SendMessage: {response.message}")
+                self.console.print(
+                    f"Erro ao chamar InsertUpdateUserState: {response.message}",
+                    style="bold red",
+                )
             return response
         except grpc.RpcError as e:
-            print(f"Erro ao chamar InsertUpdateUserState: {e}")
+            self.console.print(
+                f"Erro ao chamar InsertUpdateUserState: {e}", style="bold red"
+            )
             return None
 
     def delete_user_state(self, chat_id_data):
@@ -38,10 +46,12 @@ class RouterServiceClient:
         try:
             response = self.user_state_stub.DeleteUserState(request)
             if not response.status:
-                print(f"Erro ao chamar SendMessage: {response.message}")
+                self.console.print(
+                    f"Erro ao chamar SendMessage: {response.message}", style="bold red"
+                )
             return response
         except grpc.RpcError as e:
-            print(f"Erro ao chamar DeleteUserState: {e}")
+            self.console.print(f"Erro ao chamar DeleteUserState: {e}", style="bold red")
             return None
 
     def get_user_state(self, chat_id_data):
@@ -50,7 +60,7 @@ class RouterServiceClient:
             response = self.user_state_stub.GetUserState(request)
             return response
         except grpc.RpcError as e:
-            print(f"Erro ao chamar GetUserState: {e}")
+            self.console.print(f"Erro ao chamar GetUserState: {e}", style="bold red")
             return None
 
     def send_message(self, message_data):
@@ -61,10 +71,12 @@ class RouterServiceClient:
         try:
             response = self.send_message_stub.SendMessage(request)
             if not response.status:
-                print(f"Erro ao chamar SendMessage: {response.message}")
+                self.console.print(
+                    f"Erro ao chamar SendMessage: {response.message}", style="bold red"
+                )
             return response
         except grpc.RpcError as e:
-            print(f"Erro ao chamar SendMessage: {e}")
+            self.console.print(f"Erro ao chamar SendMessage: {e}", style="bold red")
             return None
 
     def send_image(self, message_data):
@@ -75,12 +87,14 @@ class RouterServiceClient:
         try:
             response = self.send_message_stub.SendImage(request)
             if not response.status and response.message != "arquivo não encontrado":
-                print(f"Erro ao chamar SendImage: {response.message}")
+                self.console.print(
+                    f"Erro ao chamar SendImage: {response.message}", style="bold red"
+                )
             elif response.message == "arquivo não encontrado":
                 print("Arquivo não encontrado, Carregando arquivo...")
             return response
         except grpc.RpcError as e:
-            print(f"Erro ao chamar SendImage: {e}")
+            self.console.print(f"Erro ao chamar SendImage: {e}", style="bold red")
             return None
 
     def upload_file(self, file_data):
@@ -88,10 +102,12 @@ class RouterServiceClient:
         try:
             response = self.send_message_stub.UploadFile(request)
             if not response.status:
-                print(f"Erro ao chamar UploadFile: {response.message}")
+                self.console.print(
+                    f"Erro ao chamar UploadFile: {response.message}", style="bold red"
+                )
             return response
         except grpc.RpcError as e:
-            print(f"Erro ao chamar UploadFile: {e}")
+            self.console.print(f"Erro ao chamar UploadFile: {e}", style="bold red")
             return None
 
     def transfer_to_human(self, transfer_request_data):
@@ -99,10 +115,12 @@ class RouterServiceClient:
         try:
             response = self.transfer_stub.TransferToHuman(request)
             if not response.status:
-                print(f"Erro ao chamar SendMessage: {response.message}")
+                self.console.print(
+                    f"Erro ao chamar SendMessage: {response.message}", style="bold red"
+                )
             return response
         except grpc.RpcError as e:
-            print(f"Erro ao chamar TransferToHuman: {e}")
+            self.console.print(f"Erro ao chamar TransferToHuman: {e}", style="bold red")
             return None
 
     def transfer_to_menu(self, transfer_request_data):
@@ -110,10 +128,13 @@ class RouterServiceClient:
         try:
             response = self.transfer_stub.TransferToMenu(request)
             if not response.status:
-                print(f"Erro ao chamar TransferToMenu: {response.message}")
+                self.console.print(
+                    f"Erro ao chamar TransferToMenu: {response.message}",
+                    style="bold red",
+                )
             return response
         except grpc.RpcError as e:
-            print(f"Erro ao chamar TransferToMenu: {e}")
+            self.console.print(f"Erro ao chamar TransferToMenu: {e}", style="bold red")
             return None
 
     def end_chat(self, end_chat_request_data):
@@ -121,10 +142,12 @@ class RouterServiceClient:
         try:
             response = self.end_chat_stub.EndChat(request)
             if not response.status:
-                print(f"Erro ao chamar SendMessage: {response.message}")
+                self.console.print(
+                    f"Erro ao chamar SendMessage: {response.message}", style="bold red"
+                )
             return response
         except grpc.RpcError as e:
-            print(f"Erro ao chamar EndChat: {e}")
+            self.console.print(f"Erro ao chamar EndChat: {e}", style="bold red")
             return None
 
     def get_campaign_id(self, campaign_name):
@@ -133,7 +156,7 @@ class RouterServiceClient:
             response = self.transfer_stub.GetCampaignID(request)
             return response
         except grpc.RpcError as e:
-            print(f"Erro ao chamar GetCampaignID: {e}")
+            self.console.print(f"Erro ao chamar GetCampaignID: {e}", style="bold red")
             return None
 
     def get_all_campaigns(self):
@@ -142,7 +165,7 @@ class RouterServiceClient:
             response = self.transfer_stub.GetAllCampaigns(request)
             return response
         except grpc.RpcError as e:
-            print(f"Erro ao chamar GetAllCampaigns: {e}")
+            self.console.print(f"Erro ao chamar GetAllCampaigns: {e}", style="bold red")
             return None
 
     def get_tabulation_id(self, tabulation_name):
@@ -151,7 +174,7 @@ class RouterServiceClient:
             response = self.end_chat_stub.GetTabulationID(request)
             return response
         except grpc.RpcError as e:
-            print(f"Erro ao chamar GetTabulationID: {e}")
+            self.console.print(f"Erro ao chamar GetTabulationID: {e}", style="bold red")
             return None
 
     def get_all_tabulations(self):
@@ -160,5 +183,7 @@ class RouterServiceClient:
             response = self.end_chat_stub.GetAllTabulations(request)
             return response
         except grpc.RpcError as e:
-            print(f"Erro ao chamar GetAllTabulations: {e}")
+            self.console.print(
+                f"Erro ao chamar GetAllTabulations: {e}", style="bold red"
+            )
             return None

chatgraph/types/end_types.py

@@ -25,11 +25,20 @@ class EndChatResponse:
         observations (str): As observações finais do chatbot.
     """
 
-    def __init__(self, tabulation_id: str, observations: str) -> None:
+    def __init__(
+        self,
+        tabulation_id: str,
+        tabulation_name: str | None = None,
+        observations: str | None = None,
+    ) -> None:
         """
         Finzaliza e tabula as informações do chatbot.
         """
+        if not tabulation_id and not tabulation_name:
+            raise ValueError("tabulation_id or tabulation_name must be provided.")
+
         self.tabulation_id = tabulation_id
+        self.tabulation_name = tabulation_name
         self.observations = observations
 
 
@@ -38,11 +47,17 @@ class TransferToHuman:
     Representa uma transferencia para um atendente humano.
     """
 
-    def __init__(self, campaign_id: str, observations: str) -> None:
+    def __init__(
+        self,
+        campaign_id: str,
+        campaign_name: str | None = None,
+        observations: str | None = None,
+    ) -> None:
         """
         Finzaliza e tabula as informações do chatbot.
         """
         self.campaign_id = campaign_id
+        self.campaign_name = campaign_name
         self.observations = observations
 
 

chatgraph/types/request_types.py

@@ -4,6 +4,7 @@ from chatgraph.types.message_types import Message, Button, MessageTypes, message
 from typing import Optional
 from datetime import datetime
 import json, os
+from rich.console import Console
 
 
 class ChatID:
@@ -155,6 +156,7 @@ class UserCall:
         self.__grpc_uri = grpc_uri
 
         self.__router_client = RouterServiceClient(self.__grpc_uri)
+        self.console = Console()
 
     def __str__(self):
         return f"UserCall({self.type}, {self.__type_message}, {self.__content_message}, {self.__user_state})"
@@ -220,14 +222,20 @@ class UserCall:
         if not response.status:
             raise ValueError("Erro ao enviar mensagem de botões.")
 
-    def transfer_to_human(self, message: str, campaign_name: str) -> None:
+    def transfer_to_human(
+        self, message: str, campaign_id: str = None, campaign_name: str = None
+    ) -> None:
 
-        campaign = self.__router_client.get_campaign_id({"name": campaign_name})
+        if campaign_id is None and campaign_name is None:
+            raise ValueError("Você deve informar o ID ou o nome da campanha.")
+        if not campaign_id:
+            campaign = self.__router_client.get_campaign_id({"name": campaign_name})
+            campaign_id = campaign.id
 
         response = self.__router_client.transfer_to_human(
             {
                 "chat_id": self.__user_state.chatID.to_dict(),
-                "campaign_id": campaign.id,
+                "campaign_id": campaign_id,
                 "observation": message,
             }
         )
@@ -248,13 +256,19 @@ class UserCall:
         if not response.status:
             raise ValueError("Erro ao transferir chat para menu.")
 
-    def end_chat(self, message: str, tabulation_name: str) -> None:
-        tabulation = self.__router_client.get_tabulation_id({"name": tabulation_name})
+    def end_chat(self, message: str, tabulation_id: str, tabulation_name: str) -> None:
+        if tabulation_id is None and tabulation_name is None:
+            raise ValueError("Você deve informar o ID ou o nome da tabulação.")
+        if not tabulation_id:
+            tabulation = self.__router_client.get_tabulation_id(
+                {"name": tabulation_name}
+            )
+            tabulation_id = tabulation.id
 
         response = self.__router_client.end_chat(
             {
                 "chat_id": self.__user_state.chatID.to_dict(),
-                "tabulation_id": tabulation.id,
+                "tabulation_id": tabulation_id,
                 "observation": message,
             }
         )
@@ -340,3 +354,7 @@ class UserCall:
         self.update_user_state(
             self.__user_state.menu, self.__user_state.route, observation
         )
+
+    @content_message.setter
+    def content_message(self, content_message: str):
+        self.__content_message = content_message

chatgraph/types/route.py

@@ -10,7 +10,7 @@ class Route:
         routes (list[str]): A lista de todas as rotas disponíveis no fluxo.
     """
 
-    def __init__(self, current: str, routes: list[str], separator:str='.'):
+    def __init__(self, current: str, routes: list[str], separator: str = "."):
         """
         Inicializa a rota com a rota atual e a lista de rotas disponíveis.
 
@@ -24,20 +24,20 @@ class Route:
         self.separator = separator
 
     @property
-    def previous(self)->'Route':
+    def previous(self) -> "Route":
         """
         Retorna a rota anterior a atual do usuário
         """
         return self.get_previous()
-    
+
     @property
-    def current_node(self)->str:
+    def current_node(self) -> str:
         """
         Retorna o nó atual do usuário
         """
         return self.current.split(self.separator)[-1]
-    
-    def get_previous(self) -> 'Route':
+
+    def get_previous(self) -> "Route":
         """
         Retorna o caminho anterior ao caminho atual.
 
@@ -47,13 +47,18 @@ class Route:
         Returns:
             str: O caminho anterior à rota atual.
         """
-        if self.current == 'start':
+        if self.current == "start":
             return Route(self.current, self.routes, self.separator)
 
-        previous_route = self.separator.join(self.current.split(self.separator)[:-1])
+        rotas_dedup = self.separator.join(
+            dict.fromkeys(self.current.split(self.separator))
+        )
+
+        previous_route = self.separator.join(rotas_dedup.split(self.separator)[:-1])
+
         return Route(previous_route, self.routes, self.separator)
 
-    def get_next(self, next_part: str) -> 'Route':
+    def get_next(self, next_part: str) -> "Route":
         """
         Monta e retorna o próximo caminho com base na parte fornecida.
 
@@ -69,8 +74,8 @@ class Route:
         next_part = next_part.strip().lower()
         next_route = f"{self.current.rstrip(self.separator)}.{next_part}"
         if next_part not in self.routes:
-            raise RouteError(f'Rota não encontrada: {next_part}')
-        
+            raise RouteError(f"Rota não encontrada: {next_part}")
+
         return Route(next_route, self.routes, self.separator)
 
     def __str__(self):
@@ -80,7 +85,7 @@ class Route:
         Returns:
             str: A representação em string da rota atual.
         """
-        return f'Route(current={self.current})'
+        return f"Route(current={self.current})"
 
     def __repr__(self):
         """

{chatgraph-0.6.1.dist-info → chatgraph-0.6.2.dist-info}/METADATA

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

{chatgraph-0.6.1.dist-info → chatgraph-0.6.2.dist-info}/RECORD

@@ -1,23 +1,24 @@
 chatgraph/__init__.py,sha256=cQk87BOUYKvOjgCkS5Vexy9EYeLOeuQn6vR1vXx1Akc,935
 chatgraph/auth/credentials.py,sha256=xsMEpLQnc66novPjL6upocMcnUnvFJ7yxINzUenkxmc,2388
-chatgraph/bot/chatbot_model.py,sha256=wLVXehmHEWItRwvl6J8HM96lC69FXmqsNG8KqplBboI,7373
+chatgraph/bot/chatbot_model.py,sha256=kI990THIgwcwRm69XcfSqJkUAH1Khbm5OrfDnKxMZ8o,8446
 chatgraph/bot/chatbot_router.py,sha256=bVbm9dBoC2OesetXlQJpCrkaiLM3GUro0GrOyCSgreI,2586
+chatgraph/bot/default_functions.py,sha256=OSgo3NGxT4RwGe5ohpGksaARSDSkNInLmJsBuZAuv1Y,739
 chatgraph/cli/__init__.py,sha256=tfgYhGoKy1nD4STN4xDh6J4VV55RICk7v1GZmzAg-bM,7413
 chatgraph/error/chatbot_error.py,sha256=4sEcW8vz0eQk2QDmDygucVM4caHliZW5iH7XJvmLBuk,1897
 chatgraph/error/route_error.py,sha256=CY8m82ap7-Sr-DXPsolltRW50TqD__5RyXBmNNJCWr8,793
-chatgraph/gRPC/gRPCCall.py,sha256=KXLhxfX7Mrm48BVuWhTA4xw4T4vghXnljvE37FAxmfQ,6402
+chatgraph/gRPC/gRPCCall.py,sha256=jMWPbHobyR6n942BARsQSlVmbh6dk0qUpfVgSNlGOvg,7561
 chatgraph/messages/message_consumer.py,sha256=yHhfLYjtmXQYG44o7SNO6XUaeM4-jH8BpIpPHI-mt8Y,6116
 chatgraph/pb/router.proto,sha256=0FDhMe7GrabeQz7QRJN8PN2mcuRjA5YrGt6TC9GWBak,3488
 chatgraph/pb/router_pb2.py,sha256=eSo0ncYDSG3ryIfWNFzqmG3anx-VmQ2nGXBy1gPnXmQ,7962
 chatgraph/pb/router_pb2_grpc.py,sha256=iaQQsYPc8Tihuf4sWEI48DhvhtCYtDqFBHhwRtE57ps,29886
 chatgraph/types/background_task.py,sha256=j-Bpd1l_LCDiWukhfG8wNr1F0_GUaOL_J2_BCUnTof4,928
-chatgraph/types/end_types.py,sha256=ulVq3XybRs-EGcOxw6e1FF6U2p3nbtPfOv3uRe6oJfo,1638
+chatgraph/types/end_types.py,sha256=QMNju3wpdbCpdUTSZ0oGQx9x0wgckxmw1wfGgowqCTk,2057
 chatgraph/types/image.py,sha256=-Lf9ns8_M4jmROoFnHlZiYeScYD_GhF1pgIGLVjsRtI,3216
 chatgraph/types/message_types.py,sha256=__LOk09Xfvojum0xiq9n157_3QSSLNaDMc02iq8dGpk,3434
-chatgraph/types/request_types.py,sha256=pGpJhzk6LGSrREGsG2gC14SoS9DTpUO5w9MzmLUg7PU,10500
-chatgraph/types/route.py,sha256=FO5INy5UXgicuQ8FKEZKcPv6WS5YH10dPx2OaPu_0sE,2978
-chatgraph-0.6.1.dist-info/entry_points.txt,sha256=bO9_Q-PqE5vCNNo6ke_-3j2gHfKQMDGnKDTkNuCdBuA,48
-chatgraph-0.6.1.dist-info/LICENSE,sha256=rVJozpRzDlplOpvI8A1GvmfVS0ReYdZvMWc1j2jV0v8,1090
-chatgraph-0.6.1.dist-info/METADATA,sha256=avh_0Xu-Mfkhzr4b5v9mKZyJcd2CprrvUIMBnlWWr6g,12908
-chatgraph-0.6.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-chatgraph-0.6.1.dist-info/RECORD,,
+chatgraph/types/request_types.py,sha256=DC-x4tR5WLxcI9wzaZ0bk82eeLaTOClQbal3F1KWG2w,11240
+chatgraph/types/route.py,sha256=UUlMWje8yZ4scHp3LKXrtRXycDtudIwUpwQvi3zQ2_w,3090
+chatgraph-0.6.2.dist-info/entry_points.txt,sha256=bO9_Q-PqE5vCNNo6ke_-3j2gHfKQMDGnKDTkNuCdBuA,48
+chatgraph-0.6.2.dist-info/LICENSE,sha256=rVJozpRzDlplOpvI8A1GvmfVS0ReYdZvMWc1j2jV0v8,1090
+chatgraph-0.6.2.dist-info/METADATA,sha256=-n5XOgBx9bxthqGX6W7TncIQKqXkncARgWvKcZH28FA,12908
+chatgraph-0.6.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+chatgraph-0.6.2.dist-info/RECORD,,