cs-models 0.0.606__py3-none-any.whl → 0.0.827__py3-none-any.whl

cs_models/resources/InfoTable13F/schemas.py

@@ -0,0 +1,20 @@
+from marshmallow import Schema, fields
+from ...utils.utils import Safe13FInteger
+
+
+class Infotable13FSchema(Schema):
+    ACCESSION_NUMBER = fields.String(required=True)
+    INFOTABLE_SK = Safe13FInteger(required=True)
+    NAMEOFISSUER = fields.String(required=True)
+    TITLEOFCLASS = fields.String(required=True)
+    CUSIP = fields.String(required=True)
+    FIGI = fields.String()
+    VALUE = Safe13FInteger(required=True)
+    SSHPRNAMT = Safe13FInteger(required=True)
+    SSHPRNAMTTYPE = fields.String(required=True)
+    PUTCALL = fields.String()
+    INVESTMENTDISCRETION = fields.String(required=True)
+    OTHERMANAGER = fields.String()
+    VOTING_AUTH_SOLE = Safe13FInteger(required=True)
+    VOTING_AUTH_SHARED = Safe13FInteger(required=True)
+    VOTING_AUTH_NONE = Safe13FInteger(required=True)

cs_models/resources/InterventionInterventionSalesData/models.py

@@ -0,0 +1,40 @@
+from datetime import datetime
+
+from sqlalchemy import (
+    Column,
+    Integer,
+    DateTime,
+    ForeignKey,
+    Boolean,
+    Float,
+)
+
+from ...database import Base
+
+
+class InterventionInterventionSalesDataModel(Base):
+    __tablename__ = "intervention_intervention_sales_data"
+
+    id = Column(Integer, primary_key=True)
+    intervention_id = Column(
+        Integer,
+        ForeignKey('interventions.id'),
+        nullable=False,
+    )
+    intervention_sales_data_id = Column(
+        Integer,
+        ForeignKey('intervention_sales_data.id'),
+        nullable=False,
+    )
+    score = Column(
+        Float,
+        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/InterventionInterventionSalesData/schemas.py

@@ -0,0 +1,16 @@
+from marshmallow import (
+    Schema,
+    fields,
+    validate,
+)
+
+
+class InterventionInterventionSalesDataResourceSchema(Schema):
+    not_blank = validate.Length(min=1, error='Field cannot be blank')
+
+    id = fields.Integer(dump_only=True)
+    intervention_id = fields.Integer(required=True)
+    intervention_sales_data_id = fields.Integer(required=True)
+    score = fields.Float(required=True)
+    is_deleted = fields.Boolean(allow_none=True)
+    updated_at = fields.DateTime()

cs_models/resources/InterventionSalesData/models.py

@@ -0,0 +1,51 @@
+from datetime import datetime
+
+from sqlalchemy import (
+    Column,
+    Integer,
+    DateTime,
+    ForeignKey,
+    String,
+    Text,
+)
+
+from ...database import Base
+
+
+class InterventionSalesDataModel(Base):
+    __tablename__ = "intervention_sales_data"
+
+    id = Column(Integer, primary_key=True)
+    source_type = Column(
+        String(50),
+        nullable=False
+    )
+    date = Column(DateTime, nullable=True)
+    news_id = Column(
+        Integer,
+        ForeignKey('newswires.id'),
+        nullable=True,
+    )
+    company_filing_id = Column(
+        Integer,
+        ForeignKey('company_filings.id'),
+        nullable=True,
+    )
+    sec_filing_id = Column(
+        Integer,
+        ForeignKey('companies_sec_filings.id'),
+        nullable=True,
+    )
+    sales_id = Column(
+        Integer,
+        ForeignKey('sales.id'),
+        nullable=True,
+    )
+    info = 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/InterventionSalesData/schemas.py

@@ -0,0 +1,43 @@
+from marshmallow import (
+    Schema,
+    fields,
+    validate,
+    pre_load,
+    ValidationError,
+)
+
+
+class InterventionSalesDataResourceSchema(Schema):
+    not_blank = validate.Length(min=1, error='Field cannot be blank')
+
+    id = fields.Integer(dump_only=True)
+    source_type = fields.String()
+    date = fields.DateTime(allow_none=True)
+    news_id = fields.Integer(allow_none=True)
+    company_filing_id = fields.Integer(allow_none=True)
+    sec_filing_id = fields.Integer(allow_none=True)
+    sales_id = fields.Integer(allow_none=True)
+    info = fields.String(allow_none=True)
+    updated_at = fields.DateTime()
+
+    @pre_load
+    def check_resource_info(self, in_data, **kwargs):
+        if self._get_number_of_source_fields(in_data) != 1:
+            raise ValidationError('Provide only one source info')
+        return in_data
+
+    def _get_number_of_source_fields(self, in_data, **kwargs):
+        result = 0
+        if 'news_id' in in_data:
+            if in_data['news_id'] is not None:
+                result += 1
+        if 'company_filing_id' in in_data:
+            if in_data['company_filing_id'] is not None:
+                result += 1
+        if 'sec_filing_id' in in_data:
+            if in_data['sec_filing_id'] is not None:
+                result += 1
+        if 'sales_id' in in_data:
+            if in_data['sales_id'] is not None:
+                result += 1
+        return result

cs_models/resources/LicensingCollab/models.py

@@ -5,7 +5,7 @@ from sqlalchemy import (
     Integer,
     DateTime,
     ForeignKey,
-    Text,
+    Boolean,
 )
 
 from ...database import Base
@@ -21,6 +21,7 @@ class LicensingCollabModel(Base):
         ForeignKey('newswires.id'),
         nullable=False,
     )
+    is_deleted = Column(Boolean, nullable=True)
     updated_at = Column(
         DateTime,
         nullable=False,

cs_models/resources/LicensingCollab/schemas.py

@@ -13,6 +13,7 @@ class LicensingCollabResourceSchema(Schema):
     id = fields.Integer(dump_only=True)
     news_id = fields.Integer(required=True)
     date = fields.DateTime(required=True)
+    is_deleted = fields.Boolean(allow_none=True)
     updated_at = fields.DateTime()
 
     @pre_load

cs_models/resources/Meeting/models.py

@@ -7,6 +7,7 @@ from sqlalchemy import (
     DateTime,
     Boolean,
     ForeignKey,
+    Text,
 )
 
 from ...database import Base
@@ -19,12 +20,19 @@ class MeetingModel(Base):
     meeting_name = Column(String(191), nullable=False, index=True)
     start_date = Column(DateTime, nullable=False)
     end_date = Column(DateTime, nullable=False)
+    title_release_date = Column(DateTime, nullable=True)
+    abstract_release_date = Column(DateTime, nullable=True)
     from_website = Column(Boolean, nullable=True)
     meeting_bucket_id = Column(
         Integer,
         ForeignKey('meeting_bucket.id'),
         nullable=True,
     )
+    author_pipeline = Column(Boolean, nullable=True)
+    insights_pipeline = Column(Boolean, nullable=True)
+    important_dates = Column(Text, nullable=True)
+    indexing_status = Column(String(128), nullable=True)
+    status = Column(String, nullable=True)
     updated_at = Column(
         DateTime,
         nullable=False,

cs_models/resources/Meeting/schemas.py

@@ -1,3 +1,4 @@
+import json
 from marshmallow import (
     Schema,
     fields,
@@ -5,6 +6,82 @@ from marshmallow import (
 )
 
 
+class MeetingImportantDatesField(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 MeetingIndexingStatusField(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 MeetingResourceSchema(Schema):
     not_blank = validate.Length(min=1, error='Field cannot be blank')
 
@@ -12,6 +89,13 @@ class MeetingResourceSchema(Schema):
     meeting_name = fields.String(required=True)
     start_date = fields.DateTime(required=True)
     end_date = fields.DateTime(required=True)
+    title_release_date = fields.DateTime(allow_none=True)
+    abstract_release_date = fields.DateTime(allow_none=True)
     from_website = fields.Boolean(allow_none=True)
     meeting_bucket_id = fields.Integer(allow_none=True)
+    author_pipeline = fields.Boolean(allow_none=True)
+    insights_pipeline = fields.Boolean(allow_none=True)
+    important_dates = MeetingImportantDatesField(allow_none=True)
+    indexing_status = MeetingIndexingStatusField(allow_none=True)
+    status = fields.String(allow_none=True)
     updated_at = fields.DateTime()

cs_models/resources/MeetingDataOutbox/models.py

@@ -0,0 +1,53 @@
+from datetime import datetime
+
+from sqlalchemy import (
+    Column,
+    Integer,
+    String,
+    DateTime,
+    Boolean,
+    ForeignKey,
+    Text,
+)
+
+from ...database import Base
+
+
+class MeetingDataOutboxModel(Base):
+    __tablename__ = "meeting_data_outbox"
+
+    id = Column(Integer, primary_key=True)
+    meeting_id = Column(
+        Integer,
+        ForeignKey('meetings.id'),
+        nullable=True,
+    )
+    meeting_name = Column(String(191), nullable=False, index=True)
+    start_date = Column(DateTime, nullable=False)
+    end_date = Column(DateTime, nullable=False)
+    from_website = Column(Boolean, nullable=True)
+    meeting_bucket_id = Column(
+        Integer,
+        ForeignKey('meeting_bucket.id'),
+        nullable=True,
+    )
+    reviewed = Column(Boolean, nullable=True)
+    submitted = Column(Boolean, nullable=True)
+    submitted_date = Column(DateTime, nullable=True)
+    file_id = Column(
+        Integer,
+        ForeignKey('files.id'),
+        nullable=True,
+    )
+    error = Column(Text, nullable=True)
+    checks = Column(Text, nullable=True)
+    completed = Column(Boolean, nullable=True)
+    data_entry_type = Column(String(50), nullable=True)
+    note = Column(String(128), 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/MeetingDataOutbox/schemas.py

@@ -0,0 +1,27 @@
+from marshmallow import (
+    Schema,
+    fields,
+    validate,
+)
+
+
+class MeetingDataOutboxResourceSchema(Schema):
+    not_blank = validate.Length(min=1, error='Field cannot be blank')
+
+    id = fields.Integer(dump_only=True)
+    meeting_id = fields.Integer(allow_none=True)
+    meeting_name = fields.String(required=True)
+    start_date = fields.DateTime(required=True)
+    end_date = fields.DateTime(required=True)
+    from_website = fields.Boolean(allow_none=True)
+    meeting_bucket_id = fields.Integer(allow_none=True)
+    reviewed = fields.Boolean(allow_none=True)
+    submitted = fields.Boolean(allow_none=True)
+    submitted_date = fields.DateTime(allow_none=True)
+    file_id = fields.Integer(allow_none=True)
+    error = fields.String(allow_none=True)
+    checks = fields.String(allow_none=True)
+    completed = fields.Boolean(allow_none=True)
+    data_entry_type = fields.String(allow_none=True)
+    note = fields.String(allow_none=True)
+    updated_at = fields.DateTime()

cs_models/resources/MeetingDataOutboxRow/models.py

@@ -0,0 +1,30 @@
+from datetime import datetime
+
+from sqlalchemy import (
+    Column,
+    Integer,
+    DateTime,
+    ForeignKey,
+    Text,
+)
+
+from ...database import Base
+
+
+class MeetingDataOutboxRowModel(Base):
+    __tablename__ = "meeting_data_outbox_rows"
+
+    id = Column(Integer, primary_key=True)
+    meeting_data_outbox_id = Column(
+        Integer,
+        ForeignKey('meeting_data_outbox.id'),
+        nullable=False,
+    )
+    row = 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/MeetingDataOutboxRow/schemas.py

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

cs_models/resources/MindgramOAuth/models.py

@@ -0,0 +1,35 @@
+from datetime import datetime
+from sqlalchemy import (
+    Column,
+    Integer,
+    String,
+    DateTime,
+    UniqueConstraint,
+    Text,
+)
+
+from ...database import Base
+
+
+class MindgramOAuthTokenModel(Base):
+    __tablename__ = "mindgram_oauth_tokens"
+    __table_args__ = (UniqueConstraint("user_id", "provider", name="ux_user_provider"),)
+
+    id = Column(Integer, primary_key=True, autoincrement=True)
+    user_id = Column(String(128), nullable=False)
+    provider = Column(String(32), nullable=False)
+    access_token = Column(Text, nullable=True)
+    refresh_token = Column(Text, nullable=False)
+    token_type = Column(String(32), nullable=True, default="Bearer")
+    scope = Column(Text, nullable=True)
+    expiry = Column(DateTime, nullable=True)
+    token_uri = Column(String(255), nullable=False)
+    client_id = Column(String(255), nullable=False)
+    client_secret = Column(String(255), 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/MindgramOAuth/schemas.py

@@ -0,0 +1,33 @@
+from marshmallow import (
+    Schema,
+    fields,
+    pre_load,
+)
+from ...utils.utils import pre_load_date_fields
+
+
+class MindgramOAuthTokenResourceSchema(Schema):
+    id = fields.Integer(dump_only=True)
+    user_id = fields.String(required=True)
+    provider = fields.String(required=True)
+    access_token = fields.String(allow_none=True)
+    refresh_token = fields.String(required=True)
+    token_type = fields.String(allow_none=True)
+    scope = fields.String(allow_none=True)
+    expiry = fields.DateTime(allow_none=True)
+    token_uri = fields.String(required=True)
+    client_id = fields.String(required=True)
+    client_secret = fields.String(required=True)
+    created_at = fields.DateTime(dump_only=True)
+    updated_at = fields.DateTime(dump_only=True)
+
+    @pre_load
+    def convert_string_to_datetime(self, in_data, **kwargs):
+        date_fields = ['expiry']
+
+        in_data = pre_load_date_fields(
+            in_data,
+            date_fields,
+            date_format='%Y%m%dT%H%M%S',
+        )
+        return in_data

cs_models/resources/NCTFDADrugReviewFile/models.py

@@ -0,0 +1,37 @@
+from sqlalchemy import (
+    Integer,
+    Column,
+    Boolean,
+    DateTime,
+    ForeignKey
+)
+from datetime import datetime
+from ...database import Base
+
+
+class NCTFDADrugReviewFileModel(Base):
+    __tablename__ = "nct_fda_drug_review_files"
+
+    id = Column(Integer, primary_key=True)
+    nct_study_id = Column(
+        Integer,
+        ForeignKey('nct_study.id'),
+        nullable=False,
+    )
+    fda_drug_review_file_id = Column(
+        Integer,
+        ForeignKey('fda_drug_review_files.id'),
+        nullable=False,
+    )
+    is_deleted = 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/NCTFDADrugReviewFile/schemas.py

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

cs_models/resources/NCTFacilities/models.py

@@ -20,6 +20,7 @@ class NCTFacilityModel(Base):
     )
     facility_name = Column(String(256))
     facility_city = Column(String(50))
+    facility_state = Column(String(50))
     facility_country = Column(String(50))
     facility_zip_code = Column(String(20))
     updated_at = Column(

cs_models/resources/NCTFacilities/schemas.py

@@ -12,6 +12,7 @@ class NCTFacilityResourceSchema(Schema):
     nct_study_id = fields.Integer(required=True)
     facility_name = fields.String(allow_none=True)
     facility_city = fields.String(allow_none=True)
+    facility_state = fields.String(allow_none=True)
     facility_country = fields.String(allow_none=True)
     facility_zip_code = fields.String(allow_none=True)
     updated_at = fields.DateTime(dump_only=True)

cs_models/resources/NCTParticipationCriteria/models.py

@@ -0,0 +1,28 @@
+from sqlalchemy import (
+    Integer,
+    Column,
+    DateTime,
+    ForeignKey,
+    Text,
+)
+from datetime import datetime
+from ...database import Base
+
+
+class NCTParticipationCriteriaModel(Base):
+    __tablename__ = "nct_participation_criteria"
+
+    id = Column(Integer, primary_key=True)
+    nct_study_id = Column(
+        Integer,
+        ForeignKey('nct_study.id'),
+        nullable=False,
+    )
+    participation_criteria = Column(Text)
+    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/NCTParticipationCriteria/schemas.py

@@ -0,0 +1,14 @@
+from marshmallow import (
+    Schema,
+    fields,
+    validate,
+)
+
+
+class NCTParticipationCriteriaResourceSchema(Schema):
+    not_blank = validate.Length(min=1, error="Field cannot be blank")
+
+    id = fields.Integer(dump_only=True)
+    nct_study_id = fields.Integer(required=True)
+    participation_criteria = fields.String(allow_none=True)
+    updated_at = fields.DateTime(dump_only=True)

cs_models/resources/NCTStudy/models.py

@@ -17,6 +17,7 @@ class NCTStudyModel(Base):
     nct_id = Column(String(128), nullable=False, unique=True)
     acronym = Column(String(128), nullable=True)
     brief_title = Column(Text)
+    official_title = Column(Text)
     brief_summary = Column(Text, nullable=True)
     description = Column(Text, nullable=True)
     enrollment = Column(Integer, nullable=True)
@@ -36,6 +37,7 @@ class NCTStudyModel(Base):
     conditions = Column(Text, nullable=True)
     interventions = Column(Text, nullable=True)
     industry_flag = Column(Boolean)
+    study_keywords = Column(Text)
     updated_at = Column(
         DateTime,
         nullable=False,