Remove PE number model
This commit is contained in:
@@ -1,24 +0,0 @@
|
||||
from sqlalchemy.dialects.postgresql import insert
|
||||
|
||||
from atst.database import db
|
||||
from atst.models.pe_number import PENumber
|
||||
from .exceptions import NotFoundError
|
||||
|
||||
|
||||
class PENumbers(object):
|
||||
@classmethod
|
||||
def get(cls, number):
|
||||
pe_number = db.session.query(PENumber).get(number)
|
||||
if not pe_number:
|
||||
raise NotFoundError("pe_number")
|
||||
|
||||
return pe_number
|
||||
|
||||
@classmethod
|
||||
def create_many(cls, list_of_pe_numbers):
|
||||
stmt = insert(PENumber).values(list_of_pe_numbers)
|
||||
do_update = stmt.on_conflict_do_update(
|
||||
index_elements=["number"], set_=dict(description=stmt.excluded.description)
|
||||
)
|
||||
db.session.execute(do_update)
|
||||
db.session.commit()
|
@@ -6,7 +6,6 @@ from .permissions import Permissions
|
||||
from .role import Role
|
||||
from .user import User
|
||||
from .portfolio_role import PortfolioRole
|
||||
from .pe_number import PENumber
|
||||
from .legacy_task_order import LegacyTaskOrder
|
||||
from .portfolio import Portfolio
|
||||
from .application import Application
|
||||
|
@@ -1,15 +0,0 @@
|
||||
from sqlalchemy import String, Column
|
||||
|
||||
from atst.models import Base
|
||||
|
||||
|
||||
class PENumber(Base):
|
||||
__tablename__ = "pe_numbers"
|
||||
|
||||
number = Column(String, primary_key=True)
|
||||
description = Column(String)
|
||||
|
||||
def __repr__(self): # pragma: no cover
|
||||
return "<PENumber(number='{}', description='{}')>".format(
|
||||
self.number, self.description
|
||||
)
|
Reference in New Issue
Block a user