com.amazonaws.mturk.cmd.AssignQualification.java Source code

Introduction

Here is the source code for com.amazonaws.mturk.cmd.AssignQualification.java

Source

/*
 * Copyright 2012 Amazon Technologies, Inc.
 * 
 * Licensed under the Amazon Software License (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at:
 * 
 * http://aws.amazon.com/asl
 * 
 * This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES
 * OR CONDITIONS OF ANY KIND, either express or implied. See the
 * License for the specific language governing permissions and
 * limitations under the License.
 */

package com.amazonaws.mturk.cmd;

import java.io.IOException;
import java.io.PrintWriter;

import org.apache.commons.cli.CommandLine;

public class AssignQualification extends AbstractCmd {

    private final String ARG_QUALTYPE = "qualtypeid";
    private final String ARG_WORKERID = "workerid";
    private final String ARG_INPUT_FILE = "input";
    private final String ARG_SCORE_FILE = "scorefile";
    private final String ARG_SCORE = "score";
    private final String ARG_DONOT_NOTIFY = "donotnotify";

    public static void main(String[] args) {
        AssignQualification aq = new AssignQualification();
        aq.run(args);
    }

    protected void initOptions() {
        opt.addOption(ARG_QUALTYPE, true, "Id of the Qualification Type to be assigned");
        opt.addOption(ARG_INPUT_FILE, true, "File containing the Qualification Type to be assigned");
        opt.addOption(ARG_WORKERID, true, "Id of the Worker to receive the Qualification");
        opt.addOption(ARG_SCORE_FILE, true, "File containing list of workers to be assigned the Qualification."
                + " (the columns must be titled " + ARG_WORKERID + " and " + ARG_SCORE + " )");
        opt.addOption(ARG_SCORE, true, "(optional) Integer value for the Qualification");
        opt.addOption(ARG_DONOT_NOTIFY, false, "(optional) Flag indicating not to send notification to worker.");
    }

    protected void printHelp() {
        log.info(AssignQualification.class.getName() + " -" + ARG_QUALTYPE + " [qual to grant] " + " -"
                + ARG_WORKERID + " [worker to be granted]" + " -" + ARG_SCORE + " [value to grant]");
        log.info("Or");
        log.info(AssignQualification.class.getName() + " -" + ARG_INPUT_FILE
                + " [ input file containing Qualification Type] " + " -" + ARG_SCORE_FILE + " [ score file ] "
                + " -" + ARG_SCORE + " [value to grant]");

        PrintWriter pw = new PrintWriter(System.out);
        formatter.printOptions(pw, formatter.defaultWidth, opt, formatter.defaultLeftPad, formatter.defaultDescPad);
        pw.flush();
    }

    protected void runCommand(CommandLine cmdLine) throws Exception {
        if (!cmdLine.hasOption(ARG_QUALTYPE) && !cmdLine.hasOption(ARG_INPUT_FILE)) {
            log.error("Either -" + ARG_QUALTYPE + " or -" + ARG_INPUT_FILE + " must be passed");
            System.exit(-1);

        }
        if (!cmdLine.hasOption(ARG_SCORE_FILE)) {
            if (!cmdLine.hasOption(ARG_WORKERID)) {
                log.error("Either -" + ARG_WORKERID + " or -" + ARG_SCORE_FILE + " should be passed");
                System.exit(-1);
            }
        }
        String[] qualIds = null;
        if (cmdLine.hasOption(ARG_QUALTYPE)) {
            qualIds = new String[1];
            qualIds[0] = cmdLine.getOptionValue(ARG_QUALTYPE);
        } else {
            qualIds = super.getFieldValuesFromFile(cmdLine.getOptionValue(ARG_INPUT_FILE), ARG_QUALTYPE);
            if (qualIds.length == 0) {
                log.error("qualtypeid must be present in the -" + ARG_INPUT_FILE + " file");
                System.exit(-1);
            }
        }

        String strValue = cmdLine.hasOption(ARG_SCORE) ? cmdLine.getOptionValue(ARG_SCORE) : null;
        if (cmdLine.hasOption(ARG_SCORE_FILE)) {
            assignQualification(qualIds, cmdLine.getOptionValue(ARG_SCORE_FILE),
                    cmdLine.hasOption(ARG_DONOT_NOTIFY), strValue);
        } else {
            String workerId = cmdLine.getOptionValue(ARG_WORKERID);
            Boolean notify = !cmdLine.hasOption(ARG_DONOT_NOTIFY);

            assignQualification(qualIds, workerId, strValue, notify);
        }
    }

    public void assignQualification(String[] qualIds, String workerFile, Boolean notify, String argScore)
            throws IOException {
        String[] workers = super.getFieldValuesFromFile(workerFile, ARG_WORKERID);
        String[] scores = super.getFieldValuesFromFile(workerFile, ARG_SCORE);
        for (int i = 0; i < workers.length; i++) {
            String workerScore = (scores.length < 1) ? argScore : scores[i];
            assignQualification(qualIds, workers[i], workerScore, notify);
        }
    }

    public void assignQualification(String[] qualIds, String workerId, String score, Boolean notify) {
        Integer value = null;
        if (score != null) {
            try {
                // Verify that strValue is an integer and not something else
                value = Integer.valueOf(score);
            } catch (NumberFormatException e) {
                log.error("Invalid format for -" + ARG_SCORE + " : " + score);
                System.exit(-1);
            }
        }
        for (String qualId : qualIds) {
            service.assignQualification(qualId, workerId, value, notify);
            String workerScore = value != null ? " with value " + value.toString() : " with default value";
            log.info("Assigned qualification " + qualId + " to " + workerId + workerScore);
        }

    }

}