from flask import Blueprint, jsonify, request from app.dao.provider_details_dao import ( dao_get_provider_stats, dao_get_provider_versions, dao_update_provider_details, get_provider_details_by_id, ) from app.dao.users_dao import get_user_by_id from app.errors import InvalidRequest, register_errors from app.schemas import provider_details_history_schema, provider_details_schema provider_details = Blueprint("provider_details", __name__) register_errors(provider_details) @provider_details.route("", methods=["GET"]) def get_providers(): data = dao_get_provider_stats() provider_details = [ { "id": row.id, "display_name": row.display_name, "identifier": row.identifier, "notification_type": row.notification_type, "active": row.active, "updated_at": row.updated_at, "supports_international": row.supports_international, "created_by_name": row.created_by_name, "current_month_billable_sms": row.current_month_billable_sms, } for row in data ] return jsonify(provider_details=provider_details) @provider_details.route("/", methods=["GET"]) def get_provider_by_id(provider_details_id): data = provider_details_schema.dump(get_provider_details_by_id(provider_details_id)) return jsonify(provider_details=data) @provider_details.route("//versions", methods=["GET"]) def get_provider_versions(provider_details_id): versions = dao_get_provider_versions(provider_details_id) data = provider_details_history_schema.dump(versions, many=True) return jsonify(data=data) @provider_details.route("/", methods=["POST"]) def update_provider_details(provider_details_id): valid_keys = {"priority", "created_by", "active"} req_json = request.get_json() invalid_keys = req_json.keys() - valid_keys if invalid_keys: message = "Not permitted to be updated" errors = {key: [message] for key in invalid_keys} raise InvalidRequest(errors, status_code=400) provider = get_provider_details_by_id(provider_details_id) # Handle created_by differently due to how history entry is created if "created_by" in req_json: user = get_user_by_id(req_json["created_by"]) provider.created_by_id = user.id req_json.pop("created_by") for key in req_json: setattr(provider, key, req_json[key]) dao_update_provider_details(provider) return jsonify(provider_details=provider_details_schema.dump(provider)), 200