Roman 231a3918c5
test: add api endpoints
- disperse_data
- get_range
2025-01-16 19:08:22 +08:00

32 lines
1.2 KiB
Python

from src.libs.custom_logger import get_custom_logger
import json
from urllib.parse import quote
from src.node.api_clients.base_client import BaseClient
logger = get_custom_logger(__name__)
class REST(BaseClient):
def __init__(self, rest_port):
self._rest_port = rest_port
def rest_call(self, method, endpoint, payload=None):
url = f"http://127.0.0.1:{self._rest_port}/{endpoint}"
headers = {"Content-Type": "application/json"}
return self.make_request(method, url, headers=headers, data=payload)
def rest_call_text(self, method, endpoint, payload=None):
url = f"http://127.0.0.1:{self._rest_port}/{endpoint}"
headers = {"accept": "text/plain"}
return self.make_request(method, url, headers=headers, data=payload)
def info(self):
status_response = self.rest_call("get", "cryptarchia/info")
return status_response.json()
def disperse_data(self, dispersal_request):
return self.rest_call("post", "disperse-data", json.dumps(dispersal_request))
def get_range(self, app_id, data_range):
return self.rest_call("post", "da/get-range", json.dumps({app_id, data_range}))