chatgraph 0.4.2__py3-none-any.whl → 0.5.0__py3-none-any.whl

chatgraph/types/request_types.py

@@ -1,103 +1,129 @@
-from chatgraph.gRPC.gRPCCall import WhatsappServiceClient, UserStateServiceClient
-from chatgraph.types.message_types import Message, Button, ListElements, messageTypes, MessageTypes
-from dataclasses import dataclass
+from chatgraph.gRPC.gRPCCall import RouterServiceClient
+from chatgraph.types.message_types import Message, Button, MessageTypes, messageTypes
 from typing import Optional
 from datetime import datetime
 import json, os
 
-@dataclass
+
+class ChatID:
+    """
+    Representa o ID de um chat.
+
+    Atributos:
+        user_id (str): O ID do usuário.
+        company_id (str): O ID da empresa
+    """
+
+    def __init__(
+        self,
+        user_id: str,
+        company_id: str,
+    ):
+        self.user_id = user_id
+        self.company_id = company_id
+
+    def __str__(self):
+        return f"ChatID({self.user_id}, {self.company_id})"
+
+    def to_dict(self) -> dict:
+        return {
+            "user_id": self.user_id,
+            "company_id": self.company_id,
+        }
+
+
 class UserState:
     """
     Representa o estado de um usuário.
-    
+
     Atributos:
-        customer_id (str): O ID do cliente.
+        chatID (ChatID): O ID do chat.
         menu (str): O menu atual.
-        lst_update (str): A última atualização do menu.
-        obs (dict): Observações adicionais sobre o estado do usuário.
+        route (str): A rota atual.
+        protocol (str): O protocolo atual.
+        observations (dict): Observações do estado do usuário.
     """
+
     def __init__(
         self,
-        customer_id: str,
+        chatID: ChatID,
         menu: str,
         route: str,
-        platform: str,
-        direction_in: bool,
-        voll_id: Optional[str]=None,
-        lst_update: Optional[str]=None,
-        obs: Optional[dict] = None,
-        protocol: Optional[str] = None,
+        protocol: str,
+        observation: dict,
     ) -> None:
-        
-        self.customer_id = customer_id
+
+        self.chatID = chatID
         self.menu = menu
         self.route = route
-        self.lst_update = lst_update
-        self.obs = obs
-        self.direction_in = direction_in
-        self.voll_id = voll_id
-        self.platform = platform
+        self.observation = observation
         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}"
+        return f"UserState({self.chatID}, {self.menu}, {self.route}, {self.protocol}, {self.observation})"
+
+    def __chat_id_to_dict(self) -> dict:
+        return {
+            "user_id": self.chatID.user_id,
+            "company_id": self.chatID.company_id,
+        }
+
+    def __user_state_to_dict(self) -> dict:
+        return {
+            "chat_id": {
+                "user_id": self.chatID.user_id,
+                "company_id": self.chatID.company_id,
+            },
+            "menu": self.menu,
+            "route": self.route,
+            "protocol": self.protocol,
+            "observation": json.dumps(self.observation),
+        }
 
     def insert(self, grpc_uri: Optional[str] = None) -> None:
         if grpc_uri is None:
-            grpc_uri = os.getenv('GRPC_URI')
-        user_state_client = UserStateServiceClient(grpc_uri)
-        
-        user_state_client.insert_user_state({
-            'user_id': self.customer_id,
-            'menu_id': self.menu,
-            'route': self.route,
-            'obs': json.dumps(self.obs),
-            'direction': self.direction_in,
-            'voll_id': self.voll_id,
-            'platform': self.platform,
-        })
-    
-    def update(self, grpc_uri: Optional[str] = None) -> None:
-        if grpc_uri is None:
-            grpc_uri = os.getenv('GRPC_URI')
-        user_state_client = UserStateServiceClient(grpc_uri)
-        
-        user_state_client.update_user_state({
-            'user_id': self.customer_id,
-            'menu_id': self.menu,
-            'route': self.route,
-            'obs': json.dumps(self.obs),
-        })
-    
+            grpc_uri = os.getenv("GRPC_URI")
+        router_client = RouterServiceClient(grpc_uri)
+
+        ustate = self.__user_state_to_dict()
+        result = router_client.insert_update_user_state(ustate)
+        return result
+
     def delete(self, grpc_uri: Optional[str] = None) -> None:
         if grpc_uri is None:
-            grpc_uri = os.getenv('GRPC_URI')
-        user_state_client = UserStateServiceClient(grpc_uri)
-        
-        user_state_client.delete_user_state(self.customer_id)
+            grpc_uri = os.getenv("GRPC_URI")
+        router_client = RouterServiceClient(grpc_uri)
+
+        ustate = self.__user_state_to_dict()
+        response = router_client.delete_user_state(ustate["chat_id"])
+        return response
 
     @classmethod
-    def get_user_state(cls, customer_id: str, grpc_uri: Optional[str] = None) -> 'UserState':
+    def get_user_state(
+        cls, user_id: str, company_id: str, grpc_uri: Optional[str] = None
+    ) -> "UserState":
         if grpc_uri is None:
-            grpc_uri = os.getenv('GRPC_URI')
-        user_state_client = UserStateServiceClient(grpc_uri)
-        
-        response = user_state_client.select_user_state(user_id=customer_id)
-        
+            grpc_uri = os.getenv("GRPC_URI")
+        router_client = RouterServiceClient(grpc_uri)
+
+        chat_id = {"user_id": user_id, "company_id": company_id}
+        response = router_client.get_user_state(chat_id)
+
         if not response:
             raise ValueError("Erro ao buscar estado do usuário.")
-        
-        date = datetime.strptime(response.date, "%Y-%m-%d %H:%M:%S")
+
         return cls(
-            customer_id=response.user_id,
-            menu=response.menu_id,
+            chatID=ChatID(
+                user_id=response.chat_id.user_id,
+                company_id=response.chat_id.company_id,
+            ),
+            menu=response.menu,
             route=response.route,
-            obs=json.loads(response.obs),
-            direction_in=response.direction,
-            voll_id=response.voll_id,
-            platform=response.platform,
-            lst_update=date,
+            protocol=response.protocol,
+            observation=json.loads(response.observation),
         )
+
+
 class UserCall:
     """
     Representa uma mensagem recebida ou enviada pelo chatbot.
@@ -111,32 +137,31 @@ class UserCall:
         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.
     """
+
     def __init__(
         self,
-        type: str,
-        text: str,
         user_state: UserState,
-        channel: str,
-        customer_phone: str,
-        company_phone: str,
+        type_message: str,
+        content_message: str,
         grpc_uri: str,
-        status: Optional[str] = None,
     ) -> None:
-        
+
         self.type = type
-        self.text = text
+        self.__type_message = type_message
+        self.__content_message = content_message
         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(self, message: messageTypes|Message|Button|ListElements) -> None:
+        self.__grpc_uri = grpc_uri
+
+        self.__router_client = RouterServiceClient(self.__grpc_uri)
+
+    def __str__(self):
+        return f"UserCall({self.type}, {self.__type_message}, {self.__content_message}, {self.__user_state})"
+    
+    def send(
+        self,
+        message: messageTypes | Message,
+    ) -> None:
         """
         Envia uma mensagem ao cliente.
 
@@ -144,226 +169,126 @@ class UserCall:
             message (Message|Button|ListElements): A mensagem a ser enviada.
         """
         if isinstance(message, MessageTypes):
-            message = Message(message)
-        
-        if isinstance(message, Message):
-            self.__send_text(message.text, message.absolute_text)
-            
-        elif isinstance(message, Button):
-            self.__send_button(
-                message.text, 
-                message.buttons, 
-                message.title, 
-                message.caption
-            )
-            
-        elif isinstance(message, ListElements):
-            self.__send_list(
-                text=message.text,
-                title=message.title, 
-                button_title=message.button_title, 
-                caption=message.caption,
-                element_list=message.elements, 
+            message = Message(
+                type="message",
+                detail=str(message),
             )
+
+        if isinstance(message, Message):
+            self.__send(message)
         else:
             raise ValueError("Tipo de mensagem inválido.")
-    
-    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.voll_id,
-                "message_text": text,
-                "platform": self.channel,
-            }
-        )
+
+    def __send(self, message: Message) -> None:
+
+        dict_message = message.to_dict()
+        dict_message["chat_id"] = self.__user_state.chatID.to_dict()
+        response = self.__router_client.send_message(dict_message)
 
         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.voll_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,
-        button_title: str,
-        title: str|None = None,
-        element_list: list[dict] = None,
-        caption: str|None = None,
-        ) -> None:
+    def transfer_to_human(self, message: str, campaign_name: str) -> None:
         
-        if not button_title:
-            raise NameError('Button Title é um parâmetro obrigatório!')
+        campaign = self.__router_client.get_campaign_id({"name": campaign_name})
         
-        if len(element_list) > 20:
-            raise ValueError("O número máximo de elementos é 20.")
-        
-        if not text:
-            raise ValueError("O corpo da mensagem não pode ser vazio.")
-            
-        
-        response = self.__wpp_server_client.send_list(
+        response = self.__router_client.transfer_to_human(
             {
-                "hook_id": self.company_phone,
-                "enterprise_id": self.customer_phone,
-                "unique_customer_id": self.__user_state.voll_id,
-                "message_text": text,
-                "button_title": button_title,
-                "message_caption": caption,
-                "message_title": title,
-                "options": [{"title": k, "description": v} for k,v in element_list.items()],
-            }
-        )
-
-        if not response.status:
-            raise ValueError("Erro ao enviar mensagem de lista.")
-    
-    def transfer_to_human(self, message:str, campaign_name:str) -> None:
-        response = self.__wpp_server_client.transfer_to_human(
-            {
-                "hook_id": self.company_phone,
-                "enterprise_id": self.customer_phone,
-                "unique_customer_id": self.__user_state.customer_id,
-                "voll_id": self.__user_state.voll_id,
-                "message_text": message,
-                "platform": self.channel,
-                "campaign_name": campaign_name,
+                "chat_id": self.__user_state.chatID.to_dict(),
+                "campaign_id": campaign.id,
             }
         )
 
         if not response.status:
             raise ValueError("Erro ao transferir chat para humano.")
-    
-    def end_chat(self, message:str, tabulation_name:str) -> None:
-        response = self.__wpp_server_client.end_chat(
+
+    def end_chat(self, message: str, tabulation_name: str) -> None:
+        tabulation = self.__router_client.get_tabulation_id({"name": tabulation_name})
+        
+        response = self.__router_client.end_chat(
             {
-                "tabulation_name": tabulation_name,
-                "hook_id": self.company_phone,
-                "unique_customer_id": self.__user_state.customer_id,
-                "voll_id": self.__user_state.voll_id,
-                "message_text": message,
-                "platform": self.channel,
+                "chat_id": self.__user_state.chatID.to_dict(),
+                "tabulation_id": tabulation.id,
             }
         )
 
         if not response.status:
             raise ValueError("Erro ao encerrar chat.")
-    
+
     def delete_user_state(self) -> None:
-        response = self.__user_state_client.delete_user_state(self.__user_state.customer_id)
+        response = self.__user_state.delete(self.__grpc_uri)
 
         if not response.status:
             raise ValueError("Erro ao deletar estado do usuário.")
 
     def update_user_state(
-        self, 
+        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),
-        })
+        observation: dict,
+    ) -> None:
 
-        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
-    
+        self.__user_state.observation = observation
+        self.__user_state.insert(self.__grpc_uri)
+
+    @property
+    def chatID(self):
+        return self.__user_state.chatID
+
+    @property
+    def user_id(self):
+        return self.__user_state.chatID.user_id
+
+    @property
+    def company_id(self):
+        return self.__user_state.chatID.company_id
+
     @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
-    
-    @property
-    def voll_id(self):
-        return self.__user_state.voll_id
-    
+
     @property
-    def platform(self):
-        return self.__user_state.platform
+    def protocol(self):
+        return self.__user_state.protocol
     
     @property
-    def direction_in(self):
-        return self.__user_state.direction_in
+    def observation(self):
+        return self.__user_state.observation
     
     @property
-    def lst_update(self):
-        return self.__user_state.lst_update
+    def type_message(self):
+        return self.__type_message
     
     @property
-    def protocol(self):
-        return self.__user_state.protocol
-    
+    def content_message(self):
+        return self.__content_message
+
     @menu.setter
     def menu(self, menu):
-        
-        self.update_user_state(
-            menu, 
-            self.__user_state.route, 
-            self.__user_state.obs
-        )
-    
+
+        self.update_user_state(menu, self.__user_state.route, self.__user_state.observation)
+
     @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, route, self.__user_state.observation)
+
+    @observation.setter
+    def observation(self, observation):
+
         self.update_user_state(
-            self.__user_state.menu, 
-            self.__user_state.route, 
-            obs
+            self.__user_state.menu, self.__user_state.route, observation
         )
-    

{chatgraph-0.4.2.dist-info → chatgraph-0.5.0.dist-info}/METADATA

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

chatgraph-0.5.0.dist-info/RECORD

@@ -0,0 +1,21 @@
+chatgraph/__init__.py,sha256=SAMzqUOZCW8K0mvWns1zjQFimsiZijAMwuJrH_EG-2M,713
+chatgraph/auth/credentials.py,sha256=xsMEpLQnc66novPjL6upocMcnUnvFJ7yxINzUenkxmc,2388
+chatgraph/bot/chatbot_model.py,sha256=GGeU-LZYqR7zhrP-B5gRloJi31_7Vmk7--H9bGKvcto,6826
+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=8rxs_tGPY5UeP0dnoqL8zhTZ7XqJLOJcewlK8yB8Q_E,4669
+chatgraph/messages/message_consumer.py,sha256=yHhfLYjtmXQYG44o7SNO6XUaeM4-jH8BpIpPHI-mt8Y,6116
+chatgraph/pb/router.proto,sha256=pw3hTwjdMN8-yJIAlKHNigVKYqM-w2nme9-2bCTBiJ0,2853
+chatgraph/pb/router_pb2.py,sha256=jw-Rg05NMwgOKrRsoubFFb82v_O80KgWY4tIqwUf2Aw,6571
+chatgraph/pb/router_pb2_grpc.py,sha256=KxhVxCMVD9tpuX7CyXqIkzZ4P3zfXeVT92Bt_Hi7mrE,24109
+chatgraph/types/end_types.py,sha256=--Ty2gM_y7J-yRAvZV26e4HMUpoguAMAhfOIS9-kQTk,1316
+chatgraph/types/message_types.py,sha256=NM0kjiajKnZd-Yr56Stehdpr18xc857FZW8TzmqgUmU,3369
+chatgraph/types/request_types.py,sha256=BeUI2l9XZpBf5c8xyPvKI68WZ5u6n98cC8DmUfWKlQ0,8788
+chatgraph/types/route.py,sha256=FO5INy5UXgicuQ8FKEZKcPv6WS5YH10dPx2OaPu_0sE,2978
+chatgraph-0.5.0.dist-info/entry_points.txt,sha256=bO9_Q-PqE5vCNNo6ke_-3j2gHfKQMDGnKDTkNuCdBuA,48
+chatgraph-0.5.0.dist-info/LICENSE,sha256=rVJozpRzDlplOpvI8A1GvmfVS0ReYdZvMWc1j2jV0v8,1090
+chatgraph-0.5.0.dist-info/METADATA,sha256=dDJI_cTfs7Ayhshv6VB7xTvP3aV3uN694bxMD6se65Y,12866
+chatgraph-0.5.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+chatgraph-0.5.0.dist-info/RECORD,,

chatgraph/pb/userstate.proto

@@ -1,41 +0,0 @@
-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);
-
-    rpc GetAllUserStates(Void) returns (UserStatesList);
-}
-
-message Void {
-}
-
-message UserState {
-    string user_id = 1;
-    string menu_id = 2;
-    string route = 3;
-    string obs = 4;
-    string date = 5;
-    bool direction = 6;
-    string voll_id = 7;
-    string platform = 8;
-}
-
-message UserStateId {
-    string user_id = 1;
-}
-
-message RequestStatus {
-    bool status = 1;
-    string message = 2;
-}
-
-message UserStatesList {
-    repeated UserState user_states = 1;
-}

chatgraph/pb/userstate_pb2.py

@@ -1,47 +0,0 @@
-# -*- 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\"\x06\n\x04Void\"\x8d\x01\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\x12\x11\n\tdirection\x18\x06 \x01(\x08\x12\x0f\n\x07voll_id\x18\x07 \x01(\t\x12\x10\n\x08platform\x18\x08 \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(\t\";\n\x0eUserStatesList\x12)\n\x0buser_states\x18\x01 \x03(\x0b\x32\x14.userstate.UserState2\xde\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.RequestStatus\x12>\n\x10GetAllUserStates\x12\x0f.userstate.Void\x1a\x19.userstate.UserStatesListB\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['_VOID']._serialized_start=30
-  _globals['_VOID']._serialized_end=36
-  _globals['_USERSTATE']._serialized_start=39
-  _globals['_USERSTATE']._serialized_end=180
-  _globals['_USERSTATEID']._serialized_start=182
-  _globals['_USERSTATEID']._serialized_end=212
-  _globals['_REQUESTSTATUS']._serialized_start=214
-  _globals['_REQUESTSTATUS']._serialized_end=262
-  _globals['_USERSTATESLIST']._serialized_start=264
-  _globals['_USERSTATESLIST']._serialized_end=323
-  _globals['_USERSTATESERVICE']._serialized_start=326
-  _globals['_USERSTATESERVICE']._serialized_end=676
-# @@protoc_insertion_point(module_scope)