diff options
Diffstat (limited to 'app/relay/runner.js')
| -rw-r--r-- | app/relay/runner.js | 29 |
1 files changed, 27 insertions, 2 deletions
diff --git a/app/relay/runner.js b/app/relay/runner.js index 741ef8a..4762045 100644 --- a/app/relay/runner.js +++ b/app/relay/runner.js @@ -56,6 +56,30 @@ function clear_task(is_gpu, task){ } } +function sanitize_path(f){ + return f.replace(/^\//,'').replace(/\.\./, '') +} + +export function upload_file(task, cb) { + const module = modules[task.module] + const filepath = path.join(module.cwd, sanitize_path(task.path), sanitize_path(task.filename)) + const params = [ + '-F', 'module=' + task.module, + '-F', 'activity=' + task.activity, + '-F', 'generated=' + (String(task.generated) === 'true'), + '-F', 'processed=' + (String(task.processed) === 'true'), + '-F', "file=@" + filepath, + process.env.API_REMOTE + '/api/folder/' + task.folder_id + '/upload/', + ] + console.log(params) + execFile('curl', params, cb) + // curl \ + // -F "module=samplernn" \ + // -F "activity=train" \ + // -F "file=@woods1.jpg" \ + // localhost:7013/api/folder/1/upload/ +} + export function status () { return { cpu: serialize_task(state.current_cpu_task), @@ -143,7 +167,8 @@ export function run_script(task, cb) { cb("") } const module = modules[task.module] - const { activity, interpreter, params } = build_params(module, task) + const activity = module.activities[task.activity] + const { interpreter, params } = build_params(module, activity, task) if (! interpreter) return { type: 'error', error: "No such interpreter: " + activity.interpreter } if (! activity.isScript) return { type: 'error', error: "Not a script: " + task.module } @@ -163,7 +188,7 @@ export function run_task(task, preempt, watch){ } export function run_task_with_activity(task, module, activity, preempt, watch) { - const { interpreter, params } = build_params(module, task) + const { interpreter, params } = build_params(module, activity, task) if (! interpreter) return { type: 'error', error: "No such interpreter: " + activity.interpreter } if (interpreter.gpu && state.current_gpu_task.status !== 'IDLE') { |
