cs-models 0.0.563__py3-none-any.whl → 0.0.565__py3-none-any.whl

cs_models/resources/AssistantCommand/models.py

@@ -12,6 +12,7 @@ class AssistantCommandTypeEnum(enum.Enum):
     EXTRACT = "EXTRACT"
     SUMMARIZE = "SUMMARIZE"
     QUESTION_AND_ANSWER = "QUESTION_AND_ANSWER"
+    TABLE_GENERATION = "TABLE_GENERATION"
 
 
 class AssistantCommandStatusEnum(enum.Enum):

cs_models/resources/Digest/models.py

@@ -0,0 +1,33 @@
+from datetime import datetime
+
+from sqlalchemy import (
+    Column,
+    Integer,
+    DateTime,
+    ForeignKey,
+    String,
+)
+
+from ...database import Base
+
+
+class DigestModel(Base):
+    __tablename__ = "digests"
+
+    id = Column(Integer, primary_key=True)
+    saved_search_id = Column(
+        Integer,
+        ForeignKey('user_saved_searches.id'),
+        nullable=False,
+    )
+    type = Column(String(128), nullable=False)
+    description = Column(String(256), nullable=False)
+    created_at = Column(DateTime, nullable=True)
+    sent_at = Column(DateTime, nullable=True)
+    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/Digest/schemas.py

@@ -0,0 +1,17 @@
+from marshmallow import (
+    Schema,
+    fields,
+    validate,
+)
+
+
+class DigestResourceSchema(Schema):
+    not_blank = validate.Length(min=1, error='Field cannot be blank')
+
+    id = fields.Integer(dump_only=True)
+    saved_search_id = fields.Integer(required=True)
+    type = fields.String(required=True)
+    description = fields.String(required=True)
+    created_at = fields.DateTime()
+    sent_at = fields.DateTime()
+    updated_at = fields.DateTime()

cs_models/resources/DigestNotification/models.py

@@ -0,0 +1,33 @@
+from datetime import datetime
+
+from sqlalchemy import (
+    Column,
+    Integer,
+    DateTime,
+    ForeignKey,
+)
+
+from ...database import Base
+
+
+class DigestNotificationModel(Base):
+    __tablename__ = "digest_notifications"
+
+    id = Column(Integer, primary_key=True)
+    digest_id = Column(
+        Integer,
+        ForeignKey('digests.id'),
+        nullable=False,
+    )
+    notification_id = Column(
+        Integer,
+        ForeignKey('notifications.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/DigestNotification/schemas.py

@@ -0,0 +1,14 @@
+from marshmallow import (
+    Schema,
+    fields,
+    validate,
+)
+
+
+class DigestNotificationResourceSchema(Schema):
+    not_blank = validate.Length(min=1, error='Field cannot be blank')
+
+    id = fields.Integer(dump_only=True)
+    digest_id = fields.Integer(required=True)
+    notification_id = fields.Integer(required=True)
+    updated_at = fields.DateTime()

{cs_models-0.0.563.dist-info → cs_models-0.0.565.dist-info}/METADATA

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

{cs_models-0.0.563.dist-info → cs_models-0.0.565.dist-info}/RECORD

@@ -21,7 +21,7 @@ cs_models/resources/Assignee/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NM
 cs_models/resources/Assignee/models.py,sha256=_f0_UwzF1gbOlU0t0l-9UIi0E4vaIS5VsHeOxc698SE,617
 cs_models/resources/Assignee/schemas.py,sha256=N4_upICR1Cd_B45LHpOly7FEwWhAB1fXUP4y60OiezE,269
 cs_models/resources/AssistantCommand/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-cs_models/resources/AssistantCommand/models.py,sha256=2rgpg1MgSqYZOE802kIcpjIO6vHhzuy4g8iLnK_ieus,1840
+cs_models/resources/AssistantCommand/models.py,sha256=0UItcqiK9acz1s3QgqYrjpz7BrPf_QspIkixSxJA0Y8,1882
 cs_models/resources/AssistantCommand/schemas.py,sha256=7mS-WTHwPsu8Kd_0FLgJqYwrg0qVJUp7iioWqPT1ZBE,3695
 cs_models/resources/AssistantCommandEdge/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/AssistantCommandEdge/models.py,sha256=3blYNPb6zH6ewfj-VK7z77bbtvO9Ly6os4NL2vh4wIg,1372
@@ -212,6 +212,12 @@ cs_models/resources/Designation/schemas.py,sha256=CEsHYT0DX9PJq8I879aglG_JEWiUxi
 cs_models/resources/DesignationOutbox/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/DesignationOutbox/models.py,sha256=UpJZs_JQxVr7wxMLexO-yc8Hc26-btiXZnAAuLcmT9g,919
 cs_models/resources/DesignationOutbox/schemas.py,sha256=qMvxAvy4dcUoB1-yhUAYtCAGGvJixTfviROjI91VcC0,493
+cs_models/resources/Digest/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+cs_models/resources/Digest/models.py,sha256=m95Knoddoib8MESk3txzGH1LpXwKMz3a2cr2Qe8AlE0,847
+cs_models/resources/Digest/schemas.py,sha256=hg_DBRuZgb2CwmCiSTrWScRkq14hYJ7W3iEX9AmXosI,456
+cs_models/resources/DigestNotification/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+cs_models/resources/DigestNotification/models.py,sha256=9zNVi4Hsjl_ove6aReQivP7ocJy4dxr-SUoF9XI-X90,763
+cs_models/resources/DigestNotification/schemas.py,sha256=n7jih5IiRGRwD8fQr5UfEHb0NRJuWp7HOBGXcHEWXEs,360
 cs_models/resources/DocEmbedding/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/DocEmbedding/models.py,sha256=8nkbhp2GE-sZjtg6t6ABHGpRhw5W30BSB-77VvZ8h8k,760
 cs_models/resources/DocEmbedding/schemas.py,sha256=jaQCgkk-hj2HEaadOrRzkZqtwbuqw8bniZ-Vp85o88c,448
@@ -910,7 +916,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.563.dist-info/METADATA,sha256=J2dN090i-8KIb8rzBFutNFignUZqzv4zksG2N40eJUM,791
-cs_models-0.0.563.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-cs_models-0.0.563.dist-info/top_level.txt,sha256=M7CA8Nh5t0vRManQ9gHfphhO16uhMqIbfaxr1jPDg18,10
-cs_models-0.0.563.dist-info/RECORD,,
+cs_models-0.0.565.dist-info/METADATA,sha256=cqW4m8QIE5qX_-ktMn1XSRutn-RvQlosqqCo8mpO_Oc,791
+cs_models-0.0.565.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+cs_models-0.0.565.dist-info/top_level.txt,sha256=M7CA8Nh5t0vRManQ9gHfphhO16uhMqIbfaxr1jPDg18,10
+cs_models-0.0.565.dist-info/RECORD,,