chatgraph 0.3.16__py3-none-any.whl → 0.4.1__py3-none-any.whl

chatgraph/bot/chatbot_model.py

@@ -31,7 +31,7 @@ class ChatbotApp:
         self.__message_consumer = message_consumer
         self.__routes = {}
     
-    def include_router(self, router: ChatbotRouter, prefix: str):
+    def include_router(self, router: ChatbotRouter):
         """
         Inclui um roteador de chatbot com um prefixo nas rotas da aplicação.
 
@@ -42,18 +42,7 @@ class ChatbotApp:
         Raises:
             ChatbotError: Se a rota 'start' não for encontrada no roteador.
         """
-        if 'start' not in router.routes.keys():
-            raise ChatbotError('Erro ao incluir rota, start não encontrado!')
-
-        prefixed_routes = {
-            (
-                f'start{prefix.lower()}'
-                if key.lower() == 'start'
-                else f'start{prefix.lower()}{key.lower().replace("start", "")}'
-            ): value
-            for key, value in router.routes.items()
-        }
-        self.__routes.update(prefixed_routes)
+        self.__routes.update(router.routes)
 
     def route(self, route_name: str):
         """
@@ -67,9 +56,6 @@ class ChatbotApp:
         """
         route_name = route_name.strip().lower()
 
-        if 'start' not in route_name:
-            route_name = f'start{route_name}'
-
         def decorator(func):
             params = dict()
             signature = inspect.signature(func)
@@ -121,9 +107,10 @@ class ChatbotApp:
         """
         customer_id = userCall.customer_id
         route = userCall.route.lower()
+        route_handler = route.split('.')[-1]
         menu = userCall.menu.lower()
         obs = userCall.obs
-        handler = self.__routes.get(route, None)
+        handler = self.__routes.get(route_handler, None)
 
         if not handler:
             raise ChatbotMessageError(
@@ -172,7 +159,7 @@ class ChatbotApp:
             return
 
         elif isinstance(userCall_response, RedirectResponse):
-            route = self.__adjust_route(userCall_response.route, route)
+            route = route + '.' + userCall_response.route
             userCall.route = route
             return self.process_message(userCall)
 

chatgraph/bot/chatbot_router.py

@@ -18,7 +18,17 @@ class ChatbotRouter:
         """
         Inicializa a classe ChatbotRouter com um dicionário vazio de rotas.
         """
-        self.routes = {}
+        self.__routes = {}
+    
+    @property
+    def routes(self):
+        """
+        Retorna as rotas registradas no roteador.
+
+        Returns:
+            dict: Um dicionário contendo as rotas e funções associadas.
+        """
+        return self.__routes
 
     def route(self, route_name: str):
         """
@@ -30,15 +40,14 @@ class ChatbotRouter:
         Returns:
             function: O decorador que adiciona a função à rota especificada.
         """
-        if 'start' not in route_name:
-            route_name = f'start{route_name}'
+
+        route_name = route_name.strip().lower()
 
         def decorator(func):
             params = dict()
             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
@@ -48,8 +57,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()] = {
+            self.__routes[route_name] = {
                 'function': func,
                 'params': params,
                 'return': output_param,
@@ -63,7 +71,7 @@ class ChatbotRouter:
 
         return decorator
 
-    def include_router(self, router: 'ChatbotRouter', prefix: str):
+    def include_router(self, router: 'ChatbotRouter'):
         """
         Inclui outro roteador com um prefixo nas rotas do roteador atual.
 
@@ -74,16 +82,4 @@ class ChatbotRouter:
         Raises:
             ChatbotError: Se a rota 'start' não for encontrada no roteador fornecido.
         """
-        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()}'
-                if key.lower() == 'start'
-                else f'start{prefix.lower()}{key.lower().replace("start", "")}'
-            ): value
-            for key, value in router.routes.items()
-        }
-        self.routes.update(prefixed_routes)
+        self.__routes.update(router.__routes)

chatgraph/messages/message_consumer.py

@@ -178,6 +178,7 @@ class MessageConsumer:
                 voll_id=user_state.get('voll_id', ''),
                 platform=user_state.get('platform', ''),
                 obs=obs,
+                protocol=user_state.get('protocol', ''),
             ),
             channel=message.get('channel', ''),
             customer_phone=message.get('customer_phone', ''),

chatgraph/types/request_types.py

@@ -26,6 +26,7 @@ class UserState:
         voll_id: Optional[str]=None,
         lst_update: Optional[str]=None,
         obs: Optional[dict] = None,
+        protocol: Optional[str] = None,
     ) -> None:
         
         self.customer_id = customer_id
@@ -36,6 +37,7 @@ class UserState:
         self.direction_in = direction_in
         self.voll_id = voll_id
         self.platform = platform
+        self.protocol = protocol
 
     def __str__(self):
         return f"UserState:\n\tcustomer_id={self.customer_id},\n\tmenu={self.menu},\n\troute={self.route},\n\tlst_update={self.lst_update},\n\tobs={self.obs},\n\tdirection_in={self.direction_in}"
@@ -53,6 +55,7 @@ class UserState:
             'direction': self.direction_in,
             'voll_id': self.voll_id,
             'platform': self.platform,
+            'protocol': self.protocol,
         })
     
     def update(self, grpc_uri: Optional[str] = None) -> None:
@@ -318,6 +321,26 @@ class UserCall:
     def customer_id(self):
         return self.__user_state.customer_id
     
+    @property
+    def voll_id(self):
+        return self.__user_state.voll_id
+    
+    @property
+    def platform(self):
+        return self.__user_state.platform
+    
+    @property
+    def direction_in(self):
+        return self.__user_state.direction_in
+    
+    @property
+    def lst_update(self):
+        return self.__user_state.lst_update
+    
+    @property
+    def protocol(self):
+        return self.__user_state.protocol
+    
     @menu.setter
     def menu(self, menu):
         

chatgraph/types/route.py

@@ -30,6 +30,13 @@ class Route:
         """
         return self.get_previous()
     
+    @property
+    def current_node(self)->str:
+        """
+        Retorna o nó atual do usuário
+        """
+        return self.current.split(self.separator)[-1]
+    
     def get_previous(self) -> 'Route':
         """
         Retorna o caminho anterior ao caminho atual.
@@ -41,7 +48,7 @@ class Route:
             str: O caminho anterior à rota atual.
         """
         if self.current == 'start':
-            raise RouteError('Não há caminho anterior ao start')
+            return Route(self.current, self.routes, self.separator)
 
         previous_route = self.separator.join(self.current.split(self.separator)[:-1])
         return Route(previous_route, self.routes, self.separator)
@@ -60,9 +67,9 @@ class Route:
             Route: O próximo caminho montado.
         """
         next_part = next_part.strip().lower()
-        next_route = f"{self.current.rstrip(self.separator)}{next_part}"
-        if next_route not in self.routes:
-            raise RouteError(f'Rota não encontrada: {next_route}')
+        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}')
         
         return Route(next_route, self.routes, self.separator)
 

{chatgraph-0.3.16.dist-info → chatgraph-0.4.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: chatgraph
-Version: 0.3.16
+Version: 0.4.1
 Summary: A user-friendly chatbot library
 Home-page: https://github.com/irissonnlima/chatgraph
 License: MIT
@@ -13,6 +13,8 @@ Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: grpcio (>=1.67.0,<2.0.0)
 Requires-Dist: grpcio-tools (>=1.67.0,<2.0.0)
+Requires-Dist: matplotlib (>=3.10.0,<4.0.0)
+Requires-Dist: networkx (>=3.4.2,<4.0.0)
 Requires-Dist: pika (>=1.3.2,<2.0.0)
 Requires-Dist: python-dotenv (>=1.0.1,<2.0.0)
 Requires-Dist: rich (>=13.8.1,<14.0.0)

{chatgraph-0.3.16.dist-info → chatgraph-0.4.1.dist-info}/RECORD

@@ -1,12 +1,12 @@
 chatgraph/__init__.py,sha256=lUMcFMSxKQeHVFq7CyZV2r_UOHAtGGpeXtac0RUd7po,718
 chatgraph/auth/credentials.py,sha256=xsMEpLQnc66novPjL6upocMcnUnvFJ7yxINzUenkxmc,2388
-chatgraph/bot/chatbot_model.py,sha256=uqnIqsNre2KGM4hJhUfUk3oRcapcVsabgLwx7cx9Ids,7324
-chatgraph/bot/chatbot_router.py,sha256=tl-4WP9wqUjCIwvtO9kJVUFjN2-mxNDUTRAFL9PXBgI,3028
+chatgraph/bot/chatbot_model.py,sha256=8KvMDst5ZZDs4l3zw2AeuyBM74UqyPFznHe-Ajbb8Yw,6834
+chatgraph/bot/chatbot_router.py,sha256=bVbm9dBoC2OesetXlQJpCrkaiLM3GUro0GrOyCSgreI,2586
 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=jYZg7CvtKWhtv2sQMrSunSmvnnL83MKmNVO8GWmGZ9U,7696
-chatgraph/messages/message_consumer.py,sha256=DYPZA-TTXe4jtcdRZUSfPf-QO0B_aXUzSO3FSwfhWlo,8996
+chatgraph/messages/message_consumer.py,sha256=B6ettgby39r1pUw6KyiUlpzXPy5nhFAJiye54Lckoqs,9054
 chatgraph/pb/userstate.proto,sha256=GgjD1K2t8VxrMLkceVnY043xBdXWTPxBzaBhAbAg7yw,859
 chatgraph/pb/userstate_pb2.py,sha256=aCS_Muk_uA78buV28g7yi0O2vriTJedT4HKxgzFFUVw,2856
 chatgraph/pb/userstate_pb2_grpc.py,sha256=QJEGEmdBL6bkxrTI-gNhGbZpPUFuP5jm7eFu8y21PMI,10340
@@ -15,10 +15,10 @@ chatgraph/pb/voll_pb2.py,sha256=WLwE7F1pe_elcnwRnSFqVDcFihLZ3JBq4ScTHNCqWGA,5328
 chatgraph/pb/voll_pb2_grpc.py,sha256=U-_nXTAIEzo2GMYjHiphbSyTx3zyipsnKebFuZuoU1k,17685
 chatgraph/types/end_types.py,sha256=--Ty2gM_y7J-yRAvZV26e4HMUpoguAMAhfOIS9-kQTk,1316
 chatgraph/types/message_types.py,sha256=l488K4sVhTdOY5bbk47Y-qjfvfyn4h7bpmOFf2xGods,3334
-chatgraph/types/request_types.py,sha256=Rju__bOLI3AG-l5aYTNv_SwqH_-pu1tNiiHA9PujN_M,11761
-chatgraph/types/route.py,sha256=sg0oAsfPlPciDsVa-GAr1AwwypH5EyUKqfo_IYMLOtU,2801
-chatgraph-0.3.16.dist-info/entry_points.txt,sha256=bO9_Q-PqE5vCNNo6ke_-3j2gHfKQMDGnKDTkNuCdBuA,48
-chatgraph-0.3.16.dist-info/LICENSE,sha256=rVJozpRzDlplOpvI8A1GvmfVS0ReYdZvMWc1j2jV0v8,1090
-chatgraph-0.3.16.dist-info/METADATA,sha256=L9uQXpMFqD75L9xUHxTDxSCwZ15t0waU-rWY-g50Q08,12782
-chatgraph-0.3.16.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-chatgraph-0.3.16.dist-info/RECORD,,
+chatgraph/types/request_types.py,sha256=uBnqr1KrwgehQ14_RYd82FxxWy5CBNjApyNDYWUBeKE,12331
+chatgraph/types/route.py,sha256=FO5INy5UXgicuQ8FKEZKcPv6WS5YH10dPx2OaPu_0sE,2978
+chatgraph-0.4.1.dist-info/entry_points.txt,sha256=bO9_Q-PqE5vCNNo6ke_-3j2gHfKQMDGnKDTkNuCdBuA,48
+chatgraph-0.4.1.dist-info/LICENSE,sha256=rVJozpRzDlplOpvI8A1GvmfVS0ReYdZvMWc1j2jV0v8,1090
+chatgraph-0.4.1.dist-info/METADATA,sha256=jjujZIbrgTRF86EPPGBEbtnx0tOgX1zS64WZDSh5I0Y,12866
+chatgraph-0.4.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+chatgraph-0.4.1.dist-info/RECORD,,