larigira/larigira/audiogen_http.py

import os
import logging
import posixpath
from tempfile import mkstemp
import urllib.request
from urllib.parse import urlparse

log = logging.getLogger(__name__)


def put(url, destdir=None, copy=False):
    if url.split(':')[0] not in ('http', 'https'):
        log.warning('Not a valid URL: %s', url)
        return None
    ext = url.split('.')[-1]
    if ext.lower() not in ('mp3', 'ogg', 'oga', 'wma', 'm4a'):
        log.warning('Invalid format (%s) for "%s"', ext, url)
        return None
    if not copy:
        return url
    fname = posixpath.basename(urlparse(url).path)
    # sanitize
    fname = "".join(c for c in fname
                    if c.isalnum() or c in list('._-')).rstrip()
    tmp = mkstemp(suffix='.' + ext, prefix='http-%s-' % fname, dir=destdir)
    os.close(tmp[0])
    log.info("downloading %s -> %s", url, tmp[1])
    fname, headers = urllib.request.urlretrieve(url, tmp[1])
    return 'file://%s' % os.path.realpath(tmp[1])


def generate(spec):
    '''
    resolves audiospec-static

    Recognized argument is  "paths" (list of static paths)
    '''
    if 'urls' not in spec:
        raise ValueError("Malformed audiospec: missing 'paths'")

    for url in spec['urls']:
        ret = put(url, copy=True)
        if ret is None:
            continue
        yield ret


generate.description = 'Fetch audio from an URL'