def store_main_model_in_db(model, scaler_X, scaler_y):
database = Database()
database.create_table(CREATE_MODEL_QUERY)
database.drop_table(DELETE_ALL_MODELS_QUERY)
pickled_tuple = ModelStorage.__get_tuple_pickled(
(model, scaler_X, scaler_y))
database.insert_data(INSERT_MODEL_QUERY, pickled_tuple)
database.close_connection()
def _insert(self, query: str, data: List[str]):
tuple_data = tuple(data)
try:
database = Database()
_id = database.insert_data(query, tuple_data)
self._response = self._build_response(data=jsonify({
"message": "Successful POST request",
"_id": _id
}),
status_code=201)
except psycopg2.errors.UniqueViolation:
self._response = self._build_response(data=jsonify(
{"message": "Item already exists"}),
status_code=400)
except:
self._response = self._build_response(data=jsonify(
{"message": "Internal server error"}),
status_code=500)
finally:
database.close_connection()