Rename triggerNotif to addNotification, add array support

5 files changed, 19 insertions, 20 deletions

diff --git a/app/blueprints/packages/editrequests.py b/app/blueprints/packages/editrequests.py
index ee0d162..f50a812 100644
--- a/app/blueprints/packages/editrequests.py
+++ b/app/blueprints/packages/editrequests.py
@@ -99,8 +99,8 @@ def create_edit_editrequest_page(package, id=None):
 		if wasChangeMade:
 			msg = "{}: Edit request #{} {}" \
 					.format(package.title, erequest.id, "created" if id is None else "edited")
-			triggerNotif(package.author, current_user, msg, erequest.getURL())
-			triggerNotif(erequest.author, current_user, msg, erequest.getURL())
+			addNotification(package.maintainers, current_user, msg, erequest.getURL())
+			addNotification(erequest.author, current_user, msg, erequest.getURL())
 			db.session.commit()
 			return redirect(erequest.getURL())
 		else:
@@ -142,8 +142,8 @@ def approve_editrequest_page(package, id):
 		erequest.applyAll(package)
 
 		msg = "{}: Edit request #{} merged".format(package.title, erequest.id)
-		triggerNotif(erequest.author, current_user, msg, erequest.getURL())
-		triggerNotif(package.author, current_user, msg, erequest.getURL())
+		addNotification(erequest.author, current_user, msg, erequest.getURL())
+		addNotification(package.maintainers, current_user, msg, erequest.getURL())
 		db.session.commit()
 
 	return redirect(package.getDetailsURL())
@@ -166,8 +166,8 @@ def reject_editrequest_page(package, id):
 		erequest.status = 2
 
 		msg = "{}: Edit request #{} rejected".format(package.title, erequest.id)
-		triggerNotif(erequest.author, current_user, msg, erequest.getURL())
-		triggerNotif(package.author, current_user, msg, erequest.getURL())
+		addNotification(erequest.author, current_user, msg, erequest.getURL())
+		addNotification(package.maintainers, current_user, msg, erequest.getURL())
 		db.session.commit()
 
 	return redirect(package.getDetailsURL())
diff --git a/app/blueprints/packages/packages.py b/app/blueprints/packages/packages.py
index 4ba2a73..bfa3dd9 100644
--- a/app/blueprints/packages/packages.py
+++ b/app/blueprints/packages/packages.py
@@ -265,7 +265,7 @@ def create_edit(author=None, name=None):
 			return redirect(url_for("packages.create_edit", author=author, name=name))
 
 		else:
-			triggerNotif(package.author, current_user,
+			addNotification(package.maintainers, current_user,
 					"{} edited".format(package.title), package.getDetailsURL())
 
 		form.populate_obj(package) # copy to row
@@ -337,7 +337,7 @@ def approve(package):
 		for s in screenshots:
 			s.approved = True
 
-		triggerNotif(package.author, current_user,
+		addNotification(package.maintainers, current_user,
 				"{} approved".format(package.title), package.getDetailsURL())
 		db.session.commit()
 
@@ -359,7 +359,7 @@ def remove(package):
 		package.soft_deleted = True
 
 		url = url_for("users.profile", username=package.author.username)
-		triggerNotif(package.author, current_user,
+		addNotification(package.maintainers, current_user,
 				"{} deleted".format(package.title), url)
 		db.session.commit()
 
@@ -373,7 +373,7 @@ def remove(package):
 
 		package.approved = False
 
-		triggerNotif(package.author, current_user,
+		addNotification(package.maintainers, current_user,
 				"{} unapproved".format(package.title), package.getDetailsURL())
 		db.session.commit()
 
@@ -408,19 +408,19 @@ def edit_maintainers(package):
 
 		for user in users:
 			if not user in package.maintainers:
-				triggerNotif(user, current_user,
+				addNotification(user, current_user,
 						"Added you as a maintainer of {}".format(package.title), package.getDetailsURL())
 
 		for user in package.maintainers:
 			if user != package.author and not user in users:
-				triggerNotif(user, current_user,
+				addNotification(user, current_user,
 						"Removed you as a maintainer of {}".format(package.title), package.getDetailsURL())
 
 		package.maintainers.clear()
 		package.maintainers.extend(users)
 		package.maintainers.append(package.author)
 
-		triggerNotif(package.author, current_user,
+		addNotification(package.author, current_user,
 				"Edited {} maintainers".format(package.title), package.getDetailsURL())
 
 		db.session.commit()
@@ -446,7 +446,7 @@ def remove_self_maintainers(package):
 	else:
 		package.maintainers.remove(current_user)
 
-		triggerNotif(package.author, current_user,
+		addNotification(package.author, current_user,
 				"Removed themself as a maintainer of {}".format(package.title), package.getDetailsURL())
 
 		db.session.commit()
diff --git a/app/blueprints/packages/releases.py b/app/blueprints/packages/releases.py
index 12be8f6..4498738 100644
--- a/app/blueprints/packages/releases.py
+++ b/app/blueprints/packages/releases.py
@@ -93,7 +93,7 @@ def create_release(package):
 			makeVCSRelease.apply_async((rel.id, form["vcsLabel"].data), task_id=rel.task_id)
 
 			msg = "{}: Release {} created".format(package.title, rel.title)
-			triggerNotif(package.author, current_user, msg, rel.getEditURL())
+			addNotification(package.maintainers, current_user, msg, rel.getEditURL())
 			db.session.commit()
 
 			return redirect(url_for("tasks.check", id=rel.task_id, r=rel.getEditURL()))
@@ -113,7 +113,7 @@ def create_release(package):
 				checkZipRelease.apply_async((rel.id, uploadedPath), task_id=rel.task_id)
 
 				msg = "{}: Release {} created".format(package.title, rel.title)
-				triggerNotif(package.author, current_user, msg, rel.getEditURL())
+				addNotification(package.maintainers, current_user, msg, rel.getEditURL())
 				db.session.commit()
 
 				return redirect(url_for("tasks.check", id=rel.task_id, r=rel.getEditURL()))
diff --git a/app/blueprints/packages/reviews.py b/app/blueprints/packages/reviews.py
index 4322e8d..114afae 100644
--- a/app/blueprints/packages/reviews.py
+++ b/app/blueprints/packages/reviews.py
@@ -22,7 +22,7 @@ from flask_wtf import FlaskForm
 from wtforms import *
 from wtforms.validators import *
 from app.models import db, PackageReview, Thread, ThreadReply
-from app.utils import is_package_page, triggerNotif
+from app.utils import is_package_page, addNotification
 
 class ReviewForm(FlaskForm):
 	title	= StringField("Title", [InputRequired(), Length(3,100)])
@@ -88,8 +88,7 @@ def review(package):
 		else:
 			notif_msg = "Updated review '{}' on package {}".format(form.title.data, package.title)
 
-		for maintainer in package.maintainers:
-			triggerNotif(maintainer, current_user, notif_msg, url_for("threads.view", id=thread.id))
+		addNotification(package.maintainers, current_user, notif_msg, url_for("threads.view", id=thread.id))
 
 		db.session.commit()
 
diff --git a/app/blueprints/packages/screenshots.py b/app/blueprints/packages/screenshots.py
index 2646e29..68a3075 100644
--- a/app/blueprints/packages/screenshots.py
+++ b/app/blueprints/packages/screenshots.py
@@ -62,7 +62,7 @@ def create_screenshot(package, id=None):
 
 			msg = "{}: Screenshot added {}" \
 					.format(package.title, ss.title)
-			triggerNotif(package.author, current_user, msg, package.getDetailsURL())
+			addNotification(package.maintainers, current_user, msg, package.getDetailsURL())
 			db.session.commit()
 			return redirect(package.getDetailsURL())