cs-models 0.0.655__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/resources/UserDocumentMeeting/models.py

@@ -0,0 +1,33 @@
+from datetime import datetime
+
+from sqlalchemy import (
+    Column,
+    Integer,
+    DateTime,
+    ForeignKey,
+)
+
+from ...database import Base
+
+
+class UserDocumentMeetingModel(Base):
+    __tablename__ = "user_document_meetings"
+
+    id = Column(Integer, primary_key=True)
+    user_document_id = Column(
+        Integer,
+        ForeignKey('user_documents.id'),
+        nullable=False,
+    )
+    meeting_id = Column(
+        Integer,
+        ForeignKey('meetings.id'),
+        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/UserDocumentMeeting/schemas.py

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

cs_models/resources/UserInternalDocWorkflow/models.py

@@ -0,0 +1,28 @@
+from sqlalchemy import (
+    Column,
+    Integer,
+    Text,
+    String,
+    DateTime,
+)
+from datetime import datetime
+
+from ...database import Base
+
+
+class UserInternalDocWorkflowModel(Base):
+    __tablename__ = 'user_internal_doc_workflow'
+
+    id = Column(Integer, primary_key=True)
+    user_id = Column(String(128), nullable=False, index=True)
+    internal_doc_workflow = Column(
+        Text,
+        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/UserInternalDocWorkflow/schemas.py

@@ -0,0 +1,11 @@
+from marshmallow import (
+    Schema,
+    fields,
+)
+
+
+class UserInternalDocWorkflowResourceSchema(Schema):
+    id = fields.Integer(dump_only=True)
+    user_id = fields.String(required=True)
+    internal_doc_workflow = fields.String(required=True)
+    updated_at = fields.DateTime(dump_only=True)

{cs_models-0.0.655.dist-info → cs_models-0.0.657.dist-info}/METADATA

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

{cs_models-0.0.655.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
@@ -914,6 +917,9 @@ cs_models/resources/UserDocumentCondition/schemas.py,sha256=BeeRrfrLW9N3k0IsOhoi
 cs_models/resources/UserDocumentIntervention/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/UserDocumentIntervention/models.py,sha256=XDW1UIMnsQoyszHeel_aaCGnd_XfWxIOJF-JJBn9IoM,1019
 cs_models/resources/UserDocumentIntervention/schemas.py,sha256=XKgWDYGiU7NTPqNHkqxWgbdz0wMwt8__SlxrCPhEqrU,556
+cs_models/resources/UserDocumentMeeting/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+cs_models/resources/UserDocumentMeeting/models.py,sha256=dasPUuYRJbb5rurCsWQG13b_vvJsmK51N61AeTA5rhs,770
+cs_models/resources/UserDocumentMeeting/schemas.py,sha256=ZfMPoPL_-LeTXAo1ouhbEfBwItFpTTUi4oaD509kZ4k,363
 cs_models/resources/UserDocumentTag/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/UserDocumentTag/models.py,sha256=A9fpE3iCmawx09fvQslM1zZuuWY8KNCBBTnJsbr5F2w,775
 cs_models/resources/UserDocumentTag/schemas.py,sha256=R0KOmc6XqU6dO-q7JOrjj1LEZmLWODt8gGQtVFjZXYU,413
@@ -926,6 +932,9 @@ cs_models/resources/UserExplorerColumn/schemas.py,sha256=pfL8wGtzO4KjBTuLitwSkgb
 cs_models/resources/UserFile/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/UserFile/models.py,sha256=-eQ-6VqLl2SDiQ1mpuIxSwt8rnFDF0F-jALulgqZauY,744
 cs_models/resources/UserFile/schemas.py,sha256=UN2-NlmGSW5Xvkphs7vSQjJ0uWjGNXFi_eOlytO1jV8,317
+cs_models/resources/UserInternalDocWorkflow/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+cs_models/resources/UserInternalDocWorkflow/models.py,sha256=8Dh6EeGVCG7trnr-BlYGD4rRKbu1_l20bIMU6XsruV0,699
+cs_models/resources/UserInternalDocWorkflow/schemas.py,sha256=I5waHOjEFRzcEeJ1kPk4gOi7KzTenkoj0R0oJqfwDjM,296
 cs_models/resources/UserLLMUse/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 cs_models/resources/UserLLMUse/models.py,sha256=PX7xvU74zIA5ZgE8kDhdHtKWfJoM2VubtOuJpfV4A7w,810
 cs_models/resources/UserLLMUse/schemas.py,sha256=rkBf5fPyxCzevEKeVsB74Nv98ILWQwR2IJ0Q9UwYGDc,498
@@ -1075,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.655.dist-info/METADATA,sha256=IbKLU2h4gTfqu5G39fetWmMqK2Q8MVuktZ02IAQLwmM,792
-cs_models-0.0.655.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-cs_models-0.0.655.dist-info/top_level.txt,sha256=M7CA8Nh5t0vRManQ9gHfphhO16uhMqIbfaxr1jPDg18,10
-cs_models-0.0.655.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,,