larigira/larigira/audiogen_mostrecent.py

import os
import logging
import shutil
import time
from tempfile import mkstemp
import fnmatch

from pytimeparse.timeparse import timeparse


def scan_dir(dirname, extension=None):
    if extension is None:
        extension = '*'
    for root, dirnames, filenames in os.walk(dirname):
        for fname in fnmatch.filter(filenames, extension):
            yield os.path.join(root, fname)


def get_mtime(fname):
    return int(os.path.getmtime(fname))


def recent_choose(paths, howmany, minepoch):
    found_files = {}
    for path in paths:
        if not os.path.exists(path):
            logging.warn("Can't find requested path: %s" % path)
            continue
        if os.path.isfile(path):
            found_files[path] = get_mtime(path)
        elif os.path.isdir(path):
            found_files.update({fname: get_mtime(fname)
                                for fname in scan_dir(path)})
    found_files = [(fname, mtime)
                   for (fname, mtime) in found_files.items()
                   if mtime >= minepoch]

    return [fname for fname, mtime in
            sorted(found_files, key=lambda x: x[1])[:howmany]]


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

    Recognized arguments:
        - path [mandatory]    source dir
        - maxage [default=ignored]  max age of audio files to pick
    '''
    for attr in ('path', 'maxage'):
        if attr not in spec:
            raise ValueError("Malformed audiospec: missing '%s'" % attr)

    if spec['maxage'].strip():
        try:
            maxage = int(spec['maxage'])
        except ValueError:
            maxage = timeparse(spec['maxage'])
            if maxage is None:
                raise ValueError("Unknown format for maxage: '{}'"
                                 .format(spec['maxage']))
            assert type(maxage) is int
    else:
        maxage = None

    now = int(time.time())
    minepoch = 0 if maxage is None else now - maxage

    picked = recent_choose([spec['path']], 1, minepoch)

    for path in picked:
        tmp = mkstemp(suffix=os.path.splitext(path)[-1],
                      prefix='audiogen-randomdir-')
        os.close(tmp[0])
        shutil.copy(path, tmp[1])
        yield 'file://{}'.format(tmp[1])

generate.description = 'Select most recent file inside a directory'