cs-models 0.0.747__py3-none-any.whl → 0.0.749__py3-none-any.whl

cs_models/resources/WorkbookWorkflow/models.py

@@ -0,0 +1,36 @@
+from sqlalchemy import (
+    Column,
+    Integer,
+    DateTime,
+    Boolean,
+    ForeignKey,
+)
+from datetime import datetime
+from sqlalchemy.orm import relationship
+
+from ...database import Base
+from ..WorkbookWorkflowBlock.models import WorkbookWorkflowBlockModel
+
+
+class WorkbookWorkflowModel(Base):
+    __tablename__ = 'workbook_workflows'
+
+    id = Column(Integer, primary_key=True)
+    workbook_id = Column(
+        Integer,
+        ForeignKey("workbooks.id"),
+        nullable=False,
+    )
+    is_deleted = Column(Boolean, nullable=True)
+    updated_at = Column(
+        DateTime,
+        default=datetime.utcnow,
+        onupdate=datetime.utcnow,
+    )
+
+    workflow_blocks = relationship(
+        "WorkbookWorkflowBlockModel",
+        primaryjoin="WorkbookWorkflowModel.id==WorkbookWorkflowBlockModel.workflow_id",
+        order_by=WorkbookWorkflowBlockModel.sequence_number,
+        back_populates="workbook_workflow",
+    )

cs_models/resources/WorkbookWorkflow/schemas.py

@@ -0,0 +1,22 @@
+from marshmallow import (
+    Schema,
+    fields,
+    validate,
+)
+from ..WorkbookWorkflowBlock.schemas import (
+    WorkbookWorkflowBlockResourceSchema,
+)
+
+
+class WorkbookWorkflowResourceSchema(Schema):
+    not_blank = validate.Length(min=1, error="Field cannot be blank")
+
+    id = fields.Integer(dump_only=True)
+    workbook_id = fields.Integer(required=True)
+    is_deleted = fields.Boolean(allow_none=True)
+    workflow_blocks = fields.Nested(
+        WorkbookWorkflowBlockResourceSchema(exclude=("workflow_id",)),
+        many=True,
+        dump_only=True,
+    )
+    updated_at = fields.DateTime(dump_only=True)

cs_models/resources/WorkbookWorkflowBlock/models.py

@@ -0,0 +1,34 @@
+from datetime import datetime
+from sqlalchemy import Column, DateTime, Boolean, ForeignKey, Integer, Text, String
+from sqlalchemy.orm import relationship
+from ...database import Base
+
+
+class WorkbookWorkflowBlockModel(Base):
+    __tablename__ = "workbook_workflow_blocks"
+
+    id = Column(Integer, primary_key=True)
+    workflow_id = Column(
+        Integer,
+        ForeignKey("workbook_workflows.id"),
+        nullable=False,
+    )
+    sequence_number = Column(Integer, nullable=False)
+    block_type = Column(String(50), nullable=False)
+    data = Column(Text, nullable=True)
+    is_deleted = Column(Boolean, nullable=True)
+    created_at = Column(DateTime, nullable=False, default=lambda: datetime.utcnow())
+    updated_at = Column(
+        DateTime,
+        nullable=False,
+        # https://stackoverflow.com/questions/58776476/why-doesnt-freezegun-work-with-sqlalchemy-default-values
+        default=lambda: datetime.utcnow(),
+        onupdate=lambda: datetime.utcnow(),
+    )
+
+    # These are ORM fields. Don't need to be added in the corresponding migration.
+    # https://docs.sqlalchemy.org/en/14/orm/tutorial.html#building-a-relationship
+    workbook_workflow = relationship(
+        "WorkbookWorkflowModel",
+        back_populates="workflow_blocks",
+    )

cs_models/resources/WorkbookWorkflowBlock/schemas.py

@@ -0,0 +1,57 @@
+"""Marshmallow Schema for AssistantCommand."""
+import json
+from marshmallow import Schema, fields
+from ..WorkbookBlockComment.schemas import (
+    WorkbookBlockCommentResourceSchema,
+)
+
+
+class WorkbookWorkflowBlockDataField(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 WorkbookWorkflowBlockResourceSchema(Schema):
+    """Class for AssistantCommandResource schema"""
+
+    id = fields.Integer(dump_only=True)
+    workflow_id = fields.Integer(required=True)
+    sequence_number = fields.Integer(required=True)
+    block_type = fields.String(required=True)
+    data = WorkbookWorkflowBlockDataField(required=True, allow_none=True)
+    is_deleted = fields.Boolean(allow_none=True)
+    created_at = fields.DateTime(dump_only=True)
+    updated_at = fields.DateTime(dump_only=True)

{cs_models-0.0.747.dist-info → cs_models-0.0.749.dist-info}/METADATA

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

{cs_models-0.0.747.dist-info → cs_models-0.0.749.dist-info}/RECORD

@@ -1176,6 +1176,12 @@ cs_models/resources/WorkbookBlock/schemas.py,sha256=lRon0US-VHMOGypaiUSZsskgFpjN
 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
+cs_models/resources/WorkbookWorkflow/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+cs_models/resources/WorkbookWorkflow/models.py,sha256=hwzg5DeABtljOX7c6_MgrsVGqKN4-Rxs6owURPiUs4o,934
+cs_models/resources/WorkbookWorkflow/schemas.py,sha256=HBB3p-ZBcJehee5t6XlEJcrP--UVR-CcCii-PK-_5VY,617
+cs_models/resources/WorkbookWorkflowBlock/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+cs_models/resources/WorkbookWorkflowBlock/models.py,sha256=YxDqMG1o_hawbPjlM_GwlPSQ8xGmzHTZV--fUVWOaGg,1269
+cs_models/resources/WorkbookWorkflowBlock/schemas.py,sha256=cfL_XcEjgCEVI0SyzTkl6cyavGaS_r3BL5WPfXhKPTE,1731
 cs_models/resources_aact/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources_aact/something.py,sha256=WM5m0FzNJce1BgWjDc5YVnMQoHomeeKEyi36EKcd33s,1561
 cs_models/resources_aact/BriefSummary/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -1222,7 +1228,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.747.dist-info/METADATA,sha256=mZIwG4qnjF0mB37dvpTGayPh7MiXfmpX15YBSx4Qdpo,751
-cs_models-0.0.747.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-cs_models-0.0.747.dist-info/top_level.txt,sha256=M7CA8Nh5t0vRManQ9gHfphhO16uhMqIbfaxr1jPDg18,10
-cs_models-0.0.747.dist-info/RECORD,,
+cs_models-0.0.749.dist-info/METADATA,sha256=Drg0j5Ad3m11t10J4-D5aHYy-RiizzgR2kNaJY6ZuN0,751
+cs_models-0.0.749.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+cs_models-0.0.749.dist-info/top_level.txt,sha256=M7CA8Nh5t0vRManQ9gHfphhO16uhMqIbfaxr1jPDg18,10
+cs_models-0.0.749.dist-info/RECORD,,