cs-models 0.0.665__py3-none-any.whl → 0.0.667__py3-none-any.whl

cs_models/resources/UserWorkbook/models.py

@@ -0,0 +1,31 @@
+from sqlalchemy import (
+    Column,
+    Integer,
+    String,
+    DateTime,
+    Boolean,
+    ForeignKey,
+)
+from datetime import datetime
+
+from ...database import Base
+
+
+class UserWorkbookModel(Base):
+    __tablename__ = 'user_workbooks'
+
+    id = Column(Integer, primary_key=True)
+    user_id = Column(String(128), nullable=False)
+    is_active = Column(Boolean, nullable=True)
+    workbook_id = Column(
+        Integer,
+        ForeignKey('workbooks.id'),
+        nullable=False,
+    )
+    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(),
+    )

cs_models/resources/UserWorkbook/schemas.py

@@ -0,0 +1,15 @@
+from marshmallow import (
+    Schema,
+    fields,
+    validate,
+)
+
+
+class UserWorkbookResourceSchema(Schema):
+    not_blank = validate.Length(min=1, error="Field cannot be blank")
+
+    id = fields.Integer(dump_only=True)
+    user_id = fields.String(required=True, validate=not_blank)
+    is_active = fields.Boolean(required=True)
+    workbook_id = fields.Integer(required=True)
+    updated_at = fields.DateTime(dump_only=True)

cs_models/resources/Workbook/models.py

@@ -0,0 +1,33 @@
+from sqlalchemy import (
+    Column,
+    Integer,
+    String,
+    DateTime,
+    Boolean,
+)
+from datetime import datetime
+from sqlalchemy.orm import relationship
+
+from ...database import Base
+from ..WorkbookBlock.models import WorkbookBlockModel
+
+
+class WorkbookModel(Base):
+    __tablename__ = 'workbooks'
+
+    id = Column(Integer, primary_key=True)
+    user_id = Column(String(128), nullable=False)
+    workbook_name = Column(String(128), nullable=False)
+    is_deleted = Column(Boolean, nullable=True)
+    created_at = Column(DateTime, nullable=False)
+    updated_at = Column(
+        DateTime,
+        default=datetime.utcnow,
+        onupdate=datetime.utcnow,
+    )
+
+    blocks = relationship(
+        "WorkbookBlockModel",
+        order_by=WorkbookBlockModel.sequence_number,
+        back_populates="workbook",
+    )

cs_models/resources/Workbook/schemas.py

@@ -0,0 +1,16 @@
+from marshmallow import (
+    Schema,
+    fields,
+    validate,
+)
+
+
+class WorkbookResourceSchema(Schema):
+    not_blank = validate.Length(min=1, error="Field cannot be blank")
+
+    id = fields.Integer(dump_only=True)
+    user_id = fields.String(required=True, validate=not_blank)
+    workbook_name = fields.String(required=True)
+    is_deleted = fields.Boolean(allow_none=True)
+    created_at = fields.DateTime(required=True)
+    updated_at = fields.DateTime(dump_only=True)

cs_models/resources/WorkbookBlock/models.py

@@ -0,0 +1,43 @@
+from datetime import datetime
+
+from sqlalchemy import Column, DateTime, Boolean, ForeignKey, Integer, Text, String
+from sqlalchemy.orm import relationship
+
+from ...database import Base
+from ..WorkbookBlockComment.models import WorkbookBlockCommentModel
+
+
+class WorkbookBlockModel(Base):
+    __tablename__ = "workbook_blocks"
+
+    id = Column(Integer, primary_key=True)
+    workbook_id = Column(
+        Integer,
+        ForeignKey("workbooks.id"),
+        nullable=False,
+    )
+    sequence_number = Column(Integer, nullable=False)
+    block_type = Column(String(50), nullable=False)
+    block_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 = relationship(
+        "WorkbookModel",
+        back_populates="blocks",
+    )
+
+    comments = relationship(
+        "WorkbookBlockCommentModel",
+        order_by=WorkbookBlockCommentModel.sequence_number,
+        back_populates="workbook_block",
+    )

cs_models/resources/WorkbookBlock/schemas.py

@@ -0,0 +1,15 @@
+"""Marshmallow Schema for AssistantCommand."""
+from marshmallow import Schema, fields
+
+
+class WorkbookBlockResourceSchema(Schema):
+    """Class for AssistantCommandResource schema"""
+
+    id = fields.Integer(dump_only=True)
+    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)
+    is_deleted = fields.Boolean(allow_none=True)
+    created_at = fields.DateTime(dump_only=True)
+    updated_at = fields.DateTime(dump_only=True)

cs_models/resources/WorkbookBlockComment/models.py

@@ -0,0 +1,36 @@
+from datetime import datetime
+
+from sqlalchemy import Column, DateTime, Boolean, ForeignKey, Integer, Text, String
+from sqlalchemy.orm import relationship
+
+from ...database import Base
+
+
+class WorkbookBlockCommentModel(Base):
+    __tablename__ = "workbook_block_comments"
+
+    id = Column(Integer, primary_key=True)
+    block_id = Column(
+        Integer,
+        ForeignKey("workbook_blocks.id"),
+        nullable=False,
+    )
+    user_id = Column(String(128), nullable=False)
+    sequence_number = Column(Integer, nullable=False)
+    comment = 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_block = relationship(
+        "WorkbookBlockModel",
+        back_populates="comments",
+    )

cs_models/resources/WorkbookBlockComment/schemas.py

@@ -0,0 +1,15 @@
+"""Marshmallow Schema for AssistantCommand."""
+from marshmallow import Schema, fields
+
+
+class WorkbookBlockCommentResourceSchema(Schema):
+    """Class for AssistantCommandResource schema"""
+
+    id = fields.Integer(dump_only=True)
+    block_id = fields.Integer(required=True)
+    user_id = fields.String(required=True)
+    sequence_number = fields.Integer(required=True)
+    comment = fields.String(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.665.dist-info → cs_models-0.0.667.dist-info}/METADATA

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

{cs_models-0.0.665.dist-info → cs_models-0.0.667.dist-info}/RECORD

@@ -969,6 +969,9 @@ cs_models/resources/UserWatchlist/schemas.py,sha256=gM4N7ClcMGvyaSzIg95SSOw1EbFd
 cs_models/resources/UserWatchlistAccess/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/UserWatchlistAccess/models.py,sha256=upjIxTL2WbKJF1fNVVLS1J_tw2TyDvHf0DCk5TVZQ8Q,794
 cs_models/resources/UserWatchlistAccess/schemas.py,sha256=YLC99AAtHJBltD_1vuzIPYX5uKERu6N9ljndhPJAWx8,438
+cs_models/resources/UserWorkbook/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+cs_models/resources/UserWorkbook/models.py,sha256=1f5V5o4AcHElMY31bME4S1wm06bm3SbLucYJrzk0ONk,759
+cs_models/resources/UserWorkbook/schemas.py,sha256=0roXytt6Rv8RS6Um9WPZbWz2eYptmxacc7SEp0c2vHQ,427
 cs_models/resources/ViewConditionByTA/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/ViewConditionByTA/models.py,sha256=CieIW5mvmvok4dbglxd-ijMVI7gx-6C057CViSjuWoQ,764
 cs_models/resources/ViewConditionByTA/schemas.py,sha256=Y0hoMeysHulflnPUchmDP0NDsSTKtIZH-UXRvp2fIRw,441
@@ -1047,6 +1050,15 @@ cs_models/resources/ViewPublicationTarget/schemas.py,sha256=L_WUwBOz0i4yTGdibRI_
 cs_models/resources/Watchlist/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/Watchlist/models.py,sha256=YpuJCjfZvgtK_uHJieJGAyIeC-B1Zxb-whCEKvHfsf8,1173
 cs_models/resources/Watchlist/schemas.py,sha256=2VLCMjiOA73pp_mtjo9kmWpgpOuXOCmlSQNwe39s8qw,1818
+cs_models/resources/Workbook/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+cs_models/resources/Workbook/models.py,sha256=6sj9VhKTCnBL32_pkwUqfE9ESBRu1oCL2cpQn7bxlv8,822
+cs_models/resources/Workbook/schemas.py,sha256=4gqGGN5KthKoqAYmL6QnRKMX-gGH-x61-XInNZQyTzM,475
+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=ETbgr7qkpVlX-A0r6aTnDoRLyfoveYMvoQ1KrO3aRAc,565
+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_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
@@ -1093,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.665.dist-info/METADATA,sha256=W0pIqSFDWMdtJiKZExlYyIlN_N9ysdDbjG_D1fa6nm4,792
-cs_models-0.0.665.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-cs_models-0.0.665.dist-info/top_level.txt,sha256=M7CA8Nh5t0vRManQ9gHfphhO16uhMqIbfaxr1jPDg18,10
-cs_models-0.0.665.dist-info/RECORD,,
+cs_models-0.0.667.dist-info/METADATA,sha256=O4BFF74Z3IJGYVtVAh0uqsndPymWSiNuN32o4pUCKd8,792
+cs_models-0.0.667.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+cs_models-0.0.667.dist-info/top_level.txt,sha256=M7CA8Nh5t0vRManQ9gHfphhO16uhMqIbfaxr1jPDg18,10
+cs_models-0.0.667.dist-info/RECORD,,