cs-models 0.0.699__py3-none-any.whl → 0.0.701__py3-none-any.whl

cs_models/resources/SalesEstimate/models.py

@@ -18,9 +18,13 @@ class SalesEstimateModel(Base):
     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),
+    company_name = Column(String(128), nullable=True, 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)

cs_models/resources/SalesEstimate/schemas.py

@@ -11,9 +11,13 @@ class SalesEstimateResourceSchema(Schema):
     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)
+    company_name = fields.String(allow_none=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)

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-0.0.699.dist-info → cs_models-0.0.701.dist-info}/METADATA

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

{cs_models-0.0.699.dist-info → cs_models-0.0.701.dist-info}/RECORD

@@ -861,20 +861,20 @@ cs_models/resources/Sales/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJW
 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=MfxCtV8XyWoUneYhFfgyNBaUVcp1n9c-7K1Dal8jFko,1054
-cs_models/resources/SalesEstimate/schemas.py,sha256=gA48HN9zzN77meQBM9aOCYNr1P4oHLAqAwW2MqbiXGk,713
+cs_models/resources/SalesEstimate/models.py,sha256=h_pZ9FKjVyb7JFPH45wuNQDyYGJw_t7A0LjtbVgU_24,1272
+cs_models/resources/SalesEstimate/schemas.py,sha256=T-he-qQw7mAfbCfoYWG08YhHAofaCSuPp8fdkSaIA1I,913
 cs_models/resources/SalesEstimateCompany/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-cs_models/resources/SalesEstimateCompany/models.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-cs_models/resources/SalesEstimateCompany/schemas.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=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-cs_models/resources/SalesEstimateCondition/schemas.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=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-cs_models/resources/SalesEstimateIntervention/schemas.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=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-cs_models/resources/SalesEstimateTarget/schemas.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
@@ -1162,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.699.dist-info/METADATA,sha256=XH1zCPBHfW_Vy9seqn5OfEpu96lJUSVS08e4m2846uE,792
-cs_models-0.0.699.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-cs_models-0.0.699.dist-info/top_level.txt,sha256=M7CA8Nh5t0vRManQ9gHfphhO16uhMqIbfaxr1jPDg18,10
-cs_models-0.0.699.dist-info/RECORD,,
+cs_models-0.0.701.dist-info/METADATA,sha256=1ugB4y9IEIRpLg5Qw1vpSm6vFfMRQnupv9BNxFx3ivE,792
+cs_models-0.0.701.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+cs_models-0.0.701.dist-info/top_level.txt,sha256=M7CA8Nh5t0vRManQ9gHfphhO16uhMqIbfaxr1jPDg18,10
+cs_models-0.0.701.dist-info/RECORD,,