Function get_request_function_and_path

code-generator.py:135–183

Given an endpoint (e.g. '/sapi/v1/userInfo'), determine which _request_*_api function is appropriate in the client, remove the recognized prefix plus any version segments (e.g. /v1/), parse out any version (v1, v2, etc.), and return (request_function, stripped_path, version). E

(
    endpoint: str,
)

def get_request_function_and_path(
    endpoint: str,
) -> tuple[str | None, str | None, int | None]:
    """
    Given an endpoint (e.g. '/sapi/v1/userInfo'), determine which _request_*_api
    function is appropriate in the client, remove the recognized prefix plus any
    version segments (e.g. /v1/), parse out any version (v1, v2, etc.),
    and return (request_function, stripped_path, version).

    Example:
        endpoint = '/sapi/v1/exchangeInfo'
        -> returns ('_request_margin_api', 'exchangeInfo', 1)

    If no recognized prefix is found, return (None, None, None).
    If no version is found, version will be None.
    """
    # Sort prefixes by length descending to match the longest prefix first
    sorted_prefixes = sorted(PREFIX_MAP.keys(), key=len, reverse=True)
    request_func = None
    stripped = None

    # Identify which prefix is present, if any
    matched_prefix = None
    for prefix in sorted_prefixes:
        if endpoint.startswith(prefix):
            request_func = PREFIX_MAP[prefix]
            matched_prefix = prefix
            break

    # If no recognized prefix, return null
    if not request_func or matched_prefix is None:
        return None, None, None

    stripped = endpoint[len(matched_prefix) :]

    # Attempt to parse out the version, e.g. '/v1/', '/v2/'
    version_match = re.search(r"/v(\d+)/", stripped)
    version = None
    if version_match:
        # Convert the matched text into an integer
        version = int(version_match.group(1))

    # Remove version segments like /v1/, /v2/
    if stripped:
        stripped = re.sub(r"/v\d+/", "/", stripped)
        # Strip leading/trailing slashes
        stripped = stripped.strip("/")

    return request_func, stripped, version


def check_method_in_file(method, endpoint, file_name):