Add rest APIs and wrappers

This commit is contained in:
aya 2025-08-06 22:28:04 +03:00
parent 58fa92e096
commit 3c3118ea80
2 changed files with 60 additions and 0 deletions

View File

@ -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()

View File

@ -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()