def create_service_related_downtime(params):
"""Create a service related scheduled downtime"""
body = params['body']
live = sites.live()
downtime_type: DowntimeType = body['downtime_type']
if downtime_type == 'service':
downtime_commands.schedule_service_downtime(
live,
host_name=body['host_name'],
service_description=body['service_descriptions'],
start_time=body['start_time'],
end_time=body['end_time'],
recur=body['recur'],
duration=body['duration'],
user_id=config.user.ident,
comment=body.get(
'comment',
f"Downtime for services {', '.join(body['service_descriptions'])!r}@{body['host_name']!r}"
),
)
elif downtime_type == 'servicegroup':
downtime_commands.schedule_servicegroup_service_downtime(
live,
servicegroup_name=body['servicegroup_name'],
start_time=body['start_time'],
end_time=body['end_time'],
recur=body['recur'],
duration=body['duration'],
user_id=config.user.ident,
comment=body.get('comment', f"Downtime for servicegroup {body['servicegroup_name']!r}"),
)
elif downtime_type == 'service_by_query':
try:
downtime_commands.schedule_services_downtimes_with_query(
live,
query=body['query'],
start_time=body['start_time'],
end_time=body['end_time'],
recur=body['recur'],
duration=body['duration'],
user_id=config.user.ident,
comment=body.get('comment', ''),
)
except QueryException:
return problem(
status=422,
title="Query did not match any service",
detail="The provided query returned an empty list so no downtime was set",
)
else:
return problem(status=400,
title="Unhandled downtime-type.",
detail=f"The downtime-type {downtime_type!r} is not supported.")
return Response(status=204)
def create_service_related_downtime(params):
"""Create a service related scheduled downtime"""
body = params["body"]
live = sites.live()
downtime_type: DowntimeType = body["downtime_type"]
if downtime_type == "service":
host_name = body["host_name"]
with detailed_connection(live) as conn:
site_id = Query(columns=[Hosts.name], filter_expr=Hosts.name.op("=", host_name)).value(
conn
)
downtime_commands.schedule_service_downtime(
live,
site_id,
host_name=body["host_name"],
service_description=body["service_descriptions"],
start_time=body["start_time"],
end_time=body["end_time"],
recur=body["recur"],
duration=body["duration"],
user_id=user.ident,
comment=body.get(
"comment",
f"Downtime for services {', '.join(body['service_descriptions'])!r}@{body['host_name']!r}",
),
)
elif downtime_type == "servicegroup":
downtime_commands.schedule_servicegroup_service_downtime(
live,
servicegroup_name=body["servicegroup_name"],
start_time=body["start_time"],
end_time=body["end_time"],
recur=body["recur"],
duration=body["duration"],
user_id=user.ident,
comment=body.get("comment", f"Downtime for servicegroup {body['servicegroup_name']!r}"),
)
elif downtime_type == "service_by_query":
try:
downtime_commands.schedule_services_downtimes_with_query(
live,
query=body["query"],
start_time=body["start_time"],
end_time=body["end_time"],
recur=body["recur"],
duration=body["duration"],
user_id=user.ident,
comment=body.get("comment", ""),
)
except QueryException:
return problem(
status=422,
title="Query did not match any service",
detail="The provided query returned an empty list so no downtime was set",
)
else:
return problem(
status=400,
title="Unhandled downtime-type.",
detail=f"The downtime-type {downtime_type!r} is not supported.",
)
return Response(status=204)