from flask import Blueprint, abort, current_app, jsonify, request from sqlalchemy.exc import IntegrityError from app.config import QueueNames from app.dao.annual_billing_dao import set_default_free_allowance_for_service from app.dao.dao_utils import transaction from app.dao.fact_billing_dao import fetch_usage_year_for_organization from app.dao.organization_dao import ( dao_add_service_to_organization, dao_add_user_to_organization, dao_create_organization, dao_get_organization_by_email_address, dao_get_organization_by_id, dao_get_organization_services, dao_get_organizations, dao_get_users_for_organization, dao_remove_user_from_organization, dao_update_organization, ) from app.dao.services_dao import dao_fetch_service_by_id from app.dao.templates_dao import dao_get_template_by_id from app.dao.users_dao import get_user_by_id from app.errors import InvalidRequest, register_errors from app.models import KEY_TYPE_NORMAL, Organization from app.notifications.process_notifications import ( persist_notification, send_notification_to_queue, ) from app.organization.organization_schema import ( post_create_organization_schema, post_link_service_to_organization_schema, post_update_organization_schema, ) from app.schema_validation import validate organization_blueprint = Blueprint('organization', __name__) register_errors(organization_blueprint) @organization_blueprint.errorhandler(IntegrityError) def handle_integrity_error(exc): """ Handle integrity errors caused by the unique constraint on ix_organization_name """ if 'ix_organization_name' in str(exc): return jsonify(result="error", message="Organization name already exists"), 400 if 'duplicate key value violates unique constraint "domain_pkey"' in str(exc): return jsonify(result='error', message='Domain already exists'), 400 current_app.logger.exception(exc) return jsonify(result='error', message="Internal server error"), 500 @organization_blueprint.route('', methods=['GET']) def get_organizations(): organizations = [ org.serialize_for_list() for org in dao_get_organizations() ] return jsonify(organizations) @organization_blueprint.route('/', methods=['GET']) def get_organization_by_id(organization_id): organization = dao_get_organization_by_id(organization_id) return jsonify(organization.serialize()) @organization_blueprint.route('/by-domain', methods=['GET']) def get_organization_by_domain(): domain = request.args.get('domain') if not domain or '@' in domain: abort(400) organization = dao_get_organization_by_email_address( 'example@{}'.format(request.args.get('domain')) ) if not organization: abort(404) return jsonify(organization.serialize()) @organization_blueprint.route('', methods=['POST']) def create_organization(): data = request.get_json() validate(data, post_create_organization_schema) organization = Organization(**data) dao_create_organization(organization) return jsonify(organization.serialize()), 201 @organization_blueprint.route('/', methods=['POST']) def update_organization(organization_id): data = request.get_json() validate(data, post_update_organization_schema) result = dao_update_organization(organization_id, **data) if data.get('agreement_signed') is True: # if a platform admin has manually adjusted the organization, don't tell people if data.get('agreement_signed_by_id'): send_notifications_on_mou_signed(organization_id) if result: return '', 204 else: raise InvalidRequest("Organization not found", 404) @organization_blueprint.route('//service', methods=['POST']) def link_service_to_organization(organization_id): data = request.get_json() validate(data, post_link_service_to_organization_schema) service = dao_fetch_service_by_id(data['service_id']) service.organization = None with transaction(): dao_add_service_to_organization(service, organization_id) set_default_free_allowance_for_service(service, year_start=None) return '', 204 @organization_blueprint.route('//services', methods=['GET']) def get_organization_services(organization_id): services = dao_get_organization_services(organization_id) sorted_services = sorted(services, key=lambda s: (-s.active, s.name)) return jsonify([s.serialize_for_org_dashboard() for s in sorted_services]) @organization_blueprint.route('//services-with-usage', methods=['GET']) def get_organization_services_usage(organization_id): try: year = int(request.args.get('year', 'none')) except ValueError: return jsonify(result='error', message='No valid year provided'), 400 services = fetch_usage_year_for_organization(organization_id, year) list_services = services.values() sorted_services = sorted(list_services, key=lambda s: (-s['active'], s['service_name'].lower())) return jsonify(services=sorted_services) @organization_blueprint.route('//users/', methods=['POST']) def add_user_to_organization(organization_id, user_id): new_org_user = dao_add_user_to_organization(organization_id, user_id) return jsonify(data=new_org_user.serialize()) @organization_blueprint.route('//users/', methods=['DELETE']) def remove_user_from_organization(organization_id, user_id): organization = dao_get_organization_by_id(organization_id) user = get_user_by_id(user_id=user_id) if user not in organization.users: error = 'User not found' raise InvalidRequest(error, status_code=404) dao_remove_user_from_organization(organization, user) return {}, 204 @organization_blueprint.route('//users', methods=['GET']) def get_organization_users(organization_id): org_users = dao_get_users_for_organization(organization_id) return jsonify(data=[x.serialize() for x in org_users]) def check_request_args(request): org_id = request.args.get('org_id') name = request.args.get('name', None) errors = [] if not org_id: errors.append({'org_id': ["Can't be empty"]}) if not name: errors.append({'name': ["Can't be empty"]}) if errors: raise InvalidRequest(errors, status_code=400) return org_id, name def send_notifications_on_mou_signed(organization_id): organization = dao_get_organization_by_id(organization_id) notify_service = dao_fetch_service_by_id(current_app.config['NOTIFY_SERVICE_ID']) def _send_notification(template_id, recipient, personalisation): template = dao_get_template_by_id(template_id) saved_notification = persist_notification( template_id=template.id, template_version=template.version, recipient=recipient, service=notify_service, personalisation=personalisation, notification_type=template.template_type, api_key_id=None, key_type=KEY_TYPE_NORMAL, reply_to_text=notify_service.get_default_reply_to_email_address() ) send_notification_to_queue(saved_notification, research_mode=False, queue=QueueNames.NOTIFY) personalisation = { 'mou_link': '{}/agreement/agreement.pdf'.format( current_app.config['ADMIN_BASE_URL'] ), 'org_name': organization.name, 'org_dashboard_link': '{}/organizations/{}'.format( current_app.config['ADMIN_BASE_URL'], organization.id ), 'signed_by_name': organization.agreement_signed_by.name, 'on_behalf_of_name': organization.agreement_signed_on_behalf_of_name } if not organization.agreement_signed_on_behalf_of_email_address: signer_template_id = 'MOU_SIGNER_RECEIPT_TEMPLATE_ID' else: signer_template_id = 'MOU_SIGNED_ON_BEHALF_SIGNER_RECEIPT_TEMPLATE_ID' # let the person who has been signed on behalf of know. _send_notification( current_app.config['MOU_SIGNED_ON_BEHALF_ON_BEHALF_RECEIPT_TEMPLATE_ID'], organization.agreement_signed_on_behalf_of_email_address, personalisation ) # let the person who signed know - the template is different depending on if they signed on behalf of someone _send_notification( current_app.config[signer_template_id], organization.agreement_signed_by.email_address, personalisation )