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

chatgraph/pb/userstate_pb2_grpc.py

@@ -0,0 +1,226 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+import warnings
+
+import chatgraph.pb.userstate_pb2 as userstate__pb2
+
+GRPC_GENERATED_VERSION = '1.66.2'
+GRPC_VERSION = grpc.__version__
+_version_not_supported = False
+
+try:
+    from grpc._utilities import first_version_is_lower
+    _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
+except ImportError:
+    _version_not_supported = True
+
+if _version_not_supported:
+    raise RuntimeError(
+        f'The grpc package installed is at version {GRPC_VERSION},'
+        + f' but the generated code in userstate_pb2_grpc.py depends on'
+        + f' grpcio>={GRPC_GENERATED_VERSION}.'
+        + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
+        + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
+    )
+
+
+class UserStateServiceStub(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.SelectUserState = channel.unary_unary(
+                '/userstate.UserStateService/SelectUserState',
+                request_serializer=userstate__pb2.UserStateId.SerializeToString,
+                response_deserializer=userstate__pb2.UserState.FromString,
+                _registered_method=True)
+        self.InsertUserState = channel.unary_unary(
+                '/userstate.UserStateService/InsertUserState',
+                request_serializer=userstate__pb2.UserState.SerializeToString,
+                response_deserializer=userstate__pb2.RequestStatus.FromString,
+                _registered_method=True)
+        self.UpdateUserState = channel.unary_unary(
+                '/userstate.UserStateService/UpdateUserState',
+                request_serializer=userstate__pb2.UserState.SerializeToString,
+                response_deserializer=userstate__pb2.RequestStatus.FromString,
+                _registered_method=True)
+        self.DeleteUserState = channel.unary_unary(
+                '/userstate.UserStateService/DeleteUserState',
+                request_serializer=userstate__pb2.UserStateId.SerializeToString,
+                response_deserializer=userstate__pb2.RequestStatus.FromString,
+                _registered_method=True)
+
+
+class UserStateServiceServicer(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def SelectUserState(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def InsertUserState(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def UpdateUserState(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def DeleteUserState(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_UserStateServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'SelectUserState': grpc.unary_unary_rpc_method_handler(
+                    servicer.SelectUserState,
+                    request_deserializer=userstate__pb2.UserStateId.FromString,
+                    response_serializer=userstate__pb2.UserState.SerializeToString,
+            ),
+            'InsertUserState': grpc.unary_unary_rpc_method_handler(
+                    servicer.InsertUserState,
+                    request_deserializer=userstate__pb2.UserState.FromString,
+                    response_serializer=userstate__pb2.RequestStatus.SerializeToString,
+            ),
+            'UpdateUserState': grpc.unary_unary_rpc_method_handler(
+                    servicer.UpdateUserState,
+                    request_deserializer=userstate__pb2.UserState.FromString,
+                    response_serializer=userstate__pb2.RequestStatus.SerializeToString,
+            ),
+            'DeleteUserState': grpc.unary_unary_rpc_method_handler(
+                    servicer.DeleteUserState,
+                    request_deserializer=userstate__pb2.UserStateId.FromString,
+                    response_serializer=userstate__pb2.RequestStatus.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'userstate.UserStateService', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+    server.add_registered_method_handlers('userstate.UserStateService', rpc_method_handlers)
+
+
+ # This class is part of an EXPERIMENTAL API.
+class UserStateService(object):
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def SelectUserState(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/userstate.UserStateService/SelectUserState',
+            userstate__pb2.UserStateId.SerializeToString,
+            userstate__pb2.UserState.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def InsertUserState(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/userstate.UserStateService/InsertUserState',
+            userstate__pb2.UserState.SerializeToString,
+            userstate__pb2.RequestStatus.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def UpdateUserState(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/userstate.UserStateService/UpdateUserState',
+            userstate__pb2.UserState.SerializeToString,
+            userstate__pb2.RequestStatus.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def DeleteUserState(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/userstate.UserStateService/DeleteUserState',
+            userstate__pb2.UserStateId.SerializeToString,
+            userstate__pb2.RequestStatus.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)

chatgraph/pb/voll.proto

@@ -0,0 +1,32 @@
+syntax = "proto3";
+
+package messagevoll;
+
+option go_package = "./pb/messagevoll";
+
+service MessageService {
+  rpc SendButton(MessageRequest) returns (MessageResponse);
+  rpc SendList(MessageRequest) returns (MessageResponse);
+  rpc SendText(MessageRequest) returns (MessageResponse);
+}
+
+message MessageRequest {
+  string hook_id = 1;
+  string enterprise_id = 2;
+  string unique_customer_id = 3;
+  string message_title = 4;
+  string message_text = 5;
+  string message_caption = 6;
+  string button_title = 7;
+  repeated QuickOption options = 8;
+}
+
+message MessageResponse {
+  bool status = 1;
+  string message = 2;
+}
+
+message QuickOption {
+  string title = 1;
+  string description = 2;
+}

chatgraph/pb/voll_pb2.py

@@ -0,0 +1,43 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: voll.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,
+    '',
+    'voll.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\nvoll.proto\x12\x0bmessagevoll\"\xdb\x01\n\x0eMessageRequest\x12\x0f\n\x07hook_id\x18\x01 \x01(\t\x12\x15\n\renterprise_id\x18\x02 \x01(\t\x12\x1a\n\x12unique_customer_id\x18\x03 \x01(\t\x12\x15\n\rmessage_title\x18\x04 \x01(\t\x12\x14\n\x0cmessage_text\x18\x05 \x01(\t\x12\x17\n\x0fmessage_caption\x18\x06 \x01(\t\x12\x14\n\x0c\x62utton_title\x18\x07 \x01(\t\x12)\n\x07options\x18\x08 \x03(\x0b\x32\x18.messagevoll.QuickOption\"2\n\x0fMessageResponse\x12\x0e\n\x06status\x18\x01 \x01(\x08\x12\x0f\n\x07message\x18\x02 \x01(\t\"1\n\x0bQuickOption\x12\r\n\x05title\x18\x01 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x02 \x01(\t2\xe7\x01\n\x0eMessageService\x12G\n\nSendButton\x12\x1b.messagevoll.MessageRequest\x1a\x1c.messagevoll.MessageResponse\x12\x45\n\x08SendList\x12\x1b.messagevoll.MessageRequest\x1a\x1c.messagevoll.MessageResponse\x12\x45\n\x08SendText\x12\x1b.messagevoll.MessageRequest\x1a\x1c.messagevoll.MessageResponseB\x12Z\x10./pb/messagevollb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'voll_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  _globals['DESCRIPTOR']._loaded_options = None
+  _globals['DESCRIPTOR']._serialized_options = b'Z\020./pb/messagevoll'
+  _globals['_MESSAGEREQUEST']._serialized_start=28
+  _globals['_MESSAGEREQUEST']._serialized_end=247
+  _globals['_MESSAGERESPONSE']._serialized_start=249
+  _globals['_MESSAGERESPONSE']._serialized_end=299
+  _globals['_QUICKOPTION']._serialized_start=301
+  _globals['_QUICKOPTION']._serialized_end=350
+  _globals['_MESSAGESERVICE']._serialized_start=353
+  _globals['_MESSAGESERVICE']._serialized_end=584
+# @@protoc_insertion_point(module_scope)

chatgraph/pb/voll_pb2_grpc.py

@@ -0,0 +1,183 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+import warnings
+
+import chatgraph.pb.voll_pb2 as voll__pb2
+
+GRPC_GENERATED_VERSION = '1.66.2'
+GRPC_VERSION = grpc.__version__
+_version_not_supported = False
+
+try:
+    from grpc._utilities import first_version_is_lower
+    _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
+except ImportError:
+    _version_not_supported = True
+
+if _version_not_supported:
+    raise RuntimeError(
+        f'The grpc package installed is at version {GRPC_VERSION},'
+        + f' but the generated code in voll_pb2_grpc.py depends on'
+        + f' grpcio>={GRPC_GENERATED_VERSION}.'
+        + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
+        + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
+    )
+
+
+class MessageServiceStub(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.SendButton = channel.unary_unary(
+                '/messagevoll.MessageService/SendButton',
+                request_serializer=voll__pb2.MessageRequest.SerializeToString,
+                response_deserializer=voll__pb2.MessageResponse.FromString,
+                _registered_method=True)
+        self.SendList = channel.unary_unary(
+                '/messagevoll.MessageService/SendList',
+                request_serializer=voll__pb2.MessageRequest.SerializeToString,
+                response_deserializer=voll__pb2.MessageResponse.FromString,
+                _registered_method=True)
+        self.SendText = channel.unary_unary(
+                '/messagevoll.MessageService/SendText',
+                request_serializer=voll__pb2.MessageRequest.SerializeToString,
+                response_deserializer=voll__pb2.MessageResponse.FromString,
+                _registered_method=True)
+
+
+class MessageServiceServicer(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def SendButton(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def SendList(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def SendText(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_MessageServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'SendButton': grpc.unary_unary_rpc_method_handler(
+                    servicer.SendButton,
+                    request_deserializer=voll__pb2.MessageRequest.FromString,
+                    response_serializer=voll__pb2.MessageResponse.SerializeToString,
+            ),
+            'SendList': grpc.unary_unary_rpc_method_handler(
+                    servicer.SendList,
+                    request_deserializer=voll__pb2.MessageRequest.FromString,
+                    response_serializer=voll__pb2.MessageResponse.SerializeToString,
+            ),
+            'SendText': grpc.unary_unary_rpc_method_handler(
+                    servicer.SendText,
+                    request_deserializer=voll__pb2.MessageRequest.FromString,
+                    response_serializer=voll__pb2.MessageResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'messagevoll.MessageService', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+    server.add_registered_method_handlers('messagevoll.MessageService', rpc_method_handlers)
+
+
+ # This class is part of an EXPERIMENTAL API.
+class MessageService(object):
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def SendButton(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/messagevoll.MessageService/SendButton',
+            voll__pb2.MessageRequest.SerializeToString,
+            voll__pb2.MessageResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def SendList(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/messagevoll.MessageService/SendList',
+            voll__pb2.MessageRequest.SerializeToString,
+            voll__pb2.MessageResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def SendText(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/messagevoll.MessageService/SendText',
+            voll__pb2.MessageRequest.SerializeToString,
+            voll__pb2.MessageResponse.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)

chatgraph/types/message_types.py

@@ -1,6 +1,8 @@
+from ..gRPC.gRPCCall import WhatsappServiceClient, UserStateServiceClient
+
 from dataclasses import dataclass
 from typing import Optional
-
+import json
 
 @dataclass
 class UserState:
@@ -15,11 +17,24 @@ class UserState:
     """
     customer_id: str
     menu: str
+    route: str
     lst_update: str
     obs: Optional[dict] = None
 
+
 @dataclass
-class Message:
+class Element:
+    """
+    Representa um elemento de uma lista de opções.
+    
+    Atributos:
+        title (str): O título do elemento.
+        description (str): A descrição do elemento.
+    """
+    title: str
+    description: Optional[str] = None
+
+class UserCall:
     """
     Representa uma mensagem recebida ou enviada pelo chatbot.
 
@@ -32,10 +47,168 @@ class Message:
         company_phone (str): O número de telefone da empresa que está enviando ou recebendo a mensagem.
         status (Optional[str]): O status da mensagem (por exemplo, enviada, recebida, lida, etc.). Este campo é opcional.
     """
-    type: str
-    text: str
-    user_state: UserState
-    channel: str
-    customer_phone: str
-    company_phone: str
-    status: Optional[str] = None
+    def __init__(
+        self,
+        type: str,
+        text: str,
+        user_state: UserState,
+        channel: str,
+        customer_phone: str,
+        company_phone: str,
+        grpc_uri: str,
+        status: Optional[str] = None,
+    ) -> None:
+        
+        self.type = type
+        self.text = text
+        self.__user_state = user_state
+        self.channel = channel
+        self.customer_phone = customer_phone
+        self.company_phone = company_phone
+        self.status = status
+        
+        self.grpc_uri = grpc_uri
+        
+        self.__wpp_server_client = WhatsappServiceClient(self.grpc_uri)
+        self.__user_state_client = UserStateServiceClient(self.grpc_uri)
+
+    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(
+            {
+                "hook_id": self.company_phone,
+                "enterprise_id": self.customer_phone,
+                "unique_customer_id": self.__user_state.customer_id,
+                "message_text": text
+            }
+        )
+
+        if not response.status:
+            raise ValueError("Erro ao enviar mensagem de texto.")
+    
+    def send_button(
+        self, 
+        text:str, 
+        buttons:list, 
+        title: str|None = None,
+        caption: str|None = None,
+        ) -> None:
+        if len(buttons) > 3:
+            raise ValueError("O número máximo de botões é 3.")
+        
+        response = self.__wpp_server_client.send_button(
+            {
+                "hook_id": self.company_phone,
+                "enterprise_id": self.customer_phone,
+                "unique_customer_id": self.__user_state.customer_id,
+                "message_text": text,
+                "button_title": title,
+                "message_caption": caption,
+                "message_title": title,
+                "options": [{"title": b} for b in buttons],
+            }
+        )
+
+        if not response.status:
+            raise ValueError("Erro ao enviar mensagem de botões.")
+
+    def send_list(
+        self, 
+        text:str,
+        title: str|None = None,
+        button_title: str|None = None,
+        element_list: list[Element] = None,
+        caption: str|None = None,
+        ) -> None:
+        
+        if len(element_list) > 20:
+            raise ValueError("O número máximo de elementos é 20.")
+        
+        response = self.__wpp_server_client.send_list(
+            {
+                "hook_id": self.company_phone,
+                "enterprise_id": self.customer_phone,
+                "unique_customer_id": self.__user_state.customer_id,
+                "message_text": text,
+                "button_title": button_title,
+                "message_caption": caption,
+                "message_title": title,
+                "options": [{"title": e.title, "description": e.description} for e in element_list],
+            }
+        )
+
+        if not response.status:
+            raise ValueError("Erro ao enviar mensagem de lista.")
+    
+    def delete_user_state(self) -> None:
+        response = self.__user_state_client.delete_user_state(self.__user_state.customer_id)
+
+        if not response.status:
+            raise ValueError("Erro ao deletar estado do usuário.")
+
+    def update_user_state(
+        self, 
+        menu: str,
+        route: str,
+        obs: dict,
+        ) -> None:
+        
+        response = self.__user_state_client.update_user_state({
+            "user_id": self.__user_state.customer_id,
+            "menu_id": menu,
+            "route": route,
+            "obs": json.dumps(obs),
+        })
+
+        if not response.status:
+            raise ValueError("Erro ao atualizar estado do usuário.")
+        
+        self.__user_state.menu = menu
+        self.__user_state.route = route
+        self.__user_state.obs = obs
+    
+    @property
+    def menu(self):
+        return self.__user_state.menu
+    
+    @property
+    def route(self):
+        return self.__user_state.route
+    
+    @property
+    def obs(self):
+        return self.__user_state.obs
+    
+    @property
+    def customer_id(self):
+        return self.__user_state.customer_id
+    
+    @menu.setter
+    def menu(self, menu):
+        
+        self.update_user_state(
+            menu, 
+            self.__user_state.route, 
+            self.__user_state.obs
+        )
+    
+    @route.setter
+    def route(self, route):
+        
+        self.update_user_state(
+            self.__user_state.menu, 
+            route, 
+            self.__user_state.obs
+        )
+    
+    @obs.setter
+    def obs(self, obs):
+        
+        self.update_user_state(
+            self.__user_state.menu, 
+            self.__user_state.route, 
+            obs
+        )
+