tools/load_file.py

import io
from urllib.parse import urlparse

import requests


def load_file(path_or_url: str) -> io.BytesIO:
    if _is_url(path_or_url):
        response = requests.get(path_or_url)
        response.raise_for_status()
        return io.BytesIO(response.content)
    else:
        with open(path_or_url, 'rb') as f:
            return io.BytesIO(f.read())


def _is_url(path_or_url):
    try:
        result = urlparse(path_or_url)
        return result.scheme in ('http', 'https')
    except ValueError:
        return False