convertVirtualHubsToRealHubs.py

import concurrent.futures

import requests

from utils import utils
from utils import utils

real_hubs = []


def get_real_hub(env, orgId, virtualHub):
    """
    Retrieves the real hub information based on the provided parameters.

    Parameters:
    - env (str): The environment code used to construct the request URL.
    - orgId (str): The organization ID associated with the hub.
    - virtualHub (str): The virtual hub for which the real hub information is requested.

    Returns:
    None
    """
    url = f"http://tesseract.{utils.convert_env(env)}.milezero.com/Tesseract-war/api/config/hub/{orgId}?countryCode=USA&companyBrandCode=01&posInstanceNumber={virtualHub}"

    response = requests.get(url=url, timeout=5).json()

    try:
        real_hubs.append(response["hubName"])
        print(f"{virtualHub} -> {response['hubName']}")
    except Exception:
        print(f"{virtualHub}")


def main():
    """
    Main function to retrieve real hub information for multiple virtual hubs.

    Parameters:
    None

    Returns:
    None
    """
    env = utils.select_env()
    orgId = utils.select_org(env)

    virtual_hubs = []

    virtual_hubs = list(set(virtual_hubs))

    with concurrent.futures.ThreadPoolExecutor() as pool:
        for vhub in virtual_hubs:
            pool.submit(get_real_hub, env, orgId, vhub)

    pool.shutdown(wait=True)

    print(f"{len(virtual_hubs)} in, {len(real_hubs)} out")


main()