def incident_update(incident_update_id, user_id):
incident_update_notification_module = IncidentUpdateNotificationModule()
subscriber_module = SubscriberModule()
task_module = TaskModule()
for subscriber in subscriber_module.get_iterator():
notification = incident_update_notification_module.is_subscriber_notified(
incident_update_id, subscriber.id)
if notification:
# Send notification.id to the queue again
task_module.delay("notify_subscriber",
{"notification_id": notification.id}, user_id)
else:
# Create new notification and send to queue
new_notification = incident_update_notification_module.insert_one({
"status":
"pending",
"incident_update_id":
incident_update_id,
"subscriber_id":
subscriber.id
})
if new_notification:
# Send new_notification.id to the Queue
task_module.delay("notify_subscriber",
{"notification_id": new_notification.id},
user_id)
return {"status": "passed", "result": "{}", "notify_type": "passed"}
def get(self, request, incident_id, update_id):
self.__context = Context()
self.__incident = IncidentModule()
self.__incident_update = IncidentUpdateModule()
self.__incident_update_component = IncidentUpdateComponentModule()
self.__component = ComponentModule()
self.__component_group = ComponentGroupModule()
self.__incident_update_notification = IncidentUpdateNotificationModule(
)
self.__correlation_id = request.META[
"X-Correlation-ID"] if "X-Correlation-ID" in request.META else ""
incident = self.__incident.get_one_by_id(incident_id)
if not incident:
raise Http404("Incident not found.")
update = self.__incident_update.get_one_by_id(update_id)
if not update:
raise Http404("Incident update not found.")
update["datetime"] = update["datetime"].strftime("%b %d %Y %H:%M:%S")
update["message"] = markdown2.markdown(update["message"])
update[
"notified_subscribers"] = self.__incident_update_notification.count_by_update_status(
update["id"], IncidentUpdateNotificationModule.SUCCESS)
update[
"failed_subscribers"] = self.__incident_update_notification.count_by_update_status(
update["id"], IncidentUpdateNotificationModule.FAILED)
components = self.__format_components(self.__component.get_all())
affected_components = self.__format_affected_components(
self.__incident_update_component.get_all(update_id))
self.__context.autoload_options()
self.__context.autoload_user(
request.user.id if request.user.is_authenticated else None)
self.__context.push({
"page_title":
_("View Incident Update · %s") % self.__context.get(
"app_name", os.getenv("APP_NAME", "Silverback")),
"update":
update,
"incident":
incident,
"components":
components,
"affected_components":
affected_components
})
return render(request, self.template_name, self.__context.get())
def __init__(self):
self.__request = Request()
self.__response = Response()
self.__helpers = Helpers()
self.__form = Form()
self.__incident = IncidentModule()
self.__incident_update = IncidentUpdateModule()
self.__task = Task_Module()
self.__notification = NotificationModule()
self.__subscriber = SubscriberModule()
self.__incident_update_notification = IncidentUpdateNotificationModule()
self.__logger = self.__helpers.get_logger(__name__)
self.__form.add_validator(ExtraRules())
def incident_update(incident_update_id, user_id, correlation_id=""):
logger = Helpers().get_logger(__name__)
logger.info(
_("Worker started processing incident_update task with parameters %(parameters)s {'correlationId':'%(correlationId)s'}"
) % {
"parameters": json.dumps({}),
"correlationId": correlation_id
})
incident_update_notification_module = IncidentUpdateNotificationModule()
subscriber_module = SubscriberModule()
task_module = TaskModule()
for subscriber in subscriber_module.get_iterator():
notification = incident_update_notification_module.is_subscriber_notified(
incident_update_id, subscriber.id)
if notification:
# Send notification.id to the queue again
task_module.delay("notify_subscriber",
{"notification_id": notification.id}, user_id)
else:
# Create new notification and send to queue
new_notification = incident_update_notification_module.insert_one({
"status":
"pending",
"incident_update_id":
incident_update_id,
"subscriber_id":
subscriber.id
})
if new_notification:
# Send new_notification.id to the Queue
task_module.delay("notify_subscriber",
{"notification_id": new_notification.id},
user_id)
return {"status": "passed", "result": "{}", "notify_type": "passed"}
def notify_subscriber(notification_id):
option_entity = OptionEntity()
incident_update_notification_module = IncidentUpdateNotificationModule()
app_name = option_entity.get_value_by_key("app_name")
app_email = option_entity.get_value_by_key("app_email")
app_url = option_entity.get_value_by_key("app_url")
notification = incident_update_notification_module.get_one_by_id(
notification_id)
incident_update = notification["incident_update"]
incident = notification["incident_update"].incident
subscriber = notification["subscriber"]
data = {}
data["incident_uri"] = incident.uri
data["incident_update_time"] = incident_update.datetime.strftime(
"%b %d %Y %H:%M:%S")
data["incident_type"] = incident_update.status.title()
data["incident_update"] = markdown2.markdown(incident_update.message)
if notification["status"] == "pending":
# send the notification for the first time
if subscriber.type == SubscriberModule.EMAIL:
status = __deliver_email(
app_name, app_email, app_url, [subscriber.email],
_("%(app_name)s Incident Update: %(incident_name)s") % {
"app_name": app_name,
"incident_name": incident.name
}, "mails/incident_update.html", data, False)
elif subscriber.type == SubscriberModule.PHONE:
status = __deliver_sms(
app_name, subscriber.phone, "%s%s" %
(app_url.strip("/"),
reverse("app.web.incidents", kwargs={'uri': incident.uri})))
elif subscriber.type == SubscriberModule.ENDPOINT:
status = __deliver_webhook(subscriber.endpoint,
subscriber.auth_token, '{}' % ())
if status:
# message sent
incident_update_notification_module.update_one_by_id(
notification["id"], {"status": "success"})
else:
# message failed
incident_update_notification_module.update_one_by_id(
notification["id"], {"status": "failed"})
elif notification["status"] == "failed":
# Retry to send the notification
if subscriber.type == SubscriberModule.EMAIL:
status = __deliver_email(
app_name, app_email, app_url, [subscriber.email],
_("%(app_name)s Incident Update: %(incident_name)s") % {
"app_name": app_name,
"incident_name": incident.name
}, "mails/incident_update.html", data, False)
elif subscriber.type == SubscriberModule.PHONE:
status = __deliver_sms(
app_name, subscriber.phone, "%s%s" %
(app_url.strip("/"),
reverse("app.web.incidents", kwargs={'uri': incident.uri})))
elif subscriber.type == SubscriberModule.ENDPOINT:
status = __deliver_webhook(subscriber.endpoint,
subscriber.auth_token, '{}' % ())
if status:
# message sent again
incident_update_notification_module.update_one_by_id(
notification["id"], {"status": "success"})
else:
print("skip subscriber#%s!" % subscriber.id)
return {"status": "passed", "result": "{}", "notify_type": "passed"}
class IncidentUpdates(View):
__request = None
__response = None
__helpers = None
__form = None
__logger = None
__user_id = None
__incident = None
__incident_update = None
__task = None
__notification = None
__subscriber = None
__incident_update_notification = None
__correlation_id = None
def __init__(self):
self.__request = Request()
self.__response = Response()
self.__helpers = Helpers()
self.__form = Form()
self.__incident = IncidentModule()
self.__incident_update = IncidentUpdateModule()
self.__task = Task_Module()
self.__notification = NotificationModule()
self.__subscriber = SubscriberModule()
self.__incident_update_notification = IncidentUpdateNotificationModule()
self.__logger = self.__helpers.get_logger(__name__)
self.__form.add_validator(ExtraRules())
@allow_if_authenticated
def post(self, request, incident_id):
self.__correlation_id = request.META["X-Correlation-ID"] if "X-Correlation-ID" in request.META else ""
self.__user_id = request.user.id
self.__request.set_request(request)
request_data = self.__request.get_request_data("post", {
"status": "",
"notify_subscribers": "",
"message": "",
"datetime": "",
})
self.__form.add_inputs({
'message': {
'value': request_data["message"],
'sanitize': {
'strip': {}
},
'validate': {}
},
'datetime': {
'value': request_data["datetime"],
'sanitize': {
'strip': {}
},
'validate': {}
},
'status': {
'value': request_data["status"],
'validate': {
'any_of': {
'param': [["investigating", "identified", "monitoring", "update", "resolved"]],
'error': _('Error! Status is invalid.')
}
}
},
'notify_subscribers': {
'value': request_data["notify_subscribers"],
'validate': {
'any_of': {
'param': [["on", "off"]],
'error': _('Error! Notify subscribers is invalid.')
}
}
}
})
self.__form.process()
if not self.__form.is_passed():
return JsonResponse(self.__response.send_errors_failure(self.__form.get_errors(), {}, self.__correlation_id))
result = self.__incident_update.insert_one({
"notify_subscribers": self.__form.get_sinput("notify_subscribers"),
"datetime": DateTimeField().clean(self.__form.get_sinput("datetime")),
"total_suscribers": self.__subscriber.count_by_status(SubscriberModule.VERIFIED),
"message": self.__form.get_sinput("message"),
"status": self.__form.get_sinput("status"),
"incident_id": incident_id
})
if self.__form.get_sinput("status") == "resolved":
self.__incident.update_one_by_id(incident_id, {
"status": "closed"
})
else:
self.__incident.update_one_by_id(incident_id, {
"status": "open"
})
if result:
return JsonResponse(self.__response.send_private_success([{
"type": "success",
"message": _("Incident update created successfully.")
}], {}, self.__correlation_id))
else:
return JsonResponse(self.__response.send_private_failure([{
"type": "error",
"message": _("Error! Something goes wrong while creating update.")
}], {}, self.__correlation_id))
@allow_if_authenticated
def get(self, request, incident_id):
self.__correlation_id = request.META["X-Correlation-ID"] if "X-Correlation-ID" in request.META else ""
self.__request.set_request(request)
request_data = self.__request.get_request_data("get", {
"offset": "",
"limit": ""
})
try:
offset = int(request_data["offset"])
limit = int(request_data["limit"])
except Exception:
offset = 0
limit = 0
return JsonResponse(self.__response.send_private_success([], {
'updates': self.__format_incident_updates(self.__incident_update.get_all(incident_id, offset, limit), incident_id),
'metadata': {
'offset': offset,
'limit': limit,
'count': self.__incident_update.count_all(incident_id)
}
}, self.__correlation_id))
def __format_incident_updates(self, updates, incident_id):
updates_list = []
for update in updates:
notified_subscribers = self.__incident_update_notification.count_by_update_status(
update.id,
IncidentUpdateNotificationModule.SUCCESS
)
progress = int(notified_subscribers/update.total_suscribers) * 100 if update.total_suscribers > 0 else 0
updates_list.append({
"id": update.id,
"status": update.status.title(),
"notify_subscribers": update.notify_subscribers.title(),
"datetime": update.datetime.strftime("%b %d %Y %H:%M:%S"),
"progress": progress if progress <= 100 else 100,
"created_at": update.created_at.strftime("%b %d %Y %H:%M:%S"),
"view_url": reverse("app.web.admin.incident_update.view", kwargs={'incident_id': incident_id, "update_id": update.id}),
"edit_url": reverse("app.web.admin.incident_update.edit", kwargs={'incident_id': incident_id, "update_id": update.id}),
"delete_url": reverse("app.api.private.v1.admin.incident_update.endpoint", kwargs={'incident_id': incident_id, "update_id": update.id})
})
return updates_list