|
@@ -5,6 +5,7 @@ import random
|
|
|
from tempfile import mkstemp
|
|
|
|
|
|
from larigira.fsutils import scan_dir
|
|
|
+log = logging.getLogger(__name__)
|
|
|
|
|
|
|
|
|
def generate(spec):
|
|
@@ -23,7 +24,7 @@ def generate(spec):
|
|
|
found_files = set()
|
|
|
for path in spec['paths']:
|
|
|
if not os.path.exists(path):
|
|
|
- logging.warning("Can't find requested path: %s", path)
|
|
|
+ log.warning("Can't find requested path: %s", path)
|
|
|
continue
|
|
|
if os.path.isfile(path):
|
|
|
found_files.add(path)
|
|
@@ -34,9 +35,11 @@ def generate(spec):
|
|
|
|
|
|
for path in picked:
|
|
|
tmp = mkstemp(suffix=os.path.splitext(path)[-1],
|
|
|
- prefix='audiogen-randomdir-')
|
|
|
+ prefix='randomdir-')
|
|
|
os.close(tmp[0])
|
|
|
shutil.copy(path, tmp[1])
|
|
|
+ log.info("copying %s -> %s", path, os.path.basename(tmp[1]))
|
|
|
yield 'file://{}'.format(tmp[1])
|
|
|
|
|
|
+
|
|
|
generate.description = 'Picks random files from a specified directory'
|