cs-models 0.0.698__py3-none-any.whl → 0.0.700__py3-none-any.whl

cs_models/resources/SalesEstimate/models.py

@@ -0,0 +1,38 @@
+from datetime import datetime
+
+from sqlalchemy import (
+    Column,
+    Integer,
+    DateTime,
+    String,
+    DECIMAL,
+    Float,
+)
+
+from ...database import Base
+
+
+class SalesEstimateModel(Base):
+    __tablename__ = "sales_estimates"
+
+    id = Column(Integer, primary_key=True)
+    fiscal_year = Column(Integer, nullable=True)
+    request_id = Column(String(20), nullable=True)
+    metric = Column(String(50), nullable=True)
+    drug_name = Column(String(128), nullable=False, index=True),
+    currency = Column(String(20), nullable=True)
+    fiscal_end_date = Column(DateTime, nullable=True)
+    estimate_date = Column(DateTime, nullable=True)
+    estimate_count = Column(Integer, nullable=True)
+    mean = Column(DECIMAL(13, 2), nullable=True)
+    median = Column(DECIMAL(13, 2), nullable=True)
+    sd = Column(Float, nullable=True)
+    high = Column(DECIMAL(13, 2), nullable=True)
+    low = Column(DECIMAL(13, 2), 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/SalesEstimate/schemas.py

@@ -0,0 +1,25 @@
+from marshmallow import (
+    Schema,
+    fields,
+    validate,
+)
+
+
+class SalesEstimateResourceSchema(Schema):
+    not_blank = validate.Length(min=1, error='Field cannot be blank')
+
+    id = fields.Integer(dump_only=True)
+    fiscal_year = fields.Integer(allow_none=True)
+    request_id = fields.String(allow_none=True)
+    metric = fields.String(allow_none=True)
+    drug_name = fields.String(required=True)
+    currency = fields.String(allow_none=True)
+    fiscal_end_date = fields.DateTime(allow_none=True)
+    estimate_date = fields.DateTime(allow_none=True)
+    estimate_count = fields.Integer(allow_none=True)
+    mean = fields.Decimal(allow_none=True)
+    median = fields.Decimal(allow_none=True)
+    sd = fields.Float(allow_none=True)
+    high = fields.Decimal(allow_none=True)
+    low = fields.Decimal(allow_none=True)
+    updated_at = fields.DateTime()

cs_models/resources/SalesEstimateCompany/models.py

@@ -0,0 +1,38 @@
+from sqlalchemy import (
+    Column,
+    Integer,
+    DateTime,
+    ForeignKey,
+)
+from datetime import datetime
+
+from ...database import Base
+
+
+class SalesEstimateCompanyModel(Base):
+    __tablename__ = 'sales_estimate_companies'
+
+    id = Column(Integer, primary_key=True)
+    company_sec_id = Column(
+        Integer,
+        ForeignKey('companies_sec.id'),
+        nullable=True,
+    )
+    company_ous_id = Column(
+        Integer,
+        ForeignKey('companies_ous.id'),
+        nullable=True,
+    )
+    sales_estimate_id = Column(
+        Integer,
+        ForeignKey('sales_estimates.id'),
+        nullable=False,
+    )
+    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/SalesEstimateCompany/schemas.py

@@ -0,0 +1,35 @@
+from marshmallow import (
+    Schema,
+    fields,
+    validate,
+    pre_load,
+    ValidationError,
+)
+
+
+class SalesEstimateCompanyResourceSchema(Schema):
+    not_blank = validate.Length(min=1, error='Field cannot be blank')
+
+    id = fields.Integer(dump_only=True)
+    sales_estimate_id = fields.Integer(required=True)
+    company_sec_id = fields.Integer(allow_none=True)
+    company_ous_id = fields.Integer(allow_none=True)
+    date = fields.DateTime(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/SalesEstimateCondition/models.py

@@ -0,0 +1,41 @@
+from datetime import datetime
+
+from sqlalchemy import (
+    Column,
+    Integer,
+    DateTime,
+    ForeignKey,
+    Float,
+    Boolean,
+)
+
+from ...database import Base
+
+
+class SalesEstimateConditionModel(Base):
+    __tablename__ = "sales_estimate_conditions"
+
+    id = Column(Integer, primary_key=True)
+    sales_estimate_id = Column(
+        Integer,
+        ForeignKey('sales_estimates.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/SalesEstimateCondition/schemas.py

@@ -0,0 +1,17 @@
+from marshmallow import (
+    Schema,
+    fields,
+    validate,
+)
+
+
+class SalesEstimateConditionResourceSchema(Schema):
+    not_blank = validate.Length(min=1, error='Field cannot be blank')
+
+    id = fields.Integer(dump_only=True)
+    sales_estimate_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/SalesEstimateIntervention/models.py

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

cs_models/resources/SalesEstimateIntervention/schemas.py

@@ -0,0 +1,17 @@
+from marshmallow import (
+    Schema,
+    fields,
+    validate,
+)
+
+
+class SalesEstimateInterventionResourceSchema(Schema):
+    not_blank = validate.Length(min=1, error='Field cannot be blank')
+
+    id = fields.Integer(dump_only=True)
+    sales_estimate_id = fields.Integer(required=True)
+    intervention_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/SalesEstimateTarget/models.py

@@ -0,0 +1,41 @@
+from datetime import datetime
+
+from sqlalchemy import (
+    Column,
+    Integer,
+    DateTime,
+    ForeignKey,
+    Float,
+    Boolean,
+)
+
+from ...database import Base
+
+
+class SalesEstimateTargetModel(Base):
+    __tablename__ = "sales_estimate_targets"
+
+    id = Column(Integer, primary_key=True)
+    sales_estimate_id = Column(
+        Integer,
+        ForeignKey('sales_estimates.id'),
+        nullable=False,
+    )
+    target_id = Column(
+        Integer,
+        ForeignKey('targets.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/SalesEstimateTarget/schemas.py

@@ -0,0 +1,17 @@
+from marshmallow import (
+    Schema,
+    fields,
+    validate,
+)
+
+
+class SalesEstimateTargetResourceSchema(Schema):
+    not_blank = validate.Length(min=1, error='Field cannot be blank')
+
+    id = fields.Integer(dump_only=True)
+    sales_estimate_id = fields.Integer(required=True)
+    target_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/Transcript/models.py

@@ -21,6 +21,7 @@ class TranscriptModel(Base):
     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,

cs_models/resources/Transcript/schemas.py

@@ -16,4 +16,5 @@ class TranscriptResourceSchema(Schema):
     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-0.0.698.dist-info → cs_models-0.0.700.dist-info}/METADATA

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

{cs_models-0.0.698.dist-info → cs_models-0.0.700.dist-info}/RECORD

@@ -860,6 +860,21 @@ cs_models/resources/SRCIdCondition/schemas.py,sha256=w_cvdCMZSI17PLftw4tST_9r1yT
 cs_models/resources/Sales/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/Sales/models.py,sha256=gJeTtpVVKo3OKb5fYRuxdLorPrXV7RJh4b0-YFIywDg,954
 cs_models/resources/Sales/schemas.py,sha256=GTA2CNBOxn65mUornpk2eGM4exkPm7W7zO1ZJ4yiLW8,559
+cs_models/resources/SalesEstimate/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+cs_models/resources/SalesEstimate/models.py,sha256=YNK2BSheuAiQ6RI4wnfv5IXHc9p4WRbi5scY9xqgvjA,1205
+cs_models/resources/SalesEstimate/schemas.py,sha256=OntDcEUI64wyZGAIB5_dir9oS5-5taMohRnbmO4_CS4,863
+cs_models/resources/SalesEstimateCompany/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+cs_models/resources/SalesEstimateCompany/models.py,sha256=wnbMVuQKVv08B-_VOgGtqHgToTbmfTCjdUxPW0Nkcuo,940
+cs_models/resources/SalesEstimateCompany/schemas.py,sha256=EOPE_FWKffdM2i_l-PzhD4wEo9pV7PQZtz9EVehNIBw,1149
+cs_models/resources/SalesEstimateCondition/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+cs_models/resources/SalesEstimateCondition/models.py,sha256=ItKOmBwVe_MyTIUyHFrDdXFV0Vz-FNyevATMdn25AXM,961
+cs_models/resources/SalesEstimateCondition/schemas.py,sha256=GnF9hThDzpExhFf86V9-cLxzT4pNSgSSvKKs0IFqLbs,501
+cs_models/resources/SalesEstimateIntervention/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+cs_models/resources/SalesEstimateIntervention/models.py,sha256=4oAbpjc5D27WZaVT4D5BfB3hnc_DQg3VEL8o8ld6bXI,973
+cs_models/resources/SalesEstimateIntervention/schemas.py,sha256=RQXJiNwPRdWkuBp_0oK9RqpZMiKMMxrBcLuuEokhm08,507
+cs_models/resources/SalesEstimateTarget/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+cs_models/resources/SalesEstimateTarget/models.py,sha256=1VKRqnIbN1ldk-UGw4TxmhzRHC-qa4ArxpkYLnkRuYk,949
+cs_models/resources/SalesEstimateTarget/schemas.py,sha256=94kjvgpavcEkvJ9RABMo9w7lRsUgrcWHs5I9-ewCq7A,495
 cs_models/resources/SalesProductsMap/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/SalesProductsMap/models.py,sha256=kObHYA2q2wNY8QTqZd0mP5nSdFGoqnXRwlzOSSjvdK8,858
 cs_models/resources/SalesProductsMap/schemas.py,sha256=zocpRe3by4_wUKwy-Cp2I190-scyG9wbgjZs37Je728,401
@@ -911,8 +926,8 @@ cs_models/resources/TickerCompany/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeR
 cs_models/resources/TickerCompany/models.py,sha256=trbXLaiO0GPke7pzDxZ-cXSspmOaYi3EF0bendx6pjg,884
 cs_models/resources/TickerCompany/schemas.py,sha256=5PwrHDwYxDQyEdC5hLq8lD9bbYDFJMgLcPtAtu0-pwY,1152
 cs_models/resources/Transcript/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-cs_models/resources/Transcript/models.py,sha256=TAu87b5NgFy9c9Gb3y7YZt5DaYbcMjNjjicLWL2WEt8,760
-cs_models/resources/Transcript/schemas.py,sha256=vKGpqYUSHxc5ZhtFhXUUBBBADA7A_Dc2NO9fA5vbOjU,578
+cs_models/resources/Transcript/models.py,sha256=dZMz-JjbDiBHXfo0AN08qh8HXpQrYfiRuK6LHD1ykDs,821
+cs_models/resources/Transcript/schemas.py,sha256=fxhdWjAydlk5iwBP7xpfEsHxGvd45BEOzClnLs3oxb0,636
 cs_models/resources/TranscriptCondition/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/TranscriptCondition/models.py,sha256=WqFUWpIjqDRBkvyjW4D8QwN4bUePzSAFdAZ0nsH0nvY,946
 cs_models/resources/TranscriptCondition/schemas.py,sha256=096N5lJ1oSP1ziMraeaaXuTIhVTR3AY2jGmFei2K5XM,494
@@ -1147,7 +1162,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.698.dist-info/METADATA,sha256=L6cIBfEaGYgzAUiGR-xxQ9I7qGKx4G-Ux41I1fyWPWY,792
-cs_models-0.0.698.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-cs_models-0.0.698.dist-info/top_level.txt,sha256=M7CA8Nh5t0vRManQ9gHfphhO16uhMqIbfaxr1jPDg18,10
-cs_models-0.0.698.dist-info/RECORD,,
+cs_models-0.0.700.dist-info/METADATA,sha256=1Zg6DZRRsRuN9-dlNR7jWP6Tb5uAqbpVYQVghkUQrLc,792
+cs_models-0.0.700.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+cs_models-0.0.700.dist-info/top_level.txt,sha256=M7CA8Nh5t0vRManQ9gHfphhO16uhMqIbfaxr1jPDg18,10
+cs_models-0.0.700.dist-info/RECORD,,