feat: remove `query_data` invocation, add `assetnum` parameter to `predict_run`, and refine fallback imports.

main
MrWaradana 3 weeks ago
parent 6ceb295781
commit 56ec7a1bad

@ -3,28 +3,28 @@ import time
# prefer package-relative imports, but allow running this file directly as a script
try:
from src.modules.equipment.insert_actual_data import query_data, insert_lcca_maximo_corrective_data, insert_ms_equipment_data, insert_acquisition_cost_data
from src.modules.equipment.Prediksi import Prediksi, main as predict_run
from src.modules.equipment.Eac import Eac, main as eac_run
from .insert_actual_data import query_data, insert_lcca_maximo_corrective_data, insert_ms_equipment_data, insert_acquisition_cost_data
from .Prediksi import Prediksi, main as predict_run
from .Eac import Eac, main as eac_run
except ImportError:
# fallback when there's no parent package (e.g., python run.py)
from insert_actual_data import query_data, insert_lcca_maximo_corrective_data, insert_ms_equipment_data, insert_acquisition_cost_data
from Prediksi import Prediksi, main as predict_run
from Eac import Eac, main as eac_run
from insert_actual_data import query_data
from Prediksi import main as predict_run
from Eac import main as eac_run
# Panggil fungsi
async def main():
start_time = time.time()
try:
await query_data()
except Exception as e:
print(f"Error in query_data: {str(e)}")
return
# try:
# await query_data()
# except Exception as e:
# print(f"Error in query_data: {str(e)}")
# return
try:
prediction_result = await predict_run()
prediction_result = await predict_run(assetnum="A22277")
if prediction_result is False:
print("Prediction step failed or was skipped. Skipping EAC run.")
return

Loading…
Cancel
Save