mirror of
https://github.com/GSA/notifications-api.git
synced 2026-01-08 19:54:13 -05:00
61 lines
1.8 KiB
Python
61 lines
1.8 KiB
Python
from datetime import timedelta
|
|
|
|
from sqlalchemy import select
|
|
|
|
from app import db
|
|
from app.enums import InvitedUserStatus
|
|
from app.models import InvitedUser
|
|
from app.utils import utc_now
|
|
|
|
|
|
def save_invited_user(invited_user):
|
|
db.session.add(invited_user)
|
|
db.session.commit()
|
|
|
|
|
|
def get_invited_user_by_service_and_id(service_id, invited_user_id):
|
|
|
|
stmt = select(InvitedUser).where(
|
|
InvitedUser.service_id == service_id,
|
|
InvitedUser.id == invited_user_id,
|
|
)
|
|
return db.session.execute(stmt).scalars().one()
|
|
|
|
|
|
def get_expired_invite_by_service_and_id(service_id, invited_user_id):
|
|
stmt = select(InvitedUser).where(
|
|
InvitedUser.service_id == service_id,
|
|
InvitedUser.id == invited_user_id,
|
|
InvitedUser.status == InvitedUserStatus.EXPIRED,
|
|
)
|
|
return db.session.execute(stmt).scalars().one()
|
|
|
|
|
|
def get_invited_user_by_id(invited_user_id):
|
|
stmt = select(InvitedUser).where(InvitedUser.id == invited_user_id)
|
|
return db.session.execute(stmt).scalars().one()
|
|
|
|
|
|
def get_expired_invited_users_for_service(service_id):
|
|
# TODO why does this return all invited users?
|
|
stmt = select(InvitedUser).where(InvitedUser.service_id == service_id)
|
|
return db.session.execute(stmt).scalars().all()
|
|
|
|
|
|
def get_invited_users_for_service(service_id):
|
|
stmt = select(InvitedUser).where(InvitedUser.service_id == service_id)
|
|
return db.session.execute(stmt).scalars().all()
|
|
|
|
|
|
def expire_invitations_created_more_than_two_days_ago():
|
|
expired = (
|
|
db.session.query(InvitedUser)
|
|
.where(
|
|
InvitedUser.created_at <= utc_now() - timedelta(days=2),
|
|
InvitedUser.status.in_((InvitedUserStatus.PENDING,)),
|
|
)
|
|
.update({InvitedUser.status: InvitedUserStatus.EXPIRED})
|
|
)
|
|
db.session.commit()
|
|
return expired
|