ngrok public link reciever

ngrok_link_reciever.py

import requests as r


def get_ngrok_url(protocol: str, url_with_protocol=True):
    if protocol not in ("https", "http"):
        raise ValueError("protocol param must be in ('https', 'http')")
    suffix = "" if protocol == "https" else " (http)"
    uri = f"http://localhost:4040/api/tunnels/command_line{suffix}"
    try:
        response = r.get(uri)
        url = response.json().get("public_url")
        if url_with_protocol:
            return url
        url = url.split("://")[1]
        return url
    except r.ConnectionError:
        raise ConnectionError(f"Can't connect to {uri}. Probably, you didn't started ngrok.")