add crunch_scripts/file-select

[arvados.git] / crunch_scripts / file-select

diff --git a/crunch_scripts/file-select b/crunch_scripts/file-select
new file mode 100755 (executable)
index 0000000..fb3f761
--- /dev/null
+++ b/crunch_scripts/file-select
@@ -0,0 +1,15 @@
+#!/usr/bin/env python
+
+import arvados
+import os
+import re
+
+this_job = arvados.current_job()
+this_task = arvados.current_task()
+this_job_input = this_job['script_parameters']['input']
+manifest_text = ""
+for f in arvados.CollectionReader(this_job_input).all_files():
+    if f.name() in this_job['script_parameters']['names']:
+        manifest_text += f.as_manifest()
+
+this_task.set_output(arvados.Keep.put(manifest_text))