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

cs_models/resources/UserMeetingFavorite/models.py

@@ -0,0 +1,36 @@
+from sqlalchemy import (
+    Column,
+    Integer,
+    String,
+    DateTime,
+    ForeignKey,
+    Boolean,
+)
+from datetime import datetime
+
+from ...database import Base
+
+
+class UserMeetingFavoriteModel(Base):
+    __tablename__ = 'user_meeting_favorite'
+
+    id = Column(Integer, primary_key=True)
+    user_id = Column(String(128), nullable=False)
+    meeting_id = Column(
+        Integer,
+        ForeignKey('meetings.id'),
+        nullable=False,
+    )
+    pubmed_id = Column(
+        Integer,
+        ForeignKey('pubmed.id'),
+        nullable=False,
+    )
+    favorite_status = 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/UserMeetingFavorite/schemas.py

@@ -0,0 +1,13 @@
+from marshmallow import (
+    Schema,
+    fields,
+)
+
+
+class UserMeetingFavoriteResourceSchema(Schema):
+    id = fields.Integer(dump_only=True)
+    user_id = fields.String(required=True)
+    meeting_id = fields.Integer(required=True)
+    pubmed_id = fields.Integer(required=True)
+    favorite_status = fields.Boolean(allow_none=True)
+    updated_at = fields.DateTime(dump_only=True)

cs_models/resources/UserSavedSearch/models.py

@@ -25,6 +25,7 @@ class UserSavedSearchModel(Base):
     instant_notification = Column(Boolean, nullable=True)
     daily_digest = Column(Boolean, nullable=True)
     weekly_digest = Column(Boolean, nullable=True)
+    questions = Column(Text, nullable=True)
     updated_at = Column(
         DateTime,
         nullable=False,

cs_models/resources/UserSavedSearch/schemas.py

@@ -16,4 +16,5 @@ class UserSavedSearchResourceSchema(Schema):
     instant_notification = fields.Boolean(allow_none=True)
     daily_digest = fields.Boolean(allow_none=True)
     weekly_digest = fields.Boolean(allow_none=True)
+    questions = fields.String(allow_none=True)
     updated_at = fields.DateTime(dump_only=True)

cs_models/resources/UserSavedSearchAccess/models.py

@@ -0,0 +1,34 @@
+from sqlalchemy import (
+    Column,
+    Integer,
+    String,
+    DateTime,
+    Boolean,
+    ForeignKey,
+)
+from datetime import datetime
+
+from ...database import Base
+
+
+class UserSavedSearchAccessModel(Base):
+    __tablename__ = 'user_saved_searches_access'
+
+    id = Column(Integer, primary_key=True)
+    user_id = Column(String(128), nullable=False, index=True)
+    saved_search_id = Column(
+        Integer,
+        ForeignKey('user_saved_searches.id'),
+        nullable=False,
+    )
+    is_deleted = Column(Boolean, nullable=True)
+    instant_notification = Column(Boolean, nullable=True)
+    daily_digest = Column(Boolean, nullable=True)
+    weekly_digest = 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/UserSavedSearchAccess/schemas.py

@@ -0,0 +1,18 @@
+from marshmallow import (
+    Schema,
+    fields,
+    validate,
+)
+
+
+class UserSavedSearchAccessResourceSchema(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)
+    saved_search_id = fields.Integer(required=True)
+    is_deleted = fields.Boolean(allow_none=True)
+    instant_notification = fields.Boolean(allow_none=True)
+    daily_digest = fields.Boolean(allow_none=True)
+    weekly_digest = fields.Boolean(allow_none=True)
+    updated_at = fields.DateTime(dump_only=True)

cs_models/resources/UserSavedSearchCollectionMap/models.py

@@ -0,0 +1,33 @@
+from sqlalchemy import (
+    Column,
+    Integer,
+    String,
+    DateTime,
+    ForeignKey,
+)
+from datetime import datetime
+
+from ...database import Base
+
+
+class UserSavedSearchCollectionMapModel(Base):
+    __tablename__ = 'user_saved_searches_collection_map'
+
+    id = Column(Integer, primary_key=True)
+    saved_search_id = Column(
+        Integer,
+        ForeignKey('user_saved_searches.id'),
+        nullable=False,
+    )
+    collection_id = Column(
+        Integer,
+        ForeignKey('collections.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/UserSavedSearchCollectionMap/schemas.py

@@ -0,0 +1,14 @@
+from marshmallow import (
+    Schema,
+    fields,
+    validate,
+)
+
+
+class UserSavedSearchCollectionMapResourceSchema(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)
+    collection_id = fields.Integer(required=True)
+    updated_at = fields.DateTime(dump_only=True)

cs_models/resources/UserSavedSearchDigest/models.py

@@ -0,0 +1,36 @@
+from sqlalchemy import (
+    Column,
+    Integer,
+    String,
+    DateTime,
+    Boolean,
+    ForeignKey,
+)
+from datetime import datetime
+
+from ...database import Base
+
+
+class UserSavedSearchDigestModel(Base):
+    __tablename__ = 'user_saved_search_digests'
+
+    id = Column(Integer, primary_key=True)
+    user_id = Column(String(128), nullable=False, index=True)
+    saved_search_id = Column(
+        Integer,
+        ForeignKey('user_saved_searches.id'),
+        nullable=False,
+    )
+    assistant_user_query_id = Column(
+        Integer,
+        ForeignKey('assistant_user_queries.id'),
+        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(),
+    )

cs_models/resources/UserSavedSearchDigest/schemas.py

@@ -0,0 +1,16 @@
+from marshmallow import (
+    Schema,
+    fields,
+    validate,
+)
+
+
+class UserSavedSearchDigestResourceSchema(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)
+    saved_search_id = fields.Integer(required=True)
+    assistant_user_query_id = fields.Integer(required=True)
+    is_deleted = fields.Boolean(allow_none=True)
+    updated_at = fields.DateTime(dump_only=True)

cs_models/resources/UserVAParameterView/models.py

@@ -0,0 +1,27 @@
+from sqlalchemy import (
+    Column,
+    Integer,
+    String,
+    DateTime,
+    Boolean,
+)
+from datetime import datetime
+
+from ...database import Base
+
+
+class UserVAParameterViewModel(Base):
+    __tablename__ = 'user_va_parameter_views'
+
+    id = Column(Integer, primary_key=True)
+    user_id = Column(String(128), nullable=False)
+    name = Column(String(128), nullable=False)
+    condition_norm = Column(String(191), index=True, 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(),
+    )

cs_models/resources/UserVAParameterView/schemas.py

@@ -0,0 +1,13 @@
+from marshmallow import (
+    Schema,
+    fields,
+)
+
+
+class UserVAParameterViewResourceSchema(Schema):
+    id = fields.Integer(dump_only=True)
+    user_id = fields.String(required=True)
+    name = fields.String(required=True)
+    condition_norm = fields.String(required=True)
+    is_deleted = fields.Boolean(allow_none=True)
+    updated_at = fields.DateTime(dump_only=True)

cs_models/resources/UserWatchlistAccess/models.py

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

cs_models/resources/UserWatchlistAccess/schemas.py

@@ -0,0 +1,15 @@
+from marshmallow import (
+    Schema,
+    fields,
+    validate,
+)
+
+
+class UserWatchlistAccessResourceSchema(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)
+    watchlist_id = fields.Integer(required=True)
+    is_deleted = fields.Boolean(allow_none=True)
+    updated_at = fields.DateTime(dump_only=True)

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/UserWorkbookWorkflows/models.py

@@ -0,0 +1,27 @@
+from sqlalchemy import (
+    Column,
+    Integer,
+    String,
+    DateTime,
+    Boolean,
+    Text,
+)
+from datetime import datetime
+
+from ...database import Base
+
+
+class UserWorkbookWorkflowsModel(Base):
+    __tablename__ = 'user_workbook_workflows'
+
+    id = Column(Integer, primary_key=True)
+    user_id = Column(String(128), nullable=False)
+    is_active = Column(Boolean, nullable=True)
+    workflow_template = Column(Text, 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/UserWorkbookWorkflows/schemas.py

@@ -0,0 +1,15 @@
+from marshmallow import (
+    Schema,
+    fields,
+    validate,
+)
+
+
+class UserWorkbookWorkflowsResourceSchema(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)
+    workflow_template = fields.String(allow_none=True)
+    updated_at = fields.DateTime(dump_only=True)

cs_models/resources/UserWorkbookWorkflowsShared/models.py

@@ -0,0 +1,31 @@
+from sqlalchemy import (
+    Column,
+    Integer,
+    String,
+    DateTime,
+    Boolean,
+    ForeignKey,
+)
+from datetime import datetime
+
+from ...database import Base
+
+
+class UserWorkbookWorkflowsSharedModel(Base):
+    __tablename__ = 'user_workbook_workflows_shared'
+
+    id = Column(Integer, primary_key=True)
+    user_id = Column(String(128), nullable=False)
+    user_workbook_workflow_id = Column(
+        Integer,
+        ForeignKey('user_workbook_workflows.id'),
+        nullable=False,
+    )
+    is_active = 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/UserWorkbookWorkflowsShared/schemas.py

@@ -0,0 +1,15 @@
+from marshmallow import (
+    Schema,
+    fields,
+    validate,
+)
+
+
+class UserWorkbookWorkflowsSharedResourceSchema(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)
+    user_workbook_workflow_id = fields.Integer(required=True)
+    is_active = fields.Boolean(required=True)
+    updated_at = fields.DateTime(dump_only=True)

cs_models/resources/VACompanyMap/models.py

@@ -0,0 +1,37 @@
+from sqlalchemy import (
+    Column,
+    Integer,
+    String,
+    DateTime,
+    ForeignKey,
+)
+from datetime import datetime
+
+from ...database import Base
+
+
+class VACompanyMapModel(Base):
+    __tablename__ = 'va_company_map'
+
+    id = Column(Integer, primary_key=True)
+    cid = Column(Integer, nullable=False, index=True)
+    capiqcid = Column(Integer, nullable=True)
+    cik = Column(String(191), nullable=False, index=True)
+    ticker = Column(String(50), nullable=True)
+    name = Column(String(191), 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/VACompanyMap/schemas.py

@@ -0,0 +1,38 @@
+from marshmallow import (
+    Schema,
+    fields,
+    validate,
+    pre_load,
+    ValidationError,
+)
+
+
+class VACompanyMapResourceSchema(Schema):
+    not_blank = validate.Length(min=1, error='Field cannot be blank')
+
+    id = fields.Integer(dump_only=True)
+    cid = fields.Integer(required=True)
+    capiqcid = fields.Integer(allow_none=True)
+    cik = fields.String(allow_none=True)
+    ticker = fields.String(allow_none=True)
+    name = 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(dump_only=True)
+
+    @pre_load
+    def check_company_ids(self, in_data, **kwargs):
+        if self._get_number_of_company_fields(in_data) > 1:
+            raise ValidationError('Provide either company_sec_id or '
+                                  'company_ous_id, not both')
+        return in_data
+
+    def _get_number_of_company_fields(self, in_data, **kwargs):
+        result = 0
+        if 'company_sec_id' in in_data:
+            if in_data['company_sec_id'] is not None:
+                result += 1
+        if 'company_ous_id' in in_data:
+            if in_data['company_ous_id'] is not None:
+                result += 1
+        return result

cs_models/resources/VAParameter/models.py

@@ -0,0 +1,36 @@
+from sqlalchemy import (
+    Column,
+    Integer,
+    String,
+    DateTime,
+    Text,
+    Boolean,
+)
+from datetime import datetime
+
+from ...database import Base
+
+
+class VAParameterModel(Base):
+    __tablename__ = 'va_parameters'
+
+    id = Column(Integer, primary_key=True)
+    pid = Column(Integer, nullable=False, index=True)
+    cid = Column(Integer, nullable=False, index=True)
+    pname = Column(String(191), nullable=False)
+    ciso = Column(String(20), nullable=True)
+    u = Column(Integer, nullable=True)
+    sign = Column(String(10), nullable=True)
+    ppid = Column(Integer, nullable=True, index=True)
+    ftid = Column(Integer, nullable=True)
+    dpname = Column(String(191), nullable=True)
+    scid = Column(Integer, nullable=True)
+    so = Column(Integer, nullable=True)
+    llm_output = Column(Text, nullable=True)
+    needs_review = Column(Boolean, nullable=True)
+    updated_at = Column(
+        DateTime,
+        nullable=False,
+        default=datetime.utcnow,
+        onupdate=datetime.utcnow,
+    )

cs_models/resources/VAParameter/schemas.py

@@ -0,0 +1,25 @@
+from marshmallow import (
+    Schema,
+    fields,
+    validate,
+)
+
+
+class VAParameterResourceSchema(Schema):
+    not_blank = validate.Length(min=1, error='Field cannot be blank')
+
+    id = fields.Integer(dump_only=True)
+    pid = fields.Integer(required=True)
+    cid = fields.Integer(required=True)
+    pname = fields.String(required=True)
+    ciso = fields.String(allow_none=True)
+    u = fields.Integer(allow_none=True)
+    sign = fields.String(allow_none=True)
+    ppid = fields.Integer(allow_none=True)
+    ftid = fields.Integer(allow_none=True)
+    dpname = fields.String(allow_none=True)
+    scid = fields.Integer(allow_none=True)
+    so = fields.Integer(allow_none=True)
+    llm_output = fields.String(allow_none=True)
+    needs_review = fields.Boolean(allow_none=True)
+    updated_at = fields.DateTime(dump_only=True)

cs_models/resources/VAParameterCondition/models.py

@@ -0,0 +1,41 @@
+from datetime import datetime
+
+from sqlalchemy import (
+    Column,
+    Integer,
+    DateTime,
+    ForeignKey,
+    Float,
+    Boolean,
+)
+
+from ...database import Base
+
+
+class VAParameterConditionModel(Base):
+    __tablename__ = "va_parameter_conditions"
+
+    id = Column(Integer, primary_key=True)
+    va_parameter_id = Column(
+        Integer,
+        ForeignKey('va_parameters.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(),
+    )