mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-23 00:41:35 -05:00
32 lines
893 B
Python
32 lines
893 B
Python
from datetime import datetime, timedelta
|
|
|
|
from app import db
|
|
from app.models import INVITE_EXPIRED, InvitedUser
|
|
|
|
|
|
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):
|
|
return InvitedUser.query.filter_by(service_id=service_id, id=invited_user_id).one()
|
|
|
|
|
|
def get_invited_user_by_id(invited_user_id):
|
|
return InvitedUser.query.filter_by(id=invited_user_id).one()
|
|
|
|
|
|
def get_invited_users_for_service(service_id):
|
|
return InvitedUser.query.filter_by(service_id=service_id).all()
|
|
|
|
|
|
def expire_invitations_created_more_than_two_days_ago():
|
|
expired = (
|
|
db.session.query(InvitedUser)
|
|
.filter(InvitedUser.created_at <= datetime.utcnow() - timedelta(days=2))
|
|
.update({InvitedUser.status: INVITE_EXPIRED})
|
|
)
|
|
db.session.commit()
|
|
return expired
|