115 lines
3.6 KiB
Python
115 lines
3.6 KiB
Python
from enum import Enum
|
|
|
|
from sqlalchemy import (
|
|
Column,
|
|
Enum as SQLAEnum,
|
|
Numeric,
|
|
String,
|
|
ForeignKey,
|
|
Date,
|
|
Integer,
|
|
)
|
|
from sqlalchemy.types import ARRAY
|
|
from sqlalchemy.orm import relationship
|
|
|
|
from atst.models import Base, types, mixins
|
|
|
|
|
|
class Status(Enum):
|
|
PENDING = "Pending"
|
|
|
|
|
|
class TaskOrder(Base, mixins.TimestampsMixin):
|
|
__tablename__ = "task_orders"
|
|
|
|
id = types.Id()
|
|
|
|
workspace_id = Column(ForeignKey("workspaces.id"))
|
|
portfolio = relationship("Portfolio")
|
|
|
|
user_id = Column(ForeignKey("users.id"))
|
|
creator = relationship("User", foreign_keys="TaskOrder.user_id")
|
|
|
|
ko_id = Column(ForeignKey("users.id"))
|
|
contracting_officer = relationship("User", foreign_keys="TaskOrder.ko_id")
|
|
|
|
cor_id = Column(ForeignKey("users.id"))
|
|
contracting_officer_representative = relationship(
|
|
"User", foreign_keys="TaskOrder.cor_id"
|
|
)
|
|
|
|
so_id = Column(ForeignKey("users.id"))
|
|
security_officer = relationship("User", foreign_keys="TaskOrder.so_id")
|
|
|
|
status = Column(SQLAEnum(Status, native_enum=False))
|
|
|
|
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)) # Application 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)
|
|
performance_length = Column(Integer)
|
|
attachment_id = Column(ForeignKey("attachments.id"))
|
|
pdf = relationship("Attachment")
|
|
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_phone_number = Column(String) # Phone Number
|
|
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_phone_number = Column(String) # Phone Number
|
|
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_phone_number = Column(String) # Phone Number
|
|
so_dod_id = Column(String) # DOD ID
|
|
number = Column(String, unique=True) # Task Order Number
|
|
loa = Column(ARRAY(String)) # Line of Accounting (LOA)
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
if "status" not in kwargs:
|
|
self.status = Status.PENDING
|
|
|
|
@property
|
|
def budget(self):
|
|
return sum(
|
|
filter(None, [self.clin_01, self.clin_02, self.clin_03, self.clin_04])
|
|
)
|
|
|
|
@property
|
|
def portfolio_name(self):
|
|
return self.portfolio.name
|
|
|
|
@property
|
|
def is_pending(self):
|
|
return self.status == Status.PENDING
|
|
|
|
def to_dictionary(self):
|
|
return {
|
|
"portfolio_name": self.portfolio_name,
|
|
**{
|
|
c.name: getattr(self, c.name)
|
|
for c in self.__table__.columns
|
|
if c.name not in ["id"]
|
|
},
|
|
}
|
|
|
|
def __repr__(self):
|
|
return "<TaskOrder(number='{}', budget='{}', end_date='{}', id='{}')>".format(
|
|
self.number, self.budget, self.end_date, self.id
|
|
)
|