Remove legacy task order model

atst/domain/legacy_task_orders.py

@@ -1,60 +0,0 @@
-from sqlalchemy.orm.exc import NoResultFound
-from flask import current_app as app
-
-from atst.database import db
-from atst.models.legacy_task_order import LegacyTaskOrder, Source, FundingType
-from .exceptions import NotFoundError
-from atst.utils import update_obj
-
-
-class LegacyTaskOrders(object):
-    TASK_ORDER_DATA = [
-        col.name for col in LegacyTaskOrder.__table__.c if col.name != "id"
-    ]
-
-    @classmethod
-    def get(cls, order_number):
-        try:
-            legacy_task_order = (
-                db.session.query(LegacyTaskOrder).filter_by(number=order_number).one()
-            )
-        except NoResultFound:
-            if LegacyTaskOrders._client():
-                legacy_task_order = LegacyTaskOrders.get_from_eda(order_number)
-            else:
-                raise NotFoundError("legacy_task_order")
-
-        return legacy_task_order
-
-    @classmethod
-    def get_from_eda(cls, order_number):
-        to_data = LegacyTaskOrders._client().get_contract(order_number, status="y")
-        if to_data:
-            # TODO: we need to determine exactly what we're getting and storing from the EDA client
-            return LegacyTaskOrders.create(
-                source=Source.EDA, funding_type=FundingType.PROC, **to_data
-            )
-
-        else:
-            raise NotFoundError("legacy_task_order")
-
-    @classmethod
-    def create(cls, source=Source.MANUAL, **kwargs):
-        to_data = {k: v for k, v in kwargs.items() if v not in ["", None]}
-        legacy_task_order = LegacyTaskOrder(source=source, **to_data)
-
-        db.session.add(legacy_task_order)
-        db.session.commit()
-
-        return legacy_task_order
-
-    @classmethod
-    def _client(cls):
-        return app.eda_client
-
-    @classmethod
-    def update(cls, legacy_task_order, dct):
-        updated = update_obj(legacy_task_order, dct, ignore_vals=["", None])
-        db.session.add(updated)
-        db.session.commit()
-        return updated

atst/models/__init__.py

@@ -6,7 +6,6 @@ from .permissions import Permissions
 from .role import Role
 from .user import User
 from .portfolio_role import PortfolioRole
-from .legacy_task_order import LegacyTaskOrder
 from .portfolio import Portfolio
 from .application import Application
 from .environment import Environment

atst/models/legacy_task_order.py

@@ -1,75 +0,0 @@
-from enum import Enum
-
-from sqlalchemy import Column, Integer, String, ForeignKey, Enum as SQLAEnum, Date
-from sqlalchemy.orm import relationship
-
-from atst.models import Base, types, mixins
-
-
-class Source(Enum):
-    MANUAL = "Manual"
-    EDA = "EDA"
-
-
-class FundingType(Enum):
-    RDTE = "RDTE"
-    OM = "OM"
-    PROC = "PROC"
-    OTHER = "OTHER"
-
-
-class LegacyTaskOrder(Base, mixins.TimestampsMixin):
-    __tablename__ = "legacy_task_orders"
-
-    id = types.Id()
-    number = Column(String, unique=True)
-    source = Column(SQLAEnum(Source, native_enum=False))
-    funding_type = Column(SQLAEnum(FundingType, native_enum=False))
-    funding_type_other = Column(String)
-    clin_0001 = Column(Integer)
-    clin_0003 = Column(Integer)
-    clin_1001 = Column(Integer)
-    clin_1003 = Column(Integer)
-    clin_2001 = Column(Integer)
-    clin_2003 = Column(Integer)
-    expiration_date = Column(Date)
-
-    attachment_id = Column(ForeignKey("attachments.id"))
-    pdf = relationship("Attachment")
-
-    @property
-    def verified(self):
-        return self.source == Source.EDA
-
-    def to_dictionary(self):
-        return {
-            c.name: getattr(self, c.name)
-            for c in self.__table__.columns
-            if c.name not in ["id", "attachment_id"]
-        }
-
-    @property
-    def budget(self):
-        return sum(
-            filter(
-                None,
-                [
-                    self.clin_0001,
-                    self.clin_0003,
-                    self.clin_1001,
-                    self.clin_1003,
-                    self.clin_2001,
-                    self.clin_2003,
-                ],
-            )
-        )
-
-    def __repr__(self):  # pragma: no cover
-        return "<LegacyTaskOrder(number='{}', verified='{}', budget='{}', expiration_date='{}', pdf='{}', id='{}')>".format(
-            self.number,
-            self.verified,
-            self.budget,
-            self.expiration_date,
-            self.pdf,
-            self.id,
-        )