95 lines
2.7 KiB
Python
95 lines
2.7 KiB
Python
from sqlalchemy import and_, Column, String
|
|
from sqlalchemy.orm import relationship
|
|
from sqlalchemy.types import ARRAY
|
|
from itertools import chain
|
|
|
|
from atst.models import Base, Application, mixins, types
|
|
from atst.models.portfolio_role import PortfolioRole, Status as PortfolioRoleStatus
|
|
from atst.domain.permission_sets import PermissionSets
|
|
from atst.utils import first_or_none
|
|
from atst.database import db
|
|
|
|
|
|
class Portfolio(
|
|
Base, mixins.TimestampsMixin, mixins.AuditableMixin, mixins.DeletableMixin
|
|
):
|
|
__tablename__ = "portfolios"
|
|
|
|
id = types.Id()
|
|
name = Column(String)
|
|
defense_component = Column(String) # Department of Defense Component
|
|
|
|
app_migration = Column(String) # App Migration
|
|
complexity = Column(ARRAY(String)) # Application Complexity
|
|
complexity_other = Column(String)
|
|
description = Column(String)
|
|
dev_team = Column(ARRAY(String)) # Development Team
|
|
dev_team_other = Column(String)
|
|
native_apps = Column(String) # Native Apps
|
|
team_experience = Column(String) # Team Experience
|
|
|
|
applications = relationship(
|
|
"Application",
|
|
back_populates="portfolio",
|
|
primaryjoin=and_(Application.portfolio_id == id, Application.deleted == False),
|
|
)
|
|
roles = relationship("PortfolioRole")
|
|
|
|
task_orders = relationship("TaskOrder")
|
|
|
|
@property
|
|
def owner_role(self):
|
|
def _is_portfolio_owner(portfolio_role):
|
|
return PermissionSets.PORTFOLIO_POC in [
|
|
perms_set.name for perms_set in portfolio_role.permission_sets
|
|
]
|
|
|
|
return first_or_none(_is_portfolio_owner, self.roles)
|
|
|
|
@property
|
|
def owner(self):
|
|
owner_role = self.owner_role
|
|
return owner_role.user if owner_role else None
|
|
|
|
@property
|
|
def users(self):
|
|
return set(role.user for role in self.roles)
|
|
|
|
@property
|
|
def user_count(self):
|
|
return len(self.members)
|
|
|
|
@property
|
|
def num_task_orders(self):
|
|
return len(self.task_orders)
|
|
|
|
@property
|
|
def members(self):
|
|
return (
|
|
db.session.query(PortfolioRole)
|
|
.filter(PortfolioRole.portfolio_id == self.id)
|
|
.filter(PortfolioRole.status != PortfolioRoleStatus.DISABLED)
|
|
.all()
|
|
)
|
|
|
|
@property
|
|
def displayname(self):
|
|
return self.name
|
|
|
|
@property
|
|
def all_environments(self):
|
|
return list(chain.from_iterable(p.environments for p in self.applications))
|
|
|
|
@property
|
|
def portfolio_id(self):
|
|
return self.id
|
|
|
|
@property
|
|
def application_id(self):
|
|
return None
|
|
|
|
def __repr__(self):
|
|
return "<Portfolio(name='{}', user_count='{}', id='{}')>".format(
|
|
self.name, self.user_count, self.id
|
|
)
|