chatgraph 0.3.12__tar.gz → 0.3.14__tar.gz

{chatgraph-0.3.12 → chatgraph-0.3.14}/PKG-INFO

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

chatgraph-0.3.14/chatgraph/pb/userstate_pb2.py

@@ -0,0 +1,47 @@
+# -*- 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)

{chatgraph-0.3.12 → chatgraph-0.3.14}/chatgraph/types/request_types.py

@@ -2,6 +2,7 @@ from chatgraph.gRPC.gRPCCall import WhatsappServiceClient, UserStateServiceClien
 from chatgraph.types.message_types import Message, Button, ListElements, messageTypes, MessageTypes
 from dataclasses import dataclass
 from typing import Optional
+from datetime import datetime
 import json, os
 
 @dataclass
@@ -20,9 +21,9 @@ class UserState:
         customer_id: str,
         menu: str,
         route: str,
-        voll_id: str,
         platform: str,
         direction_in: bool,
+        voll_id: Optional[str]=None,
         lst_update: Optional[str]=None,
         obs: Optional[dict] = None,
     ) -> None:
@@ -73,6 +74,28 @@ class UserState:
         
         user_state_client.delete_user_state(self.customer_id)
 
+    @classmethod
+    def get_user_state(cls, customer_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)
+        
+        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,
+            route=response.route,
+            obs=json.loads(response.obs),
+            direction_in=response.direction,
+            voll_id=response.voll_id,
+            platform=response.platform,
+            lst_update=date,
+        )
 class UserCall:
     """
     Representa uma mensagem recebida ou enviada pelo chatbot.

{chatgraph-0.3.12 → chatgraph-0.3.14}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "chatgraph"
-version = "0.3.12"
+version = "0.3.14"
 description = "A user-friendly chatbot library"
 authors = ["Irisson N. Lima <irisson.lima@verdecard.com.br>"]
 readme = "README.md"

chatgraph-0.3.12/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\"j\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(\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=38
-  _globals['_USERSTATE']._serialized_end=144
-  _globals['_USERSTATEID']._serialized_start=146
-  _globals['_USERSTATEID']._serialized_end=176
-  _globals['_REQUESTSTATUS']._serialized_start=178
-  _globals['_REQUESTSTATUS']._serialized_end=226
-  _globals['_USERSTATESLIST']._serialized_start=228
-  _globals['_USERSTATESLIST']._serialized_end=287
-  _globals['_USERSTATESERVICE']._serialized_start=290
-  _globals['_USERSTATESERVICE']._serialized_end=640
-# @@protoc_insertion_point(module_scope)