def close_maintenance(maintenance_id):
logger.info("[%s] Close maintenance", maintenance_id)
m = Maintenance.get_by_id(maintenance_id)
if not m:
logger.info("[%s] Not found, skipping", maintenance_id)
return
if not m.escalation_tt:
logger.info("[%s] Not escalated, skipping", maintenance_id)
return
tts_name, tt_id = m.escalation_tt.split(":", 1)
tts = TTSystem.get_by_name(tts_name).get_system()
if not tts:
logger.error("[%s] TT system '%s' is not found", maintenance_id,
tts_name)
return
try:
logger.info("[%s] Closing TT %s", maintenance_id, tt_id)
tts.close_tt(tt_id,
subject="Closed",
body="Closed",
login="******")
metrics["maintenance_tt_close"] += 1
except tts.TTError as e:
logger.error("[%s] Failed to close TT %s: %s", maintenance_id, tt_id,
e)
metrics["maintenance_tt_close_fail"] += 1
def start_maintenance(maintenance_id):
logger.info("[%s] Start maintenance", maintenance_id)
m = Maintenance.get_by_id(maintenance_id)
if not m:
logger.info("[%s] Not found, skipping")
return
if not m.escalate_managed_object:
logger.info("[%s] No managed object to escalate", maintenance_id)
return
if m.escalation_tt:
logger.info("[%s] Already escalated as TT %s", maintenance_id, m.escalation_tt)
return
# Get external TT system
tts_id = m.escalate_managed_object.tt_system_id
if not tts_id:
logger.info(
"[%s] No TT mapping for object %s(%s)",
maintenance_id,
m.escalate_managed_object.name,
m.escalate_managed_object.address,
)
return
tt_system = m.escalate_managed_object.tt_system
if not tt_system:
logger.info("[%s] Cannot find TT system '%s'", maintenance_id, m.escalate_managed_object)
return
tts = tt_system.get_system()
try:
logger.info("[%s] Creating TT", maintenance_id)
tt_id = tts.create_tt(
queue=m.escalate_managed_object.tt_queue if m.escalate_managed_object.tt_queue else 1,
obj=tts_id,
reason=0,
subject=m.subject,
body=m.description or m.subject,
login="******",
timestamp=m.start,
)
logger.info("[%s] TT %s created", maintenance_id, tt_id)
if tts.promote_group_tt:
gtt = tts.create_group_tt(tt_id, m.start)
d = AffectedObjects._get_collection().find_one(
{"maintenance": m.id}, {"_id": 0, "affected_objects": 1}
)
if d:
objects = [x["object"] for x in d["affected_objects"]]
for d in ManagedObject.objects.filter(id__in=list(objects), is_managed=True):
logger.info("[%s] Appending object %s to group TT %s", maintenance_id, d, gtt)
tts.add_to_group_tt(gtt, d.tt_system_id)
if tt_id and not m.escalation_tt:
m.escalation_tt = "%s:%s" % (m.escalate_managed_object.tt_system.name, tt_id)
m.save()
metrics["maintenance_tt_create"] += 1
except tts.TTError as e:
logger.error("[%s] Failed to escalate: %s", maintenance_id, e)
metrics["maintenance_tt_fail"] += 1
def bulk_field_isinmaintenance(self, data):
if not data:
return data
if data[0]["status"] == "A":
mtc = set(Maintenance.currently_affected())
for x in data:
x["isInMaintenance"] = x["managed_object"] in mtc
else:
mos = set([x["managed_object"] for x in data])
mtc = {}
for mo in list(mos):
interval = []
for ao in AffectedObjects._get_collection().find(
{"affected_objects.object": {
"$eq": mo
}}, {
"_id": 0,
"maintenance": 1
}):
m = Maintenance.get_by_id(ao["maintenance"])
interval += [(m.start, m.stop)]
if interval:
mtc[mo] = interval
for x in data:
if x["managed_object"] in mtc:
left, right = list(zip(*mtc[x["managed_object"]]))
x["isInMaintenance"] = bisect.bisect(
right,
dateutil.parser.parse(x["timestamp"]).replace(
tzinfo=None)) != bisect.bisect(
left,
dateutil.parser.parse(
x["clear_timestamp"]).replace(tzinfo=None))
else:
x["isInMaintenance"] = False
return data