mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-21 16:01:15 -05:00
Made new /resend route in api.
Signed-off-by: Cliff Hill <Clifford.hill@gsa.gov>
This commit is contained in:
@@ -1,7 +1,7 @@
|
|||||||
from datetime import datetime, timedelta
|
from datetime import datetime, timedelta
|
||||||
|
|
||||||
from app import db
|
from app import db
|
||||||
from app.models import INVITE_EXPIRED, InvitedUser
|
from app.models import INVITE_EXPIRED, INVITE_PENDING, InvitedUser
|
||||||
|
|
||||||
|
|
||||||
def save_invited_user(invited_user):
|
def save_invited_user(invited_user):
|
||||||
@@ -31,7 +31,10 @@ def get_invited_users_for_service(service_id):
|
|||||||
def expire_invitations_created_more_than_two_days_ago():
|
def expire_invitations_created_more_than_two_days_ago():
|
||||||
expired = (
|
expired = (
|
||||||
db.session.query(InvitedUser)
|
db.session.query(InvitedUser)
|
||||||
.filter(InvitedUser.created_at <= datetime.utcnow() - timedelta(days=2))
|
.filter(
|
||||||
|
InvitedUser.created_at <= datetime.utcnow() - timedelta(days=2),
|
||||||
|
InvitedUser.status.in_(INVITE_PENDING),
|
||||||
|
)
|
||||||
.update({InvitedUser.status: INVITE_EXPIRED})
|
.update({InvitedUser.status: INVITE_EXPIRED})
|
||||||
)
|
)
|
||||||
db.session.commit()
|
db.session.commit()
|
||||||
|
|||||||
@@ -1,3 +1,7 @@
|
|||||||
|
from re import I
|
||||||
|
|
||||||
|
|
||||||
|
from datetime import datetime
|
||||||
from flask import Blueprint, current_app, jsonify, request
|
from flask import Blueprint, current_app, jsonify, request
|
||||||
from itsdangerous import BadData, SignatureExpired
|
from itsdangerous import BadData, SignatureExpired
|
||||||
from notifications_utils.url_safe_token import check_token, generate_token
|
from notifications_utils.url_safe_token import check_token, generate_token
|
||||||
@@ -12,7 +16,7 @@ from app.dao.invited_user_dao import (
|
|||||||
)
|
)
|
||||||
from app.dao.templates_dao import dao_get_template_by_id
|
from app.dao.templates_dao import dao_get_template_by_id
|
||||||
from app.errors import InvalidRequest, register_errors
|
from app.errors import InvalidRequest, register_errors
|
||||||
from app.models import EMAIL_TYPE, KEY_TYPE_NORMAL, Service
|
from app.models import EMAIL_TYPE, INVITE_PENDING, KEY_TYPE_NORMAL, Service
|
||||||
from app.notifications.process_notifications import (
|
from app.notifications.process_notifications import (
|
||||||
persist_notification,
|
persist_notification,
|
||||||
send_notification_to_queue,
|
send_notification_to_queue,
|
||||||
@@ -23,13 +27,7 @@ service_invite = Blueprint("service_invite", __name__)
|
|||||||
|
|
||||||
register_errors(service_invite)
|
register_errors(service_invite)
|
||||||
|
|
||||||
|
def _create_service_invite(invited_user, invite_link_host):
|
||||||
@service_invite.route("/service/<service_id>/invite", methods=["POST"])
|
|
||||||
def create_invited_user(service_id):
|
|
||||||
request_json = request.get_json()
|
|
||||||
invited_user = invited_user_schema.load(request_json)
|
|
||||||
save_invited_user(invited_user)
|
|
||||||
|
|
||||||
template_id = current_app.config["INVITATION_EMAIL_TEMPLATE_ID"]
|
template_id = current_app.config["INVITATION_EMAIL_TEMPLATE_ID"]
|
||||||
|
|
||||||
template = dao_get_template_by_id(template_id)
|
template = dao_get_template_by_id(template_id)
|
||||||
@@ -43,10 +41,7 @@ def create_invited_user(service_id):
|
|||||||
personalisation={
|
personalisation={
|
||||||
"user_name": invited_user.from_user.name,
|
"user_name": invited_user.from_user.name,
|
||||||
"service_name": invited_user.service.name,
|
"service_name": invited_user.service.name,
|
||||||
"url": invited_user_url(
|
"url": invited_user_url(invited_user.id, invite_link_host),
|
||||||
invited_user.id,
|
|
||||||
request_json.get("invite_link_host"),
|
|
||||||
),
|
|
||||||
},
|
},
|
||||||
notification_type=EMAIL_TYPE,
|
notification_type=EMAIL_TYPE,
|
||||||
api_key_id=None,
|
api_key_id=None,
|
||||||
@@ -56,6 +51,15 @@ def create_invited_user(service_id):
|
|||||||
|
|
||||||
send_notification_to_queue(saved_notification, queue=QueueNames.NOTIFY)
|
send_notification_to_queue(saved_notification, queue=QueueNames.NOTIFY)
|
||||||
|
|
||||||
|
|
||||||
|
@service_invite.route("/service/<service_id>/invite", methods=["POST"])
|
||||||
|
def create_invited_user(service_id):
|
||||||
|
request_json = request.get_json()
|
||||||
|
invited_user = invited_user_schema.load(request_json)
|
||||||
|
save_invited_user(invited_user)
|
||||||
|
|
||||||
|
_create_service_invite(invited_user, request_json.get("invite_link_host"))
|
||||||
|
|
||||||
return jsonify(data=invited_user_schema.dump(invited_user)), 201
|
return jsonify(data=invited_user_schema.dump(invited_user)), 201
|
||||||
|
|
||||||
|
|
||||||
@@ -92,6 +96,34 @@ def update_invited_user(service_id, invited_user_id):
|
|||||||
return jsonify(data=invited_user_schema.dump(fetched)), 200
|
return jsonify(data=invited_user_schema.dump(fetched)), 200
|
||||||
|
|
||||||
|
|
||||||
|
@service_invite.route(
|
||||||
|
"/service/<service_id>/invite/<invited_user_id>/resend", methods=["POST"]
|
||||||
|
)
|
||||||
|
def resend_service_invite(service_id, invited_user_id):
|
||||||
|
"""Resend an expired invite.
|
||||||
|
|
||||||
|
This resets the invited user's created date and status to make it a new invite, and
|
||||||
|
sends the new invite out to the user.
|
||||||
|
|
||||||
|
Note:
|
||||||
|
This ignores the POST data entirely.
|
||||||
|
"""
|
||||||
|
fetched = get_invited_user_by_service_and_id(
|
||||||
|
service_id=service_id, invited_user_id=invited_user_id
|
||||||
|
)
|
||||||
|
|
||||||
|
fetched.created_at = datetime.utcnow()
|
||||||
|
fetched.status = INVITE_PENDING
|
||||||
|
|
||||||
|
current_data = {k: v for k, v in invited_user_schema.dump(fetched).items()}
|
||||||
|
update_dict = invited_user_schema.load(current_data)
|
||||||
|
save_invited_user(update_dict)
|
||||||
|
|
||||||
|
_create_service_invite(fetched, current_app.config["ADMIN_BASE_URL"])
|
||||||
|
|
||||||
|
return jsonify(data=invited_user_schema.dump(fetched)), 200
|
||||||
|
|
||||||
|
|
||||||
def invited_user_url(invited_user_id, invite_link_host=None):
|
def invited_user_url(invited_user_id, invite_link_host=None):
|
||||||
token = generate_token(
|
token = generate_token(
|
||||||
str(invited_user_id),
|
str(invited_user_id),
|
||||||
|
|||||||
Reference in New Issue
Block a user