Add status to task order table

This commit is contained in:
Patrick Smith
2019-01-10 14:02:08 -05:00
parent 27eb15fc2d
commit dda6db8562
3 changed files with 45 additions and 2 deletions

View File

@@ -1,10 +1,16 @@
from sqlalchemy import Column, Numeric, String, ForeignKey, Date
from enum import Enum
from sqlalchemy import Column, Enum as SQLAEnum, Numeric, String, ForeignKey, Date
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"
@@ -27,6 +33,8 @@ class TaskOrder(Base, mixins.TimestampsMixin):
so_id = Column(ForeignKey("users.id"))
security_officer = relationship("User", foreign_keys="TaskOrder.so_id")
status = Column(SQLAEnum(Status, native_enum=False, default=Status.PENDING))
scope = Column(String) # Cloud Project Scope
defense_component = Column(String) # Department of Defense Component
app_migration = Column(String) # App Migration
@@ -67,6 +75,10 @@ class TaskOrder(Base, mixins.TimestampsMixin):
def portfolio_name(self):
return self.workspace.name
@property
def is_pending(self):
return self.status == Status.PENDING
def to_dictionary(self):
return {
"portfolio_name": self.portfolio_name,