Add CLIN table

atst/models/__init__.py

@@ -2,20 +2,22 @@ from sqlalchemy.ext.declarative import declarative_base
 
 Base = declarative_base()
 
-from .permissions import Permissions
-from .permission_set import PermissionSet
-from .user import User
-from .portfolio_role import PortfolioRole, Status as PortfolioRoleStatus
-from .application_role import ApplicationRole, Status as ApplicationRoleStatus
-from .environment_role import EnvironmentRole, CSPRole
-from .portfolio import Portfolio
+
 from .application import Application
-from .environment import Environment
+from .application_invitation import ApplicationInvitation
+from .application_role import ApplicationRole, Status as ApplicationRoleStatus
 from .attachment import Attachment
 from .audit_event import AuditEvent
-from .portfolio_invitation import PortfolioInvitation
-from .application_invitation import ApplicationInvitation
-from .task_order import TaskOrder
+from .clin import CLIN
+from .environment import Environment
+from .environment_role import EnvironmentRole, CSPRole
 from .notification_recipient import NotificationRecipient
+from .permissions import Permissions
+from .permission_set import PermissionSet
+from .portfolio import Portfolio
+from .portfolio_invitation import PortfolioInvitation
+from .portfolio_role import PortfolioRole, Status as PortfolioRoleStatus
+from .task_order import TaskOrder
+from .user import User
 
 from .mixins.invites import Status as InvitationStatus

atst/models/clin.py

@@ -0,0 +1,29 @@
+from enum import Enum
+from sqlalchemy import Column, Date, Enum as SQLAEnum, ForeignKey, Numeric, String
+from sqlalchemy.dialects.postgresql import ARRAY
+from sqlalchemy.orm import relationship
+
+from atst.models import Base, mixins, types
+
+
+class JEDICLINType(Enum):
+    JEDI_CLIN_1 = "jedi clin 0001"
+    JEDI_CLIN_2 = "jedi clin 0002"
+    JEDI_CLIN_3 = "jedi clin 0003"
+    JEDI_CLIN_4 = "jedi clin 0004"
+
+
+class CLIN(Base, mixins.TimestampsMixin):
+    __tablename__ = "clins"
+
+    id = types.Id()
+
+    task_order_id = Column(ForeignKey("task_orders.id"), nullable=False)
+    task_order = relationship("TaskOrder")
+
+    number = Column(String, nullable=False)
+    loas = Column(ARRAY(String), server_default="{}", nullable=False)
+    start_date = Column(Date, nullable=False)
+    end_date = Column(Date, nullable=False)
+    obligated_amount = Column(Numeric(scale=2), nullable=False)
+    jedi_clin_type = Column(SQLAEnum(JEDICLINType, native_enum=False), nullable=False)