add reset path

main
Cizz22 5 months ago
parent 2c6292fddf
commit afe23f4958

@ -399,7 +399,7 @@ async def update_equipment_for_simulation(*, db_session: DbSession, project_name
reqNodeInputs.append(eq)
print("Updating equipment for simulation")
await update_node(db_session=db_session, equipment_nodes=reqNodeInputs, project_name=project_name)
## await update_node(db_session=db_session, equipment_nodes=reqNodeInputs, project_name=project_name)
print("Updated equipment for simulation")
return results

@ -9,7 +9,7 @@ from src.database.service import CommonParameters
from src.models import StandardResponse
from .schema import AerosProjectInput, AerosMetadata
from .service import import_aro_project, fetch_aro_record
from .service import import_aro_project, fetch_aro_record, reset_project
router = APIRouter()
@ -34,6 +34,17 @@ async def getAerosMetadata(db_session: DbSession):
return {"data": result, "status": "success", "message": "Success"}
@router.get("/reset", response_model=StandardResponse[None])
async def reset_aeros_project(db_session: DbSession):
# Logic to reset the ARO project
reset_path = await reset_project(db_session=db_session)
return {
"data": reset_path,
"message": "Success"
}
# @router.post("", response_model=StandardResponse[None])
# async def import_aro_old(db_session: DbSession, aeros_project_in: AerosProjectInput, project_name: str = "trialapi"):
# # TEST ONLY

@ -181,3 +181,21 @@ async def _initialize_default_project_data(
except Exception as e:
await db_session.rollback()
raise e
def reset_project(*, db_session: DbSession):
project = await fetch_aro_record(db_session=db_session)
try:
response = await client.post(
f"{AEROS_BASE_URL}/api/Project/ImportAROFile",
content=f'"{project.aro_file_path}"',
headers={"Content-Type": "application/json"},
)
response.raise_for_status()
except Exception as e:
raise HTTPException(
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, detail=str(e)
)
return project.aro_file_path

Loading…
Cancel
Save