cs-models 0.0.627__py3-none-any.whl → 0.0.847__py3-none-any.whl

cs_models/resources/SmartGrid/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
+from ..SmartGridCell.models import SmartGridCellModel
+
+
+class SmartGridModel(Base):
+    __tablename__ = "smart_grids"
+
+    id = Column(Integer, primary_key=True)
+    user_id = Column(String(128), nullable=False)
+    workbook_id = Column(
+        Integer,
+        ForeignKey('workbooks.id'),
+        nullable=True,
+    )
+    table_info = Column(Text, nullable=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,18 @@
+from marshmallow import Schema, fields, validate
+from ..SmartGridCell.schemas import SmartGridCellResourceSchema
+
+
+class SmartGridResourceSchema(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_id = fields.Integer(allow_none=True)
+    table_info = fields.String(allow_none=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

@@ -0,0 +1,44 @@
+from datetime import datetime
+
+from sqlalchemy import Column, DateTime, Boolean, ForeignKey, Integer, Text, String
+from sqlalchemy.orm import relationship
+
+from ...database import Base
+
+
+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)
+    data = Column(Text, nullable=True)
+    user_query_id = Column(
+        Integer,
+        ForeignKey("assistant_user_queries.id"),
+        nullable=True,
+    )
+    reformulated_question = Column(String(Text), nullable=False)
+    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(),
+    )
+
+    user_query = relationship(
+        "AssistantUserQueryModel",
+    )
+
+    smart_grid = relationship(
+        "SmartGridModel",
+        back_populates="cells",
+    )

cs_models/resources/SmartGridCell/schemas.py

@@ -0,0 +1,22 @@
+"""Marshmallow Schema for AssistantCommand."""
+from marshmallow import Schema, fields
+from ..AssistantUserQuery.schemas import AssistantUserQueryResourceSchema
+
+
+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)
+    data = fields.String(allow_none=True)
+    user_query_id = fields.Integer(allow_none=True)
+    user_query = fields.Nested(
+        AssistantUserQueryResourceSchema(),
+        dump_only=True,
+    )
+    reformulated_question = fields.String(allow_none=True)
+    is_deleted = fields.Boolean(allow_none=True)
+    updated_at = fields.DateTime(dump_only=True)

cs_models/resources/Submission13F/models.py

@@ -0,0 +1,12 @@
+from sqlalchemy import Column, Integer, String, Date
+from ...database import Base
+
+
+class Submission13FModel(Base):
+    __tablename__ = '13fsubmission'
+
+    ACCESSION_NUMBER = Column(String(25), primary_key=True)
+    FILING_DATE = Column(Date, nullable=False)
+    SUBMISSIONTYPE = Column(String(10), nullable=False)
+    CIK = Column(String(10), nullable=False)
+    PERIODOFREPORT = Column(Date, nullable=False)

cs_models/resources/Submission13F/schemas.py

@@ -0,0 +1,10 @@
+from marshmallow import Schema, fields
+from ...utils.utils import Safe13FDate
+
+
+class Submission13FSchema(Schema):
+    ACCESSION_NUMBER = fields.String(required=True)
+    FILING_DATE = Safe13FDate(required=True)
+    SUBMISSIONTYPE = fields.String(required=True)
+    CIK = fields.String(required=True)
+    PERIODOFREPORT = Safe13FDate(required=True)

cs_models/resources/SummaryPage13F/models.py

@@ -0,0 +1,12 @@
+from sqlalchemy import Column, Integer, String, Date
+from ...database import Base
+
+
+class Summarypage13FModel(Base):
+    __tablename__ = '13fsummarypage'
+
+    ACCESSION_NUMBER = Column(String(25), primary_key=True)
+    OTHERINCLUDEDMANAGERSCOUNT = Column(Integer)
+    TABLEENTRYTOTAL = Column(Integer)
+    TABLEVALUETOTAL = Column(Integer)
+    ISCONFIDENTIALOMITTED = Column(String(1))

cs_models/resources/SummaryPage13F/schemas.py

@@ -0,0 +1,9 @@
+from marshmallow import Schema, fields
+
+
+class Summarypage13FSchema(Schema):
+    ACCESSION_NUMBER = fields.String(required=True)
+    OTHERINCLUDEDMANAGERSCOUNT = fields.Integer()
+    TABLEENTRYTOTAL = fields.Integer()
+    TABLEVALUETOTAL = fields.Integer()
+    ISCONFIDENTIALOMITTED = fields.String()

cs_models/resources/TableFigure/models.py

@@ -0,0 +1,40 @@
+from datetime import datetime
+
+from sqlalchemy import (
+    Column,
+    Integer,
+    DateTime,
+    String,
+    Text,
+    Boolean,
+    ForeignKey,
+)
+
+from ...database import Base
+
+
+class TableFigureModel(Base):
+    __tablename__ = "table_figures"
+
+    id = Column(Integer, primary_key=True)
+    file_id = Column(
+        Integer,
+        ForeignKey('files.id'),
+        nullable=True,
+    )
+    source_type = Column(String(50), nullable=False)
+    source_id = Column(Integer, nullable=False)
+    type = Column(String(50), nullable=False)
+    label = Column(String(50), nullable=True)
+    caption = Column(Text, nullable=True)
+    description = Column(Text, nullable=True)
+    content = Column(Text, nullable=True)
+    link = Column(String(255), nullable=True)
+    llm_processed = 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(),
+    )

cs_models/resources/TableFigure/schemas.py

@@ -0,0 +1,61 @@
+import json
+from marshmallow import (
+    Schema,
+    fields,
+    validate,
+)
+
+
+class TableFigureCaptionField(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 TableFigureResourceSchema(Schema):
+    not_blank = validate.Length(min=1, error='Field cannot be blank')
+
+    id = fields.Integer(dump_only=True)
+    file_id = fields.Integer(required=True)
+    source_type = fields.String(required=True)
+    source_id = fields.Integer(required=True)
+    type = fields.String(required=True)
+    label = fields.String(allow_none=True)
+    caption = TableFigureCaptionField(required=True, allow_none=True)
+    description = fields.String(allow_none=True)
+    content = fields.String(allow_none=True)
+    link = fields.String(allow_none=True)
+    llm_processed = fields.Boolean(allow_none=True)
+    updated_at = fields.DateTime()

cs_models/resources/TargetSyn/models.py

@@ -2,6 +2,7 @@ from sqlalchemy import (
     Column,
     Integer,
     String,
+    Boolean,
     DateTime,
     ForeignKey,
 )
@@ -20,6 +21,9 @@ class TargetSynModel(Base):
         ForeignKey('targets.id'),
         nullable=True,
     )
+    source = Column(String(50), nullable=True)
+    obsolete = Column(Boolean, nullable=True)
+    synonym_type = Column(String(50), nullable=True)
     updated_at = Column(
         DateTime,
         nullable=False,

cs_models/resources/TargetSyn/schemas.py

@@ -11,4 +11,7 @@ class TargetSynResourceSchema(Schema):
     id = fields.Integer(dump_only=True)
     synonym = fields.String(required=True)
     target_id = fields.Integer(required=True)
+    source = fields.String(allow_none=True)
+    obsolete = fields.Boolean(allow_none=True)
+    synonym_type = fields.String(allow_none=True)
     updated_at = fields.DateTime()

cs_models/resources/Transcript/models.py

@@ -0,0 +1,31 @@
+from sqlalchemy import (
+    Column,
+    Integer,
+    String,
+    DateTime,
+    Boolean,
+    Text,
+)
+from datetime import datetime
+
+from ...database import Base
+
+
+class TranscriptModel(Base):
+    __tablename__ = 'transcripts'
+
+    id = Column(Integer, primary_key=True)
+    event_id = Column(Integer, nullable=False, index=True)
+    date = Column(DateTime, nullable=False)
+    title = Column(String(191), nullable=False)
+    event_type = Column(String(50), nullable=True)
+    event_tags = Column(Text, nullable=True)
+    human_verified = Column(Boolean, nullable=True)
+    transcription_status = Column(String(50), nullable=True)
+    audio_url = Column(String(191), nullable=True)
+    updated_at = Column(
+        DateTime,
+        nullable=False,
+        default=datetime.utcnow,
+        onupdate=datetime.utcnow,
+    )

cs_models/resources/Transcript/schemas.py

@@ -0,0 +1,20 @@
+from marshmallow import (
+    Schema,
+    fields,
+    validate,
+)
+
+
+class TranscriptResourceSchema(Schema):
+    not_blank = validate.Length(min=1, error='Field cannot be blank')
+
+    id = fields.Integer(dump_only=True)
+    event_id = fields.Integer(required=True)
+    date = fields.DateTime(required=True)
+    title = fields.String(required=True)
+    event_type = fields.String(allow_none=True)
+    event_tags = fields.String(allow_none=True)
+    human_verified = fields.Boolean(allow_none=True)
+    audio_url = fields.String(allow_none=True)
+    transcription_status = fields.String(allow_none=True)
+    updated_at = fields.DateTime()

cs_models/resources/TranscriptCondition/models.py

@@ -0,0 +1,41 @@
+from datetime import datetime
+
+from sqlalchemy import (
+    Column,
+    Integer,
+    DateTime,
+    ForeignKey,
+    Float,
+    Boolean,
+)
+
+from ...database import Base
+
+
+class TranscriptConditionModel(Base):
+    __tablename__ = "transcript_conditions"
+
+    id = Column(Integer, primary_key=True)
+    transcript_id = Column(
+        Integer,
+        ForeignKey('transcripts.id'),
+        nullable=False,
+    )
+    condition_id = Column(
+        Integer,
+        ForeignKey('conditions.id'),
+        nullable=False,
+    )
+    score = Column(
+        Float,
+        nullable=False,
+    )
+    preferred = Column(Boolean, nullable=True)
+    date = 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/TranscriptCondition/schemas.py

@@ -0,0 +1,17 @@
+from marshmallow import (
+    Schema,
+    fields,
+    validate,
+)
+
+
+class TranscriptConditionResourceSchema(Schema):
+    not_blank = validate.Length(min=1, error='Field cannot be blank')
+
+    id = fields.Integer(dump_only=True)
+    transcript_id = fields.Integer(required=True)
+    condition_id = fields.Integer(required=True)
+    score = fields.Float(required=True)
+    preferred = fields.Boolean(allow_none=True)
+    date = fields.DateTime(allow_none=True)
+    updated_at = fields.DateTime()

cs_models/resources/TranscriptEquity/models.py

@@ -0,0 +1,28 @@
+from sqlalchemy import (
+    Column,
+    Integer,
+    DateTime,
+    ForeignKey,
+)
+from datetime import datetime
+
+from ...database import Base
+
+
+class TranscriptEquityModel(Base):
+    __tablename__ = 'transcripts_equities'
+
+    id = Column(Integer, primary_key=True)
+    transcript_id = Column(
+        Integer,
+        ForeignKey('transcripts.id'),
+        nullable=False,
+    )
+    equity_id = Column(Integer, index=True, nullable=False)
+    date = Column(DateTime, nullable=False)
+    updated_at = Column(
+        DateTime,
+        nullable=False,
+        default=datetime.utcnow,
+        onupdate=datetime.utcnow,
+    )

cs_models/resources/TranscriptEquity/schemas.py

@@ -0,0 +1,15 @@
+from marshmallow import (
+    Schema,
+    fields,
+    validate,
+)
+
+
+class TranscriptEquityResourceSchema(Schema):
+    not_blank = validate.Length(min=1, error='Field cannot be blank')
+
+    id = fields.Integer(dump_only=True)
+    transcript_id = fields.Integer(required=True)
+    equity_id = fields.Integer(required=True)
+    date = fields.DateTime(required=True)
+    updated_at = fields.DateTime()

cs_models/resources/TranscriptEquityCompany/models.py

@@ -0,0 +1,34 @@
+from sqlalchemy import (
+    Column,
+    Integer,
+    Text,
+    DateTime,
+    ForeignKey,
+)
+from datetime import datetime
+
+from ...database import Base
+
+
+class TranscriptEquityCompanyModel(Base):
+    __tablename__ = 'transcripts_equity_companies'
+
+    id = Column(Integer, primary_key=True)
+    equity_id = Column(Integer, nullable=False, index=True)
+    equity_details = Column(Text, nullable=True)
+    company_sec_id = Column(
+        Integer,
+        ForeignKey('companies_sec.id'),
+        nullable=True,
+    )
+    company_ous_id = Column(
+        Integer,
+        ForeignKey('companies_ous.id'),
+        nullable=True,
+    )
+    updated_at = Column(
+        DateTime,
+        nullable=False,
+        default=datetime.utcnow,
+        onupdate=datetime.utcnow,
+    )

cs_models/resources/TranscriptEquityCompany/schemas.py

@@ -0,0 +1,16 @@
+from marshmallow import (
+    Schema,
+    fields,
+    validate,
+)
+
+
+class TranscriptEquityCompanyResourceSchema(Schema):
+    not_blank = validate.Length(min=1, error='Field cannot be blank')
+
+    id = fields.Integer(dump_only=True)
+    equity_id = fields.Integer(required=True)
+    equity_details = fields.String(allow_none=True)
+    company_sec_id = fields.Integer(allow_none=True)
+    company_ous_id = fields.Integer(allow_none=True)
+    updated_at = fields.DateTime()

cs_models/resources/TranscriptGrouping/models.py

@@ -0,0 +1,23 @@
+from sqlalchemy import (
+    Column,
+    Integer,
+    String,
+    DateTime,
+)
+from datetime import datetime
+
+from ...database import Base
+
+
+class TranscriptGroupingModel(Base):
+    __tablename__ = 'transcripts_grouping'
+
+    id = Column(Integer, primary_key=True)
+    grouping_id = Column(Integer, nullable=False, index=True)
+    grouping_name = Column(String(191), nullable=False)
+    updated_at = Column(
+        DateTime,
+        nullable=False,
+        default=datetime.utcnow,
+        onupdate=datetime.utcnow,
+    )

cs_models/resources/TranscriptGrouping/schemas.py

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

cs_models/resources/TranscriptGroupingMap/models.py

@@ -0,0 +1,31 @@
+from sqlalchemy import (
+    Column,
+    Integer,
+    DateTime,
+    ForeignKey,
+)
+from datetime import datetime
+
+from ...database import Base
+
+
+class TranscriptGroupingMapModel(Base):
+    __tablename__ = 'transcripts_grouping_map'
+
+    id = Column(Integer, primary_key=True)
+    transcript_id = Column(
+        Integer,
+        ForeignKey('transcripts.id'),
+        nullable=False,
+    )
+    grouping_id = Column(
+        Integer,
+        ForeignKey('transcripts_grouping.id'),
+        nullable=False,
+    )
+    updated_at = Column(
+        DateTime,
+        nullable=False,
+        default=datetime.utcnow,
+        onupdate=datetime.utcnow,
+    )

cs_models/resources/TranscriptGroupingMap/schemas.py

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

cs_models/resources/TranscriptIntervention/models.py

@@ -0,0 +1,41 @@
+from datetime import datetime
+
+from sqlalchemy import (
+    Column,
+    Integer,
+    DateTime,
+    ForeignKey,
+    Float,
+    Boolean,
+)
+
+from ...database import Base
+
+
+class TranscriptInterventionModel(Base):
+    __tablename__ = "transcript_interventions"
+
+    id = Column(Integer, primary_key=True)
+    transcript_id = Column(
+        Integer,
+        ForeignKey('transcripts.id'),
+        nullable=False,
+    )
+    intervention_id = Column(
+        Integer,
+        ForeignKey('interventions.id'),
+        nullable=False,
+    )
+    score = Column(
+        Float,
+        nullable=False,
+    )
+    preferred = Column(Boolean, nullable=True)
+    date = 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(),
+    )