21 lines
702 B
Python
21 lines
702 B
Python
from app import db
|
|
from app.models.notification import Notification
|
|
# from app.models.users import Member
|
|
|
|
def notify(user, message, href=None):
|
|
""" Notify a user (by id, name or object reference) with a message.
|
|
An hyperlink can be added to redirect to the notification source """
|
|
|
|
# Cuz' duck typing is quite cool
|
|
# TODO: maybe abort if no user is found
|
|
if type(user) == str:
|
|
user = Member.query.filter_by(name=user).first()
|
|
if isinstance(user, Member):
|
|
user = user.id
|
|
if user and Member.query.get(user):
|
|
n = Notification(user, message, href=href)
|
|
db.session.add(n)
|
|
db.session.commit()
|
|
else:
|
|
print("User not found")
|