2025-03-05 11:07:44 +00:00

33 lines
1.2 KiB
Python

from src.libs.custom_logger import get_custom_logger
import json
from src.api_clients.base_client import BaseClient
logger = get_custom_logger(__name__)
class REST(BaseClient):
def __init__(self, rest_port, rest_host="127.0.0.1"):
self._rest_port = rest_port
self._rest_host = rest_host
def rest_call(self, method, endpoint, payload=None):
url = f"http://{self._rest_host}:{self._rest_port}/{endpoint}"
headers = {"Content-Type": "application/json", "Connection": "close"}
return self.make_request(method, url, headers=headers, data=payload)
def rest_call_text(self, method, endpoint, payload=None):
url = f"http://{self._rest_host}:{self._rest_port}/{endpoint}"
headers = {"accept": "text/plain", "Connection": "close"}
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 send_dispersal_request(self, data):
return self.rest_call("post", "disperse-data", json.dumps(data))
def send_get_range(self, query):
response = self.rest_call("post", "da/get-range", json.dumps(query))
return response.json()