You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
152 lines
4.6 KiB
Python
152 lines
4.6 KiB
Python
from enum import Enum
|
|
from typing import Optional
|
|
|
|
from sqlalchemy import (JSON, UUID, Boolean, Column, Float, ForeignKey,
|
|
Integer, String)
|
|
from sqlalchemy.orm import relationship
|
|
|
|
from src.database.core import Base, DbSession
|
|
from src.models import DefaultMixin, IdentityMixin
|
|
|
|
|
|
class OverhaulReferenceType(str, Enum):
|
|
SCOPE = "SCOPE"
|
|
ASSET = "ASSET"
|
|
|
|
|
|
class CalculationParam(Base, DefaultMixin, IdentityMixin):
|
|
__tablename__ = "oh_ms_calculation_param"
|
|
|
|
avg_failure_cost = Column(Float, nullable=False)
|
|
overhaul_cost = Column(Float, nullable=False)
|
|
|
|
# Relationships
|
|
calculation_data = relationship("CalculationData", back_populates="parameter")
|
|
results = relationship("CalculationResult", back_populates="parameter")
|
|
|
|
# @classmethod
|
|
# async def create_with_references(
|
|
# cls,
|
|
# db: DbSession,
|
|
# avg_failure_cost: float,
|
|
# overhaul_cost: float,
|
|
# created_by: str,
|
|
# # list of {"reference_type": OverhaulReferenceType, "reference_id": str}
|
|
# ):
|
|
# # Create parameter
|
|
|
|
# # Create reference links
|
|
# for ref in references:
|
|
|
|
|
|
|
|
class CalculationData(Base, DefaultMixin, IdentityMixin):
|
|
__tablename__ = "oh_tr_calculation_data"
|
|
|
|
parameter_id = Column(
|
|
UUID(as_uuid=True), ForeignKey("oh_ms_calculation_param.id"), nullable=True
|
|
)
|
|
overhaul_session_id = Column(
|
|
UUID(as_uuid=True), ForeignKey("oh_ms_overhaul.id")
|
|
)
|
|
optimum_oh_day = Column(Integer, nullable=True)
|
|
|
|
max_interval = Column(Integer, nullable=True)
|
|
|
|
rbd_simulation_id = Column(UUID(as_uuid=True), nullable=True)
|
|
|
|
optimum_analysis = Column(JSON, nullable=True)
|
|
|
|
session = relationship("OverhaulScope", lazy="raise")
|
|
|
|
parameter = relationship("CalculationParam", back_populates="calculation_data")
|
|
|
|
equipment_results = relationship(
|
|
"CalculationEquipmentResult", lazy="raise", viewonly=True
|
|
)
|
|
|
|
|
|
results = relationship("CalculationResult", lazy="raise", viewonly=True)
|
|
|
|
|
|
@classmethod
|
|
async def create_with_param(
|
|
cls,
|
|
overhaul_session_id: str,
|
|
db: DbSession,
|
|
avg_failure_cost: Optional[float],
|
|
overhaul_cost: Optional[float],
|
|
created_by: str,
|
|
params_id: Optional[UUID],
|
|
):
|
|
if not params_id:
|
|
# Create Params
|
|
params = CalculationParam(
|
|
avg_failure_cost=avg_failure_cost,
|
|
overhaul_cost=overhaul_cost,
|
|
created_by=created_by,
|
|
)
|
|
|
|
db.add(params)
|
|
await db.flush()
|
|
params_id = params.id
|
|
|
|
calculation_data = cls(
|
|
overhaul_session_id=overhaul_session_id,
|
|
created_by=created_by,
|
|
parameter_id=params_id,
|
|
)
|
|
|
|
db.add(calculation_data)
|
|
|
|
await db.commit()
|
|
await db.refresh(calculation_data)
|
|
|
|
return calculation_data
|
|
|
|
|
|
class CalculationResult(Base, DefaultMixin):
|
|
|
|
__tablename__ = "oh_tr_calculation_result"
|
|
|
|
parameter_id = Column(
|
|
UUID(as_uuid=True), ForeignKey("oh_ms_calculation_param.id"), nullable=False
|
|
)
|
|
calculation_data_id = Column(
|
|
UUID(as_uuid=True), ForeignKey("oh_tr_calculation_data.id"), nullable=False
|
|
)
|
|
day = Column(Integer, nullable=False)
|
|
corrective_cost = Column(Float, nullable=False)
|
|
overhaul_cost = Column(Float, nullable=False)
|
|
num_failures = Column(Integer, nullable=False)
|
|
|
|
parameter = relationship("CalculationParam", back_populates="results")
|
|
reference_link = relationship("CalculationData")
|
|
|
|
|
|
class CalculationEquipmentResult(Base, DefaultMixin):
|
|
|
|
__tablename__ = "oh_tr_calculation_equipment_result"
|
|
|
|
corrective_costs = Column(JSON, nullable=False)
|
|
overhaul_costs = Column(JSON, nullable=False)
|
|
daily_failures = Column(JSON, nullable=False)
|
|
procurement_costs = Column(JSON, nullable=False)
|
|
location_tag = Column(String(255), nullable=False)
|
|
material_cost = Column(Float, nullable=False)
|
|
service_cost = Column(Float, nullable=False)
|
|
calculation_data_id = Column(
|
|
UUID(as_uuid=True), ForeignKey("oh_tr_calculation_data.id"), nullable=True
|
|
)
|
|
optimum_day = Column(Integer, default=1)
|
|
is_included = Column(Boolean, default=True)
|
|
procurement_details = Column(JSON, nullable=True)
|
|
is_initial = Column(Boolean, default=True)
|
|
|
|
master_equipment = relationship(
|
|
"MasterEquipment",
|
|
lazy="joined",
|
|
primaryjoin="and_(CalculationEquipmentResult.location_tag == foreign(MasterEquipment.location_tag))",
|
|
uselist=False, # Add this if it's a one-to-one relationship
|
|
)
|