api.py

import requests
import logging
import json

class LoadBalancerAPI:
    def __init__(self, base_url):
        self.base_url = base_url

    def register_instance(self, instance_id, instance_url):
        data = {
            "url": instance_url
        }
        api_endpoint = f'{self.base_url}/api/post/register'
        
        try:
            headers = {'Content-Type': 'application/json'}
            response = requests.post(api_endpoint, data=json.dumps(data), headers=headers)
            response.raise_for_status()
            return response.json()  # Assuming the API returns JSON
        except requests.exceptions.RequestException as e:
            logging.error(f'Failed to register instance {instance_id} to load balancer: {e}')
            return None

    def get_file_structure(self):
        api_endpoint = f'{self.base_url}/api/get/file_structure'
        try:
            headers = {'Content-Type': 'application/json'}
            response = requests.get(api_endpoint, headers=headers)
            response.raise_for_status()
            return response.json()  # Assuming the API returns JSON
        except requests.exceptions.RequestException as e:
            logging.error(f'Failed to file structure: {e}')
            return None