diff --git a/atst/models/workspace.py b/atst/models/workspace.py index 126b6fd8..d02acd19 100644 --- a/atst/models/workspace.py +++ b/atst/models/workspace.py @@ -63,21 +63,3 @@ class Workspace(Base, mixins.TimestampsMixin, mixins.AuditableMixin): return "".format( self.name, self.request_id, self.user_count, self.id ) - - def _find_by_role(self, role): - try: - return [member for member in self.members if member.role.name == role] - except StopIteration: - return None - - @property - def contracting_officer(self): - return self._find_by_role("contracting_officer") - - @property - def contracting_officer_representative(self): - return self._find_by_role("contracting_officer_representative") - - @property - def security_officer(self): - return self._find_by_role("security_officer") diff --git a/atst/routes/task_orders/new.py b/atst/routes/task_orders/new.py index fb77cff6..9534fe12 100644 --- a/atst/routes/task_orders/new.py +++ b/atst/routes/task_orders/new.py @@ -161,7 +161,7 @@ class UpdateTaskOrderWorkflow(ShowTaskOrderWorkflow): if ( hasattr(self.form, field) and self.form[field].data - and not getattr(self.workspace, officer_type["role"]) + and not getattr(self.task_order, officer_type["role"]) ): prefix = officer_type["prefix"] officer_data = {