89 lines
2.4 KiB
Python
89 lines
2.4 KiB
Python
import jingrow
|
|
|
|
from press.guards import role_guard
|
|
from press.utils import get_current_team
|
|
|
|
|
|
@jingrow.whitelist()
|
|
@role_guard.document(
|
|
document_type=lambda _: "Site",
|
|
inject_values=True,
|
|
should_throw=False,
|
|
)
|
|
@role_guard.document(
|
|
document_type=lambda _: "Release Group",
|
|
inject_values=True,
|
|
should_throw=False,
|
|
)
|
|
def get_notifications(
|
|
filters=None,
|
|
order_by="creation desc",
|
|
limit_start=None,
|
|
limit_page_length=None,
|
|
sites=None,
|
|
release_groups=None,
|
|
):
|
|
if not filters:
|
|
filters = {}
|
|
|
|
PressNotification = jingrow.qb.DocType("Press Notification")
|
|
query = (
|
|
jingrow.qb.from_(PressNotification)
|
|
.select(
|
|
PressNotification.name,
|
|
PressNotification.type,
|
|
PressNotification.read,
|
|
PressNotification.title,
|
|
PressNotification.message,
|
|
PressNotification.creation,
|
|
PressNotification.is_addressed,
|
|
PressNotification.is_actionable,
|
|
PressNotification.document_type,
|
|
PressNotification.document_name,
|
|
)
|
|
.where(PressNotification.team == get_current_team())
|
|
.orderby(PressNotification.creation, order=jingrow.qb.desc)
|
|
.limit(limit_page_length)
|
|
.offset(limit_start)
|
|
)
|
|
|
|
resources = set()
|
|
if sites and isinstance(sites, list):
|
|
resources.update(sites)
|
|
if release_groups and isinstance(release_groups, list):
|
|
resources.update(release_groups)
|
|
|
|
if resources:
|
|
query = query.where(PressNotification.reference_name.isin(resources))
|
|
|
|
if filters.get("read") == "Unread":
|
|
query = query.where(PressNotification.read == 0)
|
|
|
|
notifications = query.run(as_dict=True)
|
|
|
|
for notification in notifications:
|
|
if notification.document_type == "Deploy Candidate":
|
|
rg_name = jingrow.db.get_value("Deploy Candidate", notification.document_name, "group")
|
|
notification.route = f"groups/{rg_name}/deploys/{notification.document_name}"
|
|
elif notification.document_type == "Agent Job":
|
|
site_name = jingrow.db.get_value("Agent Job", notification.document_name, "site")
|
|
notification.route = (
|
|
f"sites/{site_name}/insights/jobs/{notification.document_name}" if site_name else None
|
|
)
|
|
elif notification.document_type == "Support Access":
|
|
notification.route = "access-requests"
|
|
else:
|
|
notification.route = None
|
|
|
|
return notifications
|
|
|
|
|
|
@jingrow.whitelist()
|
|
def mark_all_notifications_as_read():
|
|
jingrow.db.set_value("Press Notification", {"team": get_current_team()}, "read", 1, update_modified=False)
|
|
|
|
|
|
@jingrow.whitelist()
|
|
def get_unread_count():
|
|
return jingrow.db.count("Press Notification", {"read": False, "team": get_current_team()})
|