chatgraph 0.3.4__py3-none-any.whl → 0.3.5__py3-none-any.whl

chatgraph/types/request_types.py

@@ -1,8 +1,8 @@
-from ..gRPC.gRPCCall import WhatsappServiceClient, UserStateServiceClient
-from .message_types import Message, Button, ListElements, messageTypes, MessageTypes
+from chatgraph.gRPC.gRPCCall import WhatsappServiceClient, UserStateServiceClient
+from chatgraph.types.message_types import Message, Button, ListElements, messageTypes, MessageTypes
 from dataclasses import dataclass
 from typing import Optional
-import json
+import json, os
 
 @dataclass
 class UserState:
@@ -15,12 +15,56 @@ class UserState:
         lst_update (str): A última atualização do menu.
         obs (dict): Observações adicionais sobre o estado do usuário.
     """
-    customer_id: str
-    menu: str
-    route: str
-    lst_update: str
-    obs: Optional[dict] = None
+    def __init__(
+        self,
+        customer_id: str,
+        menu: str,
+        route: str,
+        lst_update: Optional[str]=None,
+        obs: Optional[dict] = None,
+        direction_in: Optional[bool] = True,
+    ) -> None:
+        
+        self.customer_id = customer_id
+        self.menu = menu
+        self.route = route
+        self.lst_update = lst_update
+        self.obs = obs
+        self.direction_in = direction_in
+
+    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}"
 
+    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),
+        })
+    
+    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),
+        })
+    
+    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)
 
 class UserCall:
     """

{chatgraph-0.3.4.dist-info → chatgraph-0.3.5.dist-info}/METADATA

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

{chatgraph-0.3.4.dist-info → chatgraph-0.3.5.dist-info}/RECORD

@@ -15,10 +15,10 @@ chatgraph/pb/voll_pb2.py,sha256=IkopsIc8x-lS9j6WpL9Wk5MHl41FDa8cU0kmZRpKKKY,5291
 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=-GQAQi1iMz1ysZLAXJImIg34ObnSJnhz7zv6VBrlNuw,3364
-chatgraph/types/request_types.py,sha256=ZTJY-jc4zwPO4cfpksL7J0MKtYuTtecn9ZObkTSi9OU,8577
+chatgraph/types/request_types.py,sha256=Huv9qcaR3ltmOEtrP7lAiGVYTIGjWnGEtSN1ag5IL9o,10303
 chatgraph/types/route.py,sha256=2fuHEPKUL_Vuv4g8pB6M7eJLouxXasqjesdju6_eumE,2757
-chatgraph-0.3.4.dist-info/entry_points.txt,sha256=bO9_Q-PqE5vCNNo6ke_-3j2gHfKQMDGnKDTkNuCdBuA,48
-chatgraph-0.3.4.dist-info/LICENSE,sha256=rVJozpRzDlplOpvI8A1GvmfVS0ReYdZvMWc1j2jV0v8,1090
-chatgraph-0.3.4.dist-info/METADATA,sha256=hHLj0Qq0hX0TFE8xpHmjA4xQc_YLC51cld9uBOnTbX8,12781
-chatgraph-0.3.4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-chatgraph-0.3.4.dist-info/RECORD,,
+chatgraph-0.3.5.dist-info/entry_points.txt,sha256=bO9_Q-PqE5vCNNo6ke_-3j2gHfKQMDGnKDTkNuCdBuA,48
+chatgraph-0.3.5.dist-info/LICENSE,sha256=rVJozpRzDlplOpvI8A1GvmfVS0ReYdZvMWc1j2jV0v8,1090
+chatgraph-0.3.5.dist-info/METADATA,sha256=L_CDt9AKggDZhHZujk2AgvJxhPj5NRGhT1U02rbb0C4,12781
+chatgraph-0.3.5.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+chatgraph-0.3.5.dist-info/RECORD,,