Support BytesIo objects for tagging about tinytag HOT 4 CLOSED

import tempfile
import requests
import io
from tinytag import TinyTag

def download_file(url: str) -> io.BytesIO:
    """
    Downloads the file from the given URL and returns it as a BytesIO object
    Args:
        url: The url to run from

    Returns:
        An object containing the file
    """

    try:
        response = requests.get(url, stream=True)
        response.raise_for_status()
    except requests.exceptions.RequestException as e:
        raise Exception(f"Could not run file from {url}: {e}")

    file = io.BytesIO()
    for chunk in response.iter_content(chunk_size=8192):
        file.write(chunk)
    file.seek(0)

    return file


def read_bytes_io_as_tiny_tag(data: io.BytesIO) -> TinyTag:
    """
    Reads the given BytesIO object as a file and returns it
    Args:
        data: The BytesIO object to read

    Returns:
        The file
    """
    with tempfile.NamedTemporaryFile() as tmp:
        tmp.write(data.getvalue())
        tmp.seek(0)

        tag = TinyTag.get(tmp.name)

        return tag


def run():
    url = "https://example.com/file.mp3"
    data = download_file(url)
    tag = read_bytes_io_as_tiny_tag(data=data)

    print('This track is by %s.' % tag.artist)
    print('It is %f seconds long.' % tag.duration)