diff --git a/src/node/api_clients/rest.py b/src/node/api_clients/rest.py index 210cae38..762d5183 100644 --- a/src/node/api_clients/rest.py +++ b/src/node/api_clients/rest.py @@ -145,3 +145,33 @@ class REST(BaseClient): get_messages_response = self.rest_call("get", base_url) return get_messages_response.json() + + def set_log_level(self, log_level): + return self.rest_call("post", f"admin/v1/log-level/{quote(log_level)}") + + def get_service_peers(self): + return self.rest_call("get", "admin/v1/peers/service").json() + + def get_connected_peers(self): + return self.rest_call("get", "admin/v1/peers/connected").json() + + def get_connected_peers_on_shard(self, shard_id): + return self.rest_call("get", f"admin/v1/peers/connected/on/{quote(str(shard_id))}").json() + + def get_relay_peers(self): + return self.rest_call("get", "admin/v1/peers/relay").json() + + def get_relay_peers_on_shard(self, shard_id): + return self.rest_call("get", f"admin/v1/peers/relay/on/{quote(str(shard_id))}").json() + + def get_mesh_peers(self): + return self.rest_call("get", "admin/v1/peers/mesh").json() + + def get_mesh_peers_on_shard(self, shard_id): + return self.rest_call("get", f"admin/v1/peers/mesh/on/{quote(str(shard_id))}").json() + + def get_peer_stats(self): + return self.rest_call("get", "admin/v1/peers/stats").json() + + def get_filter_subscriptions(self): + return self.rest_call("get", "admin/v1/filter/subscriptions").json() diff --git a/src/node/waku_node.py b/src/node/waku_node.py index 99e565b0..8faadab3 100644 --- a/src/node/waku_node.py +++ b/src/node/waku_node.py @@ -534,3 +534,33 @@ class WakuNode: matches = self._docker_manager.search_log_for_keywords(self._log_path, keywords, False) assert not matches, f"Found errors {matches}" + + def set_log_level(self, log_level): + return self._api.set_log_level(log_level) + + def get_service_peers(self): + return self._api.get_service_peers() + + def get_connected_peers(self): + return self._api.get_connected_peers() + + def get_connected_peers_on_shard(self, shard_id): + return self._api.get_connected_peers_on_shard(shard_id) + + def get_relay_peers(self): + return self._api.get_relay_peers() + + def get_relay_peers_on_shard(self, shard_id): + return self._api.get_relay_peers_on_shard(shard_id) + + def get_mesh_peers(self): + return self._api.get_mesh_peers() + + def get_mesh_peers_on_shard(self, shard_id): + return self._api.get_mesh_peers_on_shard(shard_id) + + def get_peer_stats(self): + return self._api.get_peer_stats() + + def get_filter_subscriptions(self): + return self._api.get_filter_subscriptions()