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

chatgraph/__init__.py

@@ -1,21 +1,22 @@
 from .auth.credentials import Credential
 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, UserState
+from .messages.message_consumer import MessageConsumer
+from .types.message_types import UserCall, UserState, Element
 from .types.output_state import ChatbotResponse, RedirectResponse, EndChatResponse, TransferToHuman
 from .types.route import Route
 
 __all__ = [
     'ChatbotApp',
     'Credential',
-    'Message',
+    'UserCall',
     'ChatbotRouter',
     'ChatbotResponse',
     'RedirectResponse',
-    'RabbitMessageConsumer',
+    'MessageConsumer',
     'Route',
     'EndChatResponse',
     'TransferToHuman',
     'UserState',
+    'Element',
 ]

chatgraph/bot/chatbot_model.py

@@ -1,29 +1,32 @@
 import inspect
-from abc import ABC
 from functools import wraps
 from logging import debug
 import json
+from logging import error
 
 from ..error.chatbot_error import ChatbotError, ChatbotMessageError
-from ..messages.base_message_consumer import MessageConsumer
-from ..types.message_types import Message
+from ..messages.message_consumer import MessageConsumer
+from ..types.message_types import UserCall
 from ..types.output_state import ChatbotResponse, RedirectResponse, EndChatResponse, TransferToHuman
 from ..types.route import Route
 from .chatbot_router import ChatbotRouter
 
 
-class ChatbotApp(ABC):
+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 = {}
     
@@ -101,12 +104,12 @@ class ChatbotApp(ABC):
         self.__message_consumer.reprer()
         self.__message_consumer.start_consume(self.process_message)
     
-    def process_message(self, message: Message):
+    def process_message(self, userCall: UserCall):
         """
         Processa uma mensagem recebida, identificando a rota correspondente e executando a função associada.
 
         Args:
-            message (Message): A mensagem a ser processada.
+            userCall (Message): A mensagem a ser processada.
 
         Raises:
             ChatbotMessageError: Se nenhuma rota for encontrada para o menu atual do usuário.
@@ -115,60 +118,51 @@ class ChatbotApp(ABC):
         Returns:
             str: A resposta gerada pela função da rota, que pode ser uma mensagem ou o resultado de uma redireção.
         """
-        customer_id = message.user_state.customer_id
-        user_state = message.user_state
-        menu = message.user_state.menu
-        menu = menu.lower()
-        handler = self.__routes.get(menu, None)
+        customer_id = userCall.customer_id
+        route = userCall.route.lower()
+        menu = userCall.menu.lower()
+        obs = userCall.obs
+        handler = self.__routes.get(route, None)
 
         if not handler:
             raise ChatbotMessageError(
-                customer_id, f'Rota não encontrada para {menu}!'
+                customer_id, f'Rota não encontrada para {route}!'
             )
+            
         func = handler['function']
-        message_name = handler['params'].get(Message, None)
+        userCall_name = handler['params'].get(UserCall, None)
         route_state_name = handler['params'].get(Route, None)
 
         kwargs = dict()
-        if message_name:
-            kwargs[message_name] = message
+        if userCall_name:
+            kwargs[userCall_name] = userCall
         if route_state_name:
-            kwargs[route_state_name] = Route(menu, list(self.__routes.keys()))
+            kwargs[route_state_name] = Route(route, list(self.__routes.keys()))
 
-        message_response = func(**kwargs)
+        userCall_response = func(**kwargs)
+
+        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)
+            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)
 
-        if type(message_response) in (str, float, int):
-            response = ChatbotResponse(message_response)
-            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()
-            response['user_state'] = {
-                'customer_id': customer_id,
-                'menu': route,
-                'obs': user_state.obs,
-            }
-            return json.dumps(response)            
-        elif type(message_response) in (EndChatResponse, TransferToHuman):
-            response = message_response.json()
-            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)
-            message.user_state.menu = route
-            return self.process_message(message)
         else:
-            raise ChatbotError('Tipo de retorno inválido!')
+            error('Tipo de retorno inválido!')
+            return None
+        
 
     def __adjust_route(self, route: str, absolute_route: str) -> str:
         """
@@ -187,4 +181,30 @@ class ChatbotApp(ABC):
         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/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/gRPC/gRPCCall.py

@@ -0,0 +1,125 @@
+import os
+import grpc
+import chatgraph.pb.userstate_pb2 as userstate_pb2
+import chatgraph.pb.userstate_pb2_grpc as userstate_pb2_grpc
+
+import chatgraph.pb.voll_pb2 as whatsapp_pb2
+import chatgraph.pb.voll_pb2_grpc as whatsapp_pb2_grpc
+
+class WhatsappServiceClient:
+    def __init__(self, grpc_uri=None):
+        
+        self.grpc_uri = grpc_uri
+        
+        if not grpc_uri:
+            self.grpc_uri = os.getenv('GRPC_URI')
+        
+        if not self.grpc_uri:
+            raise ValueError("A variável de ambiente 'GRPC_URI' não está definida.")
+        
+        # Cria o canal gRPC
+        self.channel = grpc.insecure_channel(self.grpc_uri)
+        
+        # Cria o stub (client) para o serviço gRPC
+        self.stub = whatsapp_pb2_grpc.MessageServiceStub(self.channel)
+
+    def send_button(self, message_data):
+        # Cria o request para o método SendButton
+        request = whatsapp_pb2.MessageRequest(**message_data)
+
+        # Faz a chamada ao serviço gRPC
+        try:
+            response = self.stub.SendButton(request)
+            return response
+        except grpc.RpcError as e:
+            print(f"Erro ao fazer a requisição gRPC SendButton: {e}")
+            return None
+
+    def send_list(self, message_data):
+        # Cria o request para o método SendList
+        request = whatsapp_pb2.MessageRequest(**message_data)
+
+        # Faz a chamada ao serviço gRPC
+        try:
+            response = self.stub.SendList(request)
+            return response
+        except grpc.RpcError as e:
+            print(f"Erro ao fazer a requisição gRPC SendList: {e}")
+            return None
+
+    def send_text(self, message_data):
+        # Cria o request para o método SendText
+        request = whatsapp_pb2.MessageRequest(**message_data)
+
+        # Faz a chamada ao serviço gRPC
+        try:
+            response = self.stub.SendText(request)
+            return response
+        except grpc.RpcError as e:
+            print(f"Erro ao fazer a requisição gRPC SendText: {e}")
+            return None
+
+class UserStateServiceClient:
+    def __init__(self, grpc_uri=None):
+        
+        self.grpc_uri = grpc_uri
+        
+        if not grpc_uri:
+            self.grpc_uri = os.getenv('GRPC_URI')
+        
+        if not self.grpc_uri:
+            raise ValueError("A variável de ambiente 'GRPC_URI' não está definida.")
+        
+        # Cria o canal gRPC
+        self.channel = grpc.insecure_channel(self.grpc_uri)
+        
+        # Cria o stub (client) para o serviço gRPC
+        self.stub = userstate_pb2_grpc.UserStateServiceStub(self.channel)
+
+    def select_user_state(self, user_id):
+        # Cria o request para o método SelectUserState
+        request = userstate_pb2.UserStateId(user_id=user_id)
+
+        # Faz a chamada ao serviço gRPC
+        try:
+            response = self.stub.SelectUserState(request)
+            return response
+        except grpc.RpcError as e:
+            print(f"Erro ao fazer a requisição gRPC SelectUserState: {e}")
+            return None
+
+    def insert_user_state(self, user_state_data):
+        # Cria o request para o método InsertUserState
+        request = userstate_pb2.UserState(**user_state_data)
+
+        # Faz a chamada ao serviço gRPC
+        try:
+            response = self.stub.InsertUserState(request)
+            return response
+        except grpc.RpcError as e:
+            print(f"Erro ao fazer a requisição gRPC InsertUserState: {e}")
+            return None
+
+    def update_user_state(self, user_state_data):
+        # Cria o request para o método UpdateUserState
+        request = userstate_pb2.UserState(**user_state_data)
+
+        # Faz a chamada ao serviço gRPC
+        try:
+            response = self.stub.UpdateUserState(request)
+            return response
+        except grpc.RpcError as e:
+            print(f"Erro ao fazer a requisição gRPC UpdateUserState: {e}")
+            return None
+
+    def delete_user_state(self, user_id):
+        # Cria o request para o método DeleteUserState
+        request = userstate_pb2.UserStateId(user_id=user_id)
+
+        # Faz a chamada ao serviço gRPC
+        try:
+            response = self.stub.DeleteUserState(request)
+            return response
+        except grpc.RpcError as e:
+            print(f"Erro ao fazer a requisição gRPC DeleteUserState: {e}")
+            return None

chatgraph/messages/{rabbitMQ_message_consumer.py → message_consumer.py}

@@ -4,14 +4,13 @@ import os
 import pika
 from typing import Callable
 from ..auth.credentials import Credential
-from ..types.message_types import Message, UserState
-from .base_message_consumer import MessageConsumer
+from ..types.message_types import UserCall, UserState
 from rich.console import Console
 from rich.table import Table
 from rich.text import Text
 from rich.panel import Panel
 
-class RabbitMessageConsumer(MessageConsumer):
+class MessageConsumer:
     """
     Implementação de MessageConsumer para consumir mensagens de uma fila RabbitMQ.
 
@@ -27,6 +26,7 @@ class RabbitMessageConsumer(MessageConsumer):
         self,
         credential: Credential,
         amqp_url: str,
+        grpc_uri: str,
         queue_consume: str,
         prefetch_count: int = 1,
         virtual_host: str = '/',
@@ -45,6 +45,7 @@ class RabbitMessageConsumer(MessageConsumer):
         self.__prefetch_count = prefetch_count
         self.__queue_consume = queue_consume
         self.__amqp_url = amqp_url
+        self.__grpc_uri = grpc_uri
         self.__credentials = pika.PlainCredentials(
             credential.username, credential.password
         )
@@ -58,7 +59,8 @@ class RabbitMessageConsumer(MessageConsumer):
         queue_env: str = 'RABBIT_QUEUE',
         prefetch_env: str = 'RABBIT_PREFETCH',
         vhost_env: str = 'RABBIT_VHOST',
-    ) -> 'RabbitMessageConsumer':
+        grpc_uri: str = 'GRPC_URI',
+    ) -> 'MessageConsumer':
         """
         Carrega as configurações do RabbitMQ a partir de variáveis de ambiente e retorna uma instância de RabbitMessageConsumer.
 
@@ -82,8 +84,9 @@ class RabbitMessageConsumer(MessageConsumer):
         queue = os.getenv(queue_env)
         prefetch = os.getenv(prefetch_env, 1)
         vhost = os.getenv(vhost_env, '/')
+        grpc = os.getenv(grpc_uri)
 
-        if not username or not password or not url or not queue:
+        if not username or not password or not url or not queue or not grpc:
             raise ValueError('Corrija as variáveis de ambiente!')
 
         return cls(
@@ -92,6 +95,7 @@ class RabbitMessageConsumer(MessageConsumer):
             queue_consume=queue,
             prefetch_count=int(prefetch),
             virtual_host=vhost,
+            grpc_uri=grpc,
         )
 
     def start_consume(self, process_message: Callable) -> None:
@@ -122,7 +126,7 @@ class RabbitMessageConsumer(MessageConsumer):
                 ),
             )
 
-            info('[x] Aguardando solicitações RPC')
+            info('[x] Server inicializado! Aguardando solicitações RPC')
             channel.start_consuming()
         except pika.exceptions.StreamLostError as e:
             debug(e)
@@ -154,7 +158,7 @@ class RabbitMessageConsumer(MessageConsumer):
         )
         ch.basic_ack(delivery_tag=method.delivery_tag)
 
-    def __transform_message(self, message: dict) -> Message:
+    def __transform_message(self, message: dict) -> UserCall:
         """
         Transforma o dicionário JSON recebido em uma instância de Message.
 
@@ -166,12 +170,13 @@ class RabbitMessageConsumer(MessageConsumer):
         """
         
         user_state = message.get('user_state', {})
-        return Message(
+        return UserCall(
             type=message.get('type', ''),
             text=message.get('text', ''),
             user_state=UserState(
                 customer_id=user_state.get('customer_id', ''),
                 menu=user_state.get('menu', ''),
+                route=user_state.get('route', ''),
                 lst_update=user_state.get('lst_update', ''),
                 obs=user_state.get('obs', {}),
             ),
@@ -179,6 +184,7 @@ class RabbitMessageConsumer(MessageConsumer):
             customer_phone=message.get('customer_phone', ''),
             company_phone=message.get('company_phone', ''),
             status=message.get('status'),
+            grpc_uri=self.__grpc_uri,
         )
 
     def reprer(self):

chatgraph/pb/userstate.proto

@@ -0,0 +1,29 @@
+syntax = "proto3";
+
+package userstate;
+
+option go_package = "./pb/userstate";
+
+service UserStateService {
+    rpc SelectUserState(UserStateId) returns (UserState);
+    rpc InsertUserState(UserState) returns (RequestStatus);
+    rpc UpdateUserState(UserState) returns (RequestStatus);
+    rpc DeleteUserState(UserStateId) returns (RequestStatus);
+}
+
+message UserState {
+    string user_id = 1;
+    string menu_id = 2;
+    string route = 3;
+    string obs = 4;
+    string date = 5;
+}
+
+message UserStateId {
+    string user_id = 1;
+}
+
+message RequestStatus {
+    bool status = 1;
+    string message = 2;
+}

chatgraph/pb/userstate_pb2.py

@@ -0,0 +1,43 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: userstate.proto
+# Protobuf Python Version: 5.27.2
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    5,
+    27,
+    2,
+    '',
+    'userstate.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0fuserstate.proto\x12\tuserstate\"W\n\tUserState\x12\x0f\n\x07user_id\x18\x01 \x01(\t\x12\x0f\n\x07menu_id\x18\x02 \x01(\t\x12\r\n\x05route\x18\x03 \x01(\t\x12\x0b\n\x03obs\x18\x04 \x01(\t\x12\x0c\n\x04\x64\x61te\x18\x05 \x01(\t\"\x1e\n\x0bUserStateId\x12\x0f\n\x07user_id\x18\x01 \x01(\t\"0\n\rRequestStatus\x12\x0e\n\x06status\x18\x01 \x01(\x08\x12\x0f\n\x07message\x18\x02 \x01(\t2\x9e\x02\n\x10UserStateService\x12?\n\x0fSelectUserState\x12\x16.userstate.UserStateId\x1a\x14.userstate.UserState\x12\x41\n\x0fInsertUserState\x12\x14.userstate.UserState\x1a\x18.userstate.RequestStatus\x12\x41\n\x0fUpdateUserState\x12\x14.userstate.UserState\x1a\x18.userstate.RequestStatus\x12\x43\n\x0f\x44\x65leteUserState\x12\x16.userstate.UserStateId\x1a\x18.userstate.RequestStatusB\x10Z\x0e./pb/userstateb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'userstate_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'Z\016./pb/userstate'
+  _globals['_USERSTATE']._serialized_start=30
+  _globals['_USERSTATE']._serialized_end=117
+  _globals['_USERSTATEID']._serialized_start=119
+  _globals['_USERSTATEID']._serialized_end=149
+  _globals['_REQUESTSTATUS']._serialized_start=151
+  _globals['_REQUESTSTATUS']._serialized_end=199
+  _globals['_USERSTATESERVICE']._serialized_start=202
+  _globals['_USERSTATESERVICE']._serialized_end=488
+# @@protoc_insertion_point(module_scope)