fix: adjust get_monitor_status
method to previous changes
This commit is contained in:
parent
762dd4a657
commit
ce1cc12740
2 changed files with 31 additions and 3 deletions
15
tests/test_helper_methods.py
Normal file
15
tests/test_helper_methods.py
Normal file
|
@ -0,0 +1,15 @@
|
|||
import unittest
|
||||
|
||||
from uptime_kuma_api import MonitorStatus
|
||||
from uptime_kuma_test_case import UptimeKumaTestCase
|
||||
|
||||
|
||||
class TestHelperMethods(UptimeKumaTestCase):
|
||||
def test_monitor_status(self):
|
||||
monitor_id = self.add_monitor()
|
||||
status = self.api.get_monitor_status(monitor_id)
|
||||
self.assertTrue(type(status) == MonitorStatus)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
unittest.main()
|
|
@ -3677,9 +3677,22 @@ class UptimeKumaApi(object):
|
|||
# helper methods
|
||||
|
||||
def get_monitor_status(self, monitor_id: int) -> MonitorStatus:
|
||||
"""
|
||||
Get the monitor status.
|
||||
|
||||
:param int monitor_id: Id of the monitor.
|
||||
:return: The monitor status.
|
||||
:rtype: MonitorStatus
|
||||
:raises UptimeKumaException: If the monitor does not exist.
|
||||
|
||||
Example::
|
||||
|
||||
>>> api.get_monitor_status(1)
|
||||
<MonitorStatus.PENDING: 2>
|
||||
"""
|
||||
heartbeats = self.get_heartbeats()
|
||||
for heartbeat in heartbeats:
|
||||
if int(heartbeat["id"]) == monitor_id:
|
||||
status = heartbeat["data"][-1]["status"]
|
||||
for heartbeat_monitor_id in heartbeats:
|
||||
if heartbeat_monitor_id == monitor_id:
|
||||
status = heartbeats[heartbeat_monitor_id][-1]["status"]
|
||||
return MonitorStatus(status)
|
||||
raise UptimeKumaException("monitor does not exist")
|
||||
|
|
Loading…
Reference in a new issue