Merge pull request #503 from dod-ccpo/spike-new-workflow

Task Order Form

atst/models/__init__.py

@@ -19,3 +19,4 @@ from .request_review import RequestReview
 from .request_internal_comment import RequestInternalComment
 from .audit_event import AuditEvent
 from .invitation import Invitation
+from .task_order import TaskOrder

atst/models/task_order.py

@@ -0,0 +1,68 @@
+from sqlalchemy import Column, Numeric, String, ForeignKey, Date
+from sqlalchemy.types import ARRAY
+from sqlalchemy.orm import relationship
+
+from atst.models import Base, types, mixins
+
+
+class TaskOrder(Base, mixins.TimestampsMixin):
+    __tablename__ = "task_orders"
+
+    id = types.Id()
+
+    workspace_id = Column(ForeignKey("workspaces.id"))
+    workspace = relationship("Workspace")
+
+    user_id = Column(ForeignKey("users.id"))
+    creator = relationship("User")
+
+    scope = Column(String)  # Cloud Project Scope
+    defense_component = Column(String)  # Department of Defense Component
+    app_migration = Column(String)  # App Migration
+    native_apps = Column(String)  # Native Apps
+    complexity = Column(ARRAY(String))  # Project Complexity
+    complexity_other = Column(String)
+    dev_team = Column(ARRAY(String))  # Development Team
+    dev_team_other = Column(String)
+    team_experience = Column(String)  # Team Experience
+    start_date = Column(Date)  # Period of Performance
+    end_date = Column(Date)
+    clin_01 = Column(Numeric(scale=2))
+    clin_02 = Column(Numeric(scale=2))
+    clin_03 = Column(Numeric(scale=2))
+    clin_04 = Column(Numeric(scale=2))
+    ko_first_name = Column(String)  # First Name
+    ko_last_name = Column(String)  # Last Name
+    ko_email = Column(String)  # Email
+    ko_dod_id = Column(String)  # DOD ID
+    cor_first_name = Column(String)  # First Name
+    cor_last_name = Column(String)  # Last Name
+    cor_email = Column(String)  # Email
+    cor_dod_id = Column(String)  # DOD ID
+    so_first_name = Column(String)  # First Name
+    so_last_name = Column(String)  # Last Name
+    so_email = Column(String)  # Email
+    so_dod_id = Column(String)  # DOD ID
+    number = Column(String, unique=True)  # Task Order Number
+    loa = Column(ARRAY(String))  # Line of Accounting (LOA)
+
+    @property
+    def budget(self):
+        return sum(
+            filter(None, [self.clin_01, self.clin_02, self.clin_03, self.clin_04])
+        )
+
+    def __repr__(self):
+        return "<TaskOrder(number='{}', budget='{}', end_date='{}', id='{}')>".format(
+            self.number, self.budget, self.end_date, self.id
+        )
+
+    def to_dictionary(self):
+        return {
+            "portfolio_name": self.workspace.name,
+            **{
+                c.name: getattr(self, c.name)
+                for c in self.__table__.columns
+                if c.name not in ["id"]
+            },
+        }

atst/models/workspace.py

@@ -13,10 +13,12 @@ class Workspace(Base, mixins.TimestampsMixin, mixins.AuditableMixin):
 
     id = types.Id()
     name = Column(String)
-    request_id = Column(ForeignKey("requests.id"), nullable=False)
+    request_id = Column(ForeignKey("requests.id"), nullable=True)
     projects = relationship("Project", back_populates="workspace")
     roles = relationship("WorkspaceRole")
 
+    task_orders = relationship("TaskOrder")
+
     @property
     def owner(self):
         def _is_workspace_owner(workspace_role):