cs-models 0.0.668__py3-none-any.whl → 0.0.669__py3-none-any.whl

cs_models/resources/WorkbookBlock/schemas.py

@@ -1,10 +1,49 @@
 """Marshmallow Schema for AssistantCommand."""
+import json
 from marshmallow import Schema, fields
 from ..WorkbookBlockComment.schemas import (
     WorkbookBlockCommentResourceSchema,
 )
 
 
+class WorkbookBlockDataField(fields.Field):
+    """Field that stores result for the Assistant command."""
+
+    def _serialize(self, value, attr, obj, **kwargs):
+        """
+        In the DB, the `result` field is a text field. We persist
+        data by performing the following:
+
+        AssistantCommandModel(
+            ...
+            result=json.dumps({...}),
+        )
+
+        So here we need to perform the inverse operation (i.e `json.loads(..)`)
+        """
+        if value is None:
+            return None
+        return json.loads(value)
+
+    def _deserialize(self, value, attr, data, **kwargs):
+        """
+        In the DB, the `result` field is a text field. We persist
+        data by performing the following:
+
+
+
+        AssistantCommandModel(
+            ***AssistantCommandResourceSchema().load({
+                ...
+                "result": [{"some_key": 1}, {"some_key": 2}],
+            }),
+        )
+        """
+        if value is None:
+            return None
+        return json.dumps(value)
+
+
 class WorkbookBlockResourceSchema(Schema):
     """Class for AssistantCommandResource schema"""
 
@@ -12,7 +51,7 @@ class WorkbookBlockResourceSchema(Schema):
     workbook_id = fields.Integer(required=True)
     sequence_number = fields.Integer(required=True)
     block_type = fields.String(required=True)
-    block_data = fields.String(allow_none=True)
+    block_data = WorkbookBlockDataField(required=True, allow_none=True)
     comments = fields.Nested(
         WorkbookBlockCommentResourceSchema(exclude=("block_id",)),
         many=True,

{cs_models-0.0.668.dist-info → cs_models-0.0.669.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: cs-models
-Version: 0.0.668
+Version: 0.0.669
 Summary: MySQL db models
 Home-page: https://github.com/mindgram/cs-models
 Author: Shrey Verma

{cs_models-0.0.668.dist-info → cs_models-0.0.669.dist-info}/RECORD

@@ -1055,7 +1055,7 @@ cs_models/resources/Workbook/models.py,sha256=6sj9VhKTCnBL32_pkwUqfE9ESBRu1oCL2c
 cs_models/resources/Workbook/schemas.py,sha256=xcNoPzHK6ER06XA3fAvX3x7VtX47aBP0vrIqiJvUVKw,688
 cs_models/resources/WorkbookBlock/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/WorkbookBlock/models.py,sha256=tYxYRt6210QdvgUldtmy3p3-srlVuEBZU2yCT01Xnag,1467
-cs_models/resources/WorkbookBlock/schemas.py,sha256=vL5zIVn8qafa99hrDfQj7FT2mca53L80C_hWiDbQhLo,798
+cs_models/resources/WorkbookBlock/schemas.py,sha256=CfjGkuRff_ohR4WNSRKYkwpIrwPBvfuK9o21lNU3sWM,1859
 cs_models/resources/WorkbookBlockComment/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/WorkbookBlockComment/models.py,sha256=t1dLdweoPOzld1uBqrGJ3rrr66UkMwdiplNZEA1WtPU,1251
 cs_models/resources/WorkbookBlockComment/schemas.py,sha256=9UI_mTuPBYBk5qKUC2u0oR_clHA9XZ-bh5hhr24E-vs,563
@@ -1105,7 +1105,7 @@ cs_models/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/utils/alchemy.py,sha256=fhINGFn41owJ2DLXQKXAAtLqeZ1BRzD_qU0wPK_bsGQ,1598
 cs_models/utils/utils.py,sha256=bY623DuzycfPQiaOQT2AxfANeWfwr5w76dBuQ813-ns,3664
 cs_models/utils/profiling/__init__.py,sha256=N-73vb0M92C975fxgXyBCBjCPELl8Oh21ZY_-tzDnns,569
-cs_models-0.0.668.dist-info/METADATA,sha256=spbQxLNJpPnoyIUshWz9AkYm-6r-K3ddEFDDKSCVwZg,792
-cs_models-0.0.668.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-cs_models-0.0.668.dist-info/top_level.txt,sha256=M7CA8Nh5t0vRManQ9gHfphhO16uhMqIbfaxr1jPDg18,10
-cs_models-0.0.668.dist-info/RECORD,,
+cs_models-0.0.669.dist-info/METADATA,sha256=fUuZ7WjCpptkNy9CeIuG7goNdBkeFiWHXreJsy908xc,792
+cs_models-0.0.669.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+cs_models-0.0.669.dist-info/top_level.txt,sha256=M7CA8Nh5t0vRManQ9gHfphhO16uhMqIbfaxr1jPDg18,10
+cs_models-0.0.669.dist-info/RECORD,,