larigira/larigira/dbadmin/suggestions.py

import os

from flask import current_app

from larigira.config import get_conf
from larigira.fsutils import scan_dir_audio


def get_suggested_files():
    if not get_conf()['FILE_PATH_SUGGESTION']:
        return []
    if current_app.cache.has('dbadmin.get_suggested_files'):
        return current_app.cache.get('dbadmin.get_suggested_files')
    current_app.logger.debug('get_suggested_files MISS in cache')
    files = []
    for path in get_conf()['FILE_PATH_SUGGESTION']:
        if not os.path.isdir(path):
            current_app.logger.warn('Invalid suggestion path: %s' % path)
            continue
        pathfiles = scan_dir_audio(path)
        files.extend(pathfiles)
    current_app.logger.debug('Suggested files: %s' % ', '.join(files))

    current_app.cache.set('dbadmin.get_suggested_files', files,
                          timeout=600)  # ten minutes
    return files


def get_suggested_dirs():
    dirset = set()
    for f in get_suggested_files():
        dirpath = os.path.dirname(f)
        while dirpath:
            if dirpath in dirset:
                break
            dirset.add(dirpath)
            dirpath = os.path.dirname(dirpath)

    return list(dirset)


def get_suggested_scripts():
    base = get_conf()['SCRIPTS_PATH']
    if not base or not os.path.isdir(base):
        return []
    fnames = [f for f in os.listdir(base)
              if os.access(os.path.join(base, f), os.R_OK | os.X_OK)]
    return fnames


def get_suggestions():
    files = get_suggested_files()
    if len(files) > 200:
        current_app.logger.warn("Too many suggested files, cropping")
        files = files[:200]
    return dict(
        files=files,
        dirs=get_suggested_dirs(),
        scripts=get_suggested_scripts(),
    )