cs-models 0.0.656__py3-none-any.whl → 0.0.657__py3-none-any.whl

cs_models/resources/EventLLM/models.py

@@ -0,0 +1,51 @@
+from datetime import datetime
+
+from sqlalchemy import (
+    Column,
+    Integer,
+    String,
+    ForeignKey,
+    DateTime,
+    Text,
+)
+
+from ...database import Base
+
+
+class EventLLMModel(Base):
+    __tablename__ = "events_llm"
+
+    id = Column(Integer, primary_key=True)
+    date = Column(DateTime, nullable=False)
+    end_date = Column(DateTime, nullable=False)
+    event_type = Column(String(128), nullable=False)
+    description = Column(Text, nullable=False)
+    products = Column(Text, nullable=True)
+    company_sec_id = Column(
+        Integer,
+        ForeignKey('companies_sec.id'),
+        nullable=True,
+    )
+    company_ous_id = Column(
+        Integer,
+        ForeignKey('companies_ous.id'),
+        nullable=True,
+    )
+    news_id = Column(
+        Integer,
+        ForeignKey('newswires.id'),
+        nullable=True,
+    )
+    sec_filing_id = Column(
+        Integer,
+        ForeignKey('companies_sec_filings.id'),
+        nullable=True,
+    )
+    llm_response = 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/EventLLM/schemas.py

@@ -0,0 +1,41 @@
+from marshmallow import (
+    Schema,
+    fields,
+    validate,
+    pre_load,
+    ValidationError,
+)
+
+
+class EventLLMResourceSchema(Schema):
+    not_blank = validate.Length(min=1, error="Field cannot be blank")
+
+    id = fields.Integer(dump_only=True)
+    date = fields.DateTime(required=True)
+    end_date = fields.DateTime(required=True)
+    event_type = fields.String(required=True)
+    description = fields.String(required=True)
+    products = fields.String(allow_none=True)
+    company_sec_id = fields.Integer(allow_none=True)
+    company_ous_id = fields.Integer(allow_none=True)
+    news_id = fields.Integer(allow_none=True)
+    sec_filing_id = fields.Integer(allow_none=True)
+    llm_response = fields.String(allow_none=True)
+    updated_at = fields.DateTime(dump_only=True)
+
+    @pre_load
+    def check_company_info(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.656.dist-info → cs_models-0.0.657.dist-info}/METADATA

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

{cs_models-0.0.656.dist-info → cs_models-0.0.657.dist-info}/RECORD

@@ -289,6 +289,9 @@ cs_models/resources/EmptyCompany/schemas.py,sha256=U6fjKheQMxosVsS14C3Vz7zVNHHso
 cs_models/resources/Event/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/Event/models.py,sha256=wkRLT1tkFjXId8Ga1BiThwflXK5lRaCFqEf_guIwWM0,1227
 cs_models/resources/Event/schemas.py,sha256=63C0x7i-JUu5Q0aAZeNBd1O0_XZ7cahM3KTpkLCbCM8,1690
+cs_models/resources/EventLLM/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+cs_models/resources/EventLLM/models.py,sha256=ZsGh4a3Q606-KVFsgQD1Tdy6BuiNJ9Wx9MFwIL7N8qY,1281
+cs_models/resources/EventLLM/schemas.py,sha256=9I4kbekU5GEG2YFPfM2D5z85UqT7bAkyVjSoPANo5y4,1415
 cs_models/resources/ExplorerColumn/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/ExplorerColumn/models.py,sha256=t448jSUDUNpN3MPlA1-nf2H-K9KVfMkKIV7ruB-DSks,799
 cs_models/resources/ExplorerColumn/schemas.py,sha256=0qsNjOCkilZNa_Ly083dzp43h72MrAFM39dp2s8CnmY,442
@@ -1081,7 +1084,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.656.dist-info/METADATA,sha256=QMcUotBz6L5n1BSfrKDqTyne1XyuzQaKeDpslQzLzvk,792
-cs_models-0.0.656.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-cs_models-0.0.656.dist-info/top_level.txt,sha256=M7CA8Nh5t0vRManQ9gHfphhO16uhMqIbfaxr1jPDg18,10
-cs_models-0.0.656.dist-info/RECORD,,
+cs_models-0.0.657.dist-info/METADATA,sha256=_kW0bbGPv0xwnKC5dq92J1z8BAPPpEzcgKsB7jL6yRI,792
+cs_models-0.0.657.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+cs_models-0.0.657.dist-info/top_level.txt,sha256=M7CA8Nh5t0vRManQ9gHfphhO16uhMqIbfaxr1jPDg18,10
+cs_models-0.0.657.dist-info/RECORD,,