66 lines
2.7 KiB
Python
66 lines
2.7 KiB
Python
from flask import Blueprint, request, session, jsonify
|
|
|
|
from helix.calculators.calculator import Calculator
|
|
from helix.presenters.panel_presenter import ProjectPresenter
|
|
from helix.session_manager import SessionManager
|
|
from helix.constants import redis_constant, sql_constant
|
|
from helix.seismic_validator_user_values import SeismicValidatorUserValues
|
|
from helix.validators.file_validator import FileValidator
|
|
from helix.validators.seismic_anchor_validator import SeismicAnchorValidator
|
|
|
|
api = Blueprint('api', __name__, template_folder='templates')
|
|
|
|
|
|
@api.route("/panel_data")
|
|
def panel_data():
|
|
db_session = sql_constant.sql_session_maker()
|
|
session_manager = SessionManager(session, redis_constant.redis_store, db_session)
|
|
user_values = session_manager.user_values()
|
|
calculator = Calculator(user_values)
|
|
system_type = user_values.system_type()
|
|
module_type = user_values.module_type()
|
|
data = ProjectPresenter(system_type, module_type).get_panel_data(calculator.get_computed_csv_columns(), calculator.subarrays)
|
|
db_session.close()
|
|
return jsonify({'panel_data': data})
|
|
|
|
|
|
@api.route("/update_panel_data", methods=['POST'])
|
|
def update_panel_data():
|
|
db_session = sql_constant.sql_session_maker()
|
|
session_manager = SessionManager(session, redis_constant.redis_store, db_session)
|
|
user_seismic_data = request.get_json()
|
|
user_values = SeismicValidatorUserValues(session_manager.user_values(), user_seismic_data)
|
|
calculator = Calculator(user_values)
|
|
validator = SeismicAnchorValidator(calculator)
|
|
validation_result = validator.validate(calculator.panels)
|
|
|
|
subarrays = calculator.subarray_summary()
|
|
subarray_data = []
|
|
for subarray in subarrays:
|
|
data = {
|
|
"subarray": subarray.subarray_number,
|
|
"required_seismic_anchors": subarray.required_seismic_anchors,
|
|
"weight": round(subarray.weight),
|
|
}
|
|
subarray_data.append(data)
|
|
|
|
if not validation_result:
|
|
session_manager.save_user_provided_seismic_anchors(user_seismic_data)
|
|
panel_data = ProjectPresenter(user_values.system_type(), user_values.module_type()).get_panel_data(calculator.get_computed_csv_columns(),
|
|
calculator.subarrays)
|
|
db_session.close()
|
|
return jsonify({
|
|
"status": "success",
|
|
"error": None,
|
|
"panel_data": panel_data,
|
|
"subarray_data": subarray_data
|
|
})
|
|
else:
|
|
db_session.close()
|
|
return jsonify({
|
|
"status": "error",
|
|
"error": validation_result.value,
|
|
"panel_data": None,
|
|
"subarray_data": subarray_data
|
|
})
|