cs-models 0.0.689__py3-none-any.whl → 0.0.691__py3-none-any.whl

cs_models/resources/SmartGrid/models.py

@@ -0,0 +1,27 @@
+from datetime import datetime
+
+from sqlalchemy import Column, DateTime, Boolean, ForeignKey, Integer, Text, String
+from sqlalchemy.orm import relationship
+
+from ...database import Base
+from ..SmartGridCell.models import SmartGridCellModel
+
+
+class SmartGridModel(Base):
+    __tablename__ = "smart_grids"
+
+    id = Column(Integer, primary_key=True)
+    is_deleted = Column(Boolean, 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(),
+    )
+
+    cells = relationship(
+        "SmartGridCellModel",
+        order_by=SmartGridCellModel.row,
+        back_populates="smart_grid",
+    )

cs_models/resources/SmartGrid/schemas.py

@@ -0,0 +1,13 @@
+from marshmallow import Schema, fields
+from ..SmartGridCell.schemas import SmartGridCellResourceSchema
+
+
+class SmartGridResourceSchema(Schema):
+    id = fields.Integer(dump_only=True)
+    is_deleted = fields.Boolean(allow_none=True)
+    cells = fields.Nested(
+        SmartGridCellResourceSchema(exclude=("smart_grid_id",)),
+        many=True,
+        dump_only=True,
+    )
+    updated_at = fields.DateTime(dump_only=True)

cs_models/resources/SmartGridCell/models.py

@@ -10,6 +10,11 @@ class SmartGridCellModel(Base):
     __tablename__ = "smart_grid_cells"
 
     id = Column(Integer, primary_key=True)
+    smart_grid_id = Column(
+        Integer,
+        ForeignKey("smart_grids.id"),
+        nullable=False,
+    )
     row = Column(Integer, nullable=False)
     col = Column(Integer, nullable=False)
     type = Column(String(20), nullable=False)
@@ -31,3 +36,8 @@ class SmartGridCellModel(Base):
     user_query = relationship(
         "AssistantUserQueryModel",
     )
+
+    smart_grid = relationship(
+        "SmartGridModel",
+        back_populates="cells",
+    )

cs_models/resources/SmartGridCell/schemas.py

@@ -7,6 +7,7 @@ class SmartGridCellResourceSchema(Schema):
     """Class for AssistantCommandResource schema"""
 
     id = fields.Integer(dump_only=True)
+    smart_grid_id = fields.Integer(required=True)
     row = fields.Integer(required=True)
     col = fields.Integer(required=True)
     type = fields.String(required=True)

cs_models/resources/UserSavedSearchDigest/schemas.py

@@ -5,7 +5,7 @@ from marshmallow import (
 )
 
 
-class WatchlistDigestResourceSchema(Schema):
+class UserSavedSearchDigestResourceSchema(Schema):
     not_blank = validate.Length(min=1, error="Field cannot be blank")
 
     id = fields.Integer(dump_only=True)

{cs_models-0.0.689.dist-info → cs_models-0.0.691.dist-info}/METADATA

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

{cs_models-0.0.689.dist-info → cs_models-0.0.691.dist-info}/RECORD

@@ -869,9 +869,12 @@ cs_models/resources/ScratchpadTable/schemas.py,sha256=pQPf6xpsg0ZKDM9xwesvrpdq8H
 cs_models/resources/SearchLink/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/SearchLink/models.py,sha256=l1tPCSKHK7Wq1OUWFH72NYncEt6Q3bR0P6ii7YKoxCM,729
 cs_models/resources/SearchLink/schemas.py,sha256=A8Y6LRgw7MHs9bihf0gZPfqp6KDm0Z8k6PJSL9WVxsc,448
+cs_models/resources/SmartGrid/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+cs_models/resources/SmartGrid/models.py,sha256=Ir7ZzVLoKdbUOy8XV84T1c2GVLNcDWNKXg-yaX9uYms,808
+cs_models/resources/SmartGrid/schemas.py,sha256=tX10bmOXBump0AN4LOU-jhKRbNYiCzHsCSzHjBwsPJ8,423
 cs_models/resources/SmartGridCell/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-cs_models/resources/SmartGridCell/models.py,sha256=qolCmQ7C2BUetYADV3Ju6V5EK-byFK3RWAwU1mtYOqk,987
-cs_models/resources/SmartGridCell/schemas.py,sha256=JVPfua3I14z8h79G5kRMjZYqsyyuyHM2rELSH3cc-Zc,716
+cs_models/resources/SmartGridCell/models.py,sha256=vPXX8YaVPMjq6IgGhgAy46EnmbFgGCaHlZfNEUd9hq4,1196
+cs_models/resources/SmartGridCell/schemas.py,sha256=l6kljC4vUd1IoTwezM8UEQWnPoaJru8MqqPKlfyPdNw,766
 cs_models/resources/Stream/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/Stream/models.py,sha256=wXb7yOjW82ZqmLhRcDhjLPk6_WJ5FGmvKpOcSvDMuBM,1884
 cs_models/resources/Stream/schemas.py,sha256=ZFsdEKINFGaSBUAcQQvilq5MKpYGliZseguQoc_opxk,3111
@@ -994,7 +997,7 @@ cs_models/resources/UserSavedSearchAccess/models.py,sha256=mMbMvvMnKMTTfjGR0_z4o
 cs_models/resources/UserSavedSearchAccess/schemas.py,sha256=p7zykn8OsYpGHhfsy5aTxYFX1IYojaFcOC7HPURhxwg,605
 cs_models/resources/UserSavedSearchDigest/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/UserSavedSearchDigest/models.py,sha256=alX7owQynkde0zZO2nk3Qz8M_gmDWgbfRj4pTIZGWzA,940
-cs_models/resources/UserSavedSearchDigest/schemas.py,sha256=IN4Le_mtuL1lRF99FD9UgGlz8fq6qzp_9scUbSUF9Ag,497
+cs_models/resources/UserSavedSearchDigest/schemas.py,sha256=hhwrqEdqeSGSrASid5-n6Ll4yy2satrh37Q48wdxTZ0,503
 cs_models/resources/UserSearchHistoryItem/__init__.py,sha256=VAR16vegIC2J7P3s7EoILHGN1WtkSwABRkxajmlmPoc,2864
 cs_models/resources/UserSearchHistoryItem/models.py,sha256=GdNZsTEett2x_gU92iPbRFpozimmFWn_p9liUtXWRTk,532
 cs_models/resources/UserSearchHistoryItem/schemas.py,sha256=e68GcY-1KV75VWwWCGryTufWXHL09h7b48UhxieoaUQ,747
@@ -1144,7 +1147,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.689.dist-info/METADATA,sha256=NrNLTy2FOoM9-ZEdLOEnXSMW78hK92gFfm0wuWkSshg,792
-cs_models-0.0.689.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-cs_models-0.0.689.dist-info/top_level.txt,sha256=M7CA8Nh5t0vRManQ9gHfphhO16uhMqIbfaxr1jPDg18,10
-cs_models-0.0.689.dist-info/RECORD,,
+cs_models-0.0.691.dist-info/METADATA,sha256=nzIXX8FuqKSjRUsk4EyU0oI8rx49YUwb1JXzVnPQokY,792
+cs_models-0.0.691.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+cs_models-0.0.691.dist-info/top_level.txt,sha256=M7CA8Nh5t0vRManQ9gHfphhO16uhMqIbfaxr1jPDg18,10
+cs_models-0.0.691.dist-info/RECORD,,