self-hosted/oncall
1
0
Fork 0
mirror of https://github.com/linkedin/oncall.git synced 2025-11-28 23:20:23 +02:00
Code Issues Releases Activity

add create event notifications for auto scheduler (#392)

Browse Source
This commit is contained in:
Diego Cepeda
2023-04-04 10:06:04 -05:00
committed by GitHub
parent 5873625352
commit 86a3da74b8
2 changed files with 56 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
src/oncall/scheduler/default.py
View File

@@ -1,6 +1,7 @@
from datetime import datetime, timedelta
from pytz import timezone, utc
from oncall.utils import gen_link_id
from oncall.utils import gen_link_id, create_notification
from ..constants import EVENT_CREATED
from falcon import HTTPBadRequest
from ujson import dumps as json_dumps
import time
@@ -163,6 +164,19 @@ class Scheduler(object):
%%s, %%s, %%s, %%s, %%s, %%s
)''' % table_name
cursor.execute(query, event_args)
cursor.execute('SELECT `name` FROM `user` WHERE `id` = %s', user_id)
name = cursor.fetchone()
context = {
'team': team_id,
'role': role_id,
'full_name': name
}
create_notification(context, team_id,
[role_id],
EVENT_CREATED,
[user_id],
cursor,
start_time=event['start'])
else:
link_id = gen_link_id()
for event in events:
@@ -175,6 +189,19 @@ class Scheduler(object):
%%s, %%s, %%s, %%s, %%s, %%s, %%s
)''' % table_name
cursor.execute(query, event_args)
cursor.execute('SELECT `name` FROM `user` WHERE `id` = %s', user_id)
name = cursor.fetchone()
context = {
'team': team_id,
'role': role_id,
'full_name': name
}
create_notification(context, team_id,
[role_id],
EVENT_CREATED,
[user_id],
cursor,
start_time=event['start'])
def set_last_epoch(self, schedule_id, last_epoch, cursor):
cursor.execute('UPDATE `schedule` SET `last_epoch_scheduled` = %s WHERE `id` = %s',

29
src/oncall/scheduler/round-robin.py
View File

@@ -1,4 +1,5 @@
from oncall.utils import gen_link_id
from oncall.utils import gen_link_id, create_notification
from ..constants import EVENT_CREATED
from . import default
import logging
@@ -62,6 +63,19 @@ class Scheduler(default.Scheduler):
%%s, %%s, %%s, %%s, %%s, %%s
)''' % table_name
cursor.execute(query, event_args)
cursor.execute('SELECT `name` FROM `user` WHERE `id` = %s', user_id)
name = cursor.fetchone()
context = {
'team': team_id,
'role': role_id,
'full_name': name
}
create_notification(context, team_id,
[role_id],
EVENT_CREATED,
[user_id],
cursor,
start_time=event['start'])
else:
link_id = gen_link_id()
for event in events:
@@ -74,6 +88,19 @@ class Scheduler(default.Scheduler):
%%s, %%s, %%s, %%s, %%s, %%s, %%s
)''' % table_name
cursor.execute(query, event_args)
cursor.execute('SELECT `name` FROM `user` WHERE `id` = %s', user_id)
name = cursor.fetchone()
context = {
'team': team_id,
'role': role_id,
'full_name': name
}
create_notification(context, team_id,
[role_id],
EVENT_CREATED,
[user_id],
cursor,
start_time=event['start'])
cursor.execute('UPDATE `schedule` SET `last_scheduled_user_id` = %s WHERE `id` = %s', (user_id, schedule_id))
def populate(self, schedule, start_time, dbinfo, table_name='event'):