cs-models 0.0.730__py3-none-any.whl → 0.0.732__py3-none-any.whl

cs_models/resources/DealInvestor/models.py

@@ -25,6 +25,7 @@ class DealInvestorModel(Base):
         ForeignKey('investors.id'),
         nullable=False,
     )
+    lead = Column(Boolean, nullable=True)
     is_deleted = Column(Boolean, nullable=True)
     updated_at = Column(
         DateTime,

cs_models/resources/DealInvestor/schemas.py

@@ -11,5 +11,6 @@ class DealInvestorResourceSchema(Schema):
     id = fields.Integer(dump_only=True)
     deal_id = fields.Integer(required=True)
     investor_id = fields.Integer(required=True)
+    lead = fields.Boolean(allow_none=True)
     is_deleted = fields.Boolean(allow_none=True)
     updated_at = fields.DateTime()

cs_models/resources/VACompanyMap/models.py

@@ -0,0 +1,34 @@
+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)
+    cik = Column(String(191), nullable=False, index=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,35 @@
+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)
+    cik = 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-0.0.730.dist-info → cs_models-0.0.732.dist-info}/METADATA

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

{cs_models-0.0.730.dist-info → cs_models-0.0.732.dist-info}/RECORD

@@ -243,8 +243,8 @@ cs_models/resources/DealCondition/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeR
 cs_models/resources/DealCondition/models.py,sha256=-q6W-YKst8eFQsuB8E3s98_VIQWjSIqJFVEdtW_G28M,804
 cs_models/resources/DealCondition/schemas.py,sha256=m8ag-gkWjD6mX3Gbqp2yPQ21gxnEW3eFIUGzywuyFbA,399
 cs_models/resources/DealInvestor/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-cs_models/resources/DealInvestor/models.py,sha256=3sUkMsSnJkq1sPtgLrfiRLLWPDYEk7JnQM-YR-cRf7A,800
-cs_models/resources/DealInvestor/schemas.py,sha256=hThlFr7Hg2X6VEjK53SZrQ5NMV5gJInj5IuUkawlRsE,397
+cs_models/resources/DealInvestor/models.py,sha256=9nq0uvnYWG0LQN1Cfm5bJNY6AhtI4d2GdessRAnIQMg,842
+cs_models/resources/DealInvestor/schemas.py,sha256=QAl7cM_lGSxjJK4Zyof8yLfXqIUfuXmFjg8p9TWuKAc,440
 cs_models/resources/DealOutbox/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/DealOutbox/models.py,sha256=jf1Vh5NGKO94X-0GVDrsWCjTeYg6Wu9HCDuIH-IECyA,1297
 cs_models/resources/DealOutbox/schemas.py,sha256=-5EPiWngMfq_igX2u6-Zn8XRhq4PI-4zAPAKAIN_Vdw,1118
@@ -1041,6 +1041,9 @@ cs_models/resources/UserWatchlistAccess/schemas.py,sha256=YLC99AAtHJBltD_1vuzIPY
 cs_models/resources/UserWorkbook/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/UserWorkbook/models.py,sha256=1f5V5o4AcHElMY31bME4S1wm06bm3SbLucYJrzk0ONk,759
 cs_models/resources/UserWorkbook/schemas.py,sha256=0roXytt6Rv8RS6Um9WPZbWz2eYptmxacc7SEp0c2vHQ,427
+cs_models/resources/VACompanyMap/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+cs_models/resources/VACompanyMap/models.py,sha256=5CQ-tXrG7OKPBIaBUICykTeeRzXJFP4Pg9TVRK8uzqs,750
+cs_models/resources/VACompanyMap/schemas.py,sha256=d1UPxtu6L-scc7HucwOf_6JcQ7G1-8NlfTu8eHzZMd4,1124
 cs_models/resources/VAParameter/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/VAParameter/models.py,sha256=R2A4jsBKuev6VLcxniTcwc0urEs3b-Lx-XGTreXDoyM,956
 cs_models/resources/VAParameter/schemas.py,sha256=NNEzDQbS7pwL5KjL_kXKCO3lqExu6nMERyPe_gltkys,776
@@ -1198,7 +1201,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.730.dist-info/METADATA,sha256=YiEGLDJc5XK5__WpkL82zEPtD2un8Xgk2e_XJSt79Ag,792
-cs_models-0.0.730.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-cs_models-0.0.730.dist-info/top_level.txt,sha256=M7CA8Nh5t0vRManQ9gHfphhO16uhMqIbfaxr1jPDg18,10
-cs_models-0.0.730.dist-info/RECORD,,
+cs_models-0.0.732.dist-info/METADATA,sha256=H7W92sLIkSqxI-j0OqFAqqbjiyVUA7iefIsgpW7h4uc,792
+cs_models-0.0.732.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+cs_models-0.0.732.dist-info/top_level.txt,sha256=M7CA8Nh5t0vRManQ9gHfphhO16uhMqIbfaxr1jPDg18,10
+cs_models-0.0.732.dist-info/RECORD,,