ruby-aws 1.6.0

data/lib/amazon/webservices/mechanical_turk.rb

@@ -0,0 +1,123 @@
+# Copyright:: Copyright (c) 2007 Amazon Technologies, Inc.
+# License::   Apache License, Version 2.0
+
+require 'amazon/util/logging'
+require 'amazon/webservices/util/amazon_authentication_relay'
+require 'amazon/webservices/mturk/mechanical_turk_error_handler'
+require 'amazon/webservices/util/validation_exception'
+require 'amazon/webservices/util/rest_transport'
+require 'amazon/webservices/util/soap_transport'
+
+module Amazon
+module WebServices
+
+class MechanicalTurk
+  include Amazon::Util::Logging
+
+  SOFTWARE_NAME = 'MTurkRubySDK'
+
+  SANDBOX = 'mechanicalturk.sandbox.amazonaws.com'
+  PROD = 'mechanicalturk.amazonaws.com'
+
+  # By default, MechanicalTurk will operate on the MechanicalTurk Sandbox.  To run against the main site, pass +:Host => :Production+ into the constructor.
+  def initialize(args={})
+    name = args[:Name] || 'AWSMechanicalTurkRequester'
+    software = args.has_key?(:SoftwareName) ? "#{SOFTWARE_NAME}, #{args[:SoftwareName]}" : "#{SOFTWARE_NAME}"
+    @host = case args[:Host].to_s
+      when /^Prod/i
+        PROD
+      when /^Sandbox/i,""
+        SANDBOX
+      else
+        args[:Host].to_s
+      end
+    ssl = ( args[:UseSSL].nil? ? true : args[:UseSSL] )
+    newargs = args.merge( :Name => name, :SoftwareName => software, :Host => @host, :UseSSL => ssl)
+    if args[:Transport].to_s =~ /^SOAP/i
+      unless Util::SOAPTransport.canSOAP?
+        log "Unable to use SOAP transport. Falling back to REST."
+        args[:Transport] = :REST
+      end
+    end
+    transport = case args[:Transport]
+      when :SOAP,/^SOAP/i
+        getSOAPTransport(newargs)
+      when :REST,/^REST/i
+        getRESTTransport(newargs)
+      else
+        require 'amazon/webservices/util/soap_transport.rb'
+        allowOverride( 'Transport', args[:Transport], newargs ) { |a|
+          if Util::RESTTransport.canPost?
+            getRESTTransport(newargs)
+          elsif Util::SOAPTransport.canSOAP?
+            getSOAPTransport(newargs)
+          else
+            getRESTTransport(newargs)
+          end
+        }
+      end
+    newargs.merge!( :Transport => transport )
+    log "Generating relay with following args: #{newargs.inspect}"
+    relay = allowOverride('Relay',args[:Relay],newargs) { |a| Amazon::WebServices::Util::AmazonAuthenticationRelay.new(a) }
+    newargs.merge!( :Relay => relay )
+    log "Generating error handler with the following args: #{newargs.inspect}"
+    @errorHandler = allowOverride('ErrorHandler',args[:ErrorHandler],newargs) { |a| Amazon::WebServices::MTurk::MechanicalTurkErrorHandler.new(a) }
+  end
+
+  attr_accessor :host
+
+  def method_missing(method,*args)
+    log "Sending request: #{method} #{args.inspect}"
+    @errorHandler.dispatch(method,*args)
+  end
+
+  private
+
+  def allowOverride(name,override,args,&default)
+    newargs = args.merge( :DefaultOverride => default )
+    case override
+    when nil
+        yield( args )
+    when String,Symbol,Array,Hash,Integer
+      raise "Invalid #{name}: #{override.inspect}"
+    when Class
+      override.new( newargs )
+    else
+      override.configure( newargs ) if override.respond_to? :configure
+      override
+    end
+  end
+
+  def getRESTTransport(args)
+    endpoint = findRestEndpoint( args[:Name], args[:Host], args[:UseSSL] )
+    require 'amazon/webservices/util/rest_transport.rb'
+    @transport = Amazon::WebServices::Util::RESTTransport.new( args.merge( :Endpoint => endpoint ) )
+  end
+
+  def getSOAPTransport(args)
+    wsdl = findWSDL( args[:Name], args[:Host], args[:Version], args[:UseSSL] )
+    endpoint = findSOAPEndpoint( args[:Name], args[:Host], args[:UseSSL] )
+    require 'amazon/webservices/util/soap_transport.rb'
+    Amazon::WebServices::Util::SOAPTransport.new( args.merge( :Wsdl => wsdl, :Endpoint => endpoint ) )
+  end
+
+  def findWSDL( name, host, version, ssl )
+    if version.nil?
+      "#{ssl ? 'https' : 'http'}://#{host}/AWSMechanicalTurk/#{name}.wsdl"
+    else
+      "#{ssl ? 'https' : 'http'}://#{host}/AWSMechanicalTurk/#{version}/#{name}.wsdl"
+    end
+  end
+
+  def findSOAPEndpoint( name, host, ssl )
+    "#{ssl ? 'https' : 'http'}://#{host}/onca/soap?Service=#{name}"
+  end
+
+  def findRestEndpoint( name, host, ssl )
+    "#{ssl ? 'https' : 'http'}://#{host}/?Service=#{name}"
+  end
+
+end # MTurk
+
+end # Amazon::WebServices
+end # Amazon

data/lib/amazon/webservices/mechanical_turk_requester.rb

@@ -0,0 +1,274 @@
+# Copyright:: Copyright (c) 2007 Amazon Technologies, Inc.
+# License::   Apache License, Version 2.0
+
+require 'erb'
+require 'monitor'
+require 'amazon/util'
+require 'amazon/webservices/util/xml_simplifier'
+require 'amazon/webservices/util/convenience_wrapper'
+require 'amazon/webservices/mechanical_turk'
+
+module Amazon
+module WebServices
+
+class MechanicalTurkRequester < Amazon::WebServices::Util::ConvenienceWrapper
+
+  WSDL_VERSION = "2012-03-25"
+
+  ABANDONMENT_RATE_QUALIFICATION_TYPE_ID = "00000000000000000070";
+  APPROVAL_RATE_QUALIFICATION_TYPE_ID = "000000000000000000L0";
+  REJECTION_RATE_QUALIFICATION_TYPE_ID = "000000000000000000S0";
+  RETURN_RATE_QUALIFICATION_TYPE_ID = "000000000000000000E0";
+  SUBMISSION_RATE_QUALIFICATION_TYPE_ID = "00000000000000000000";
+  LOCALE_QUALIFICATION_TYPE_ID = "00000000000000000071";
+
+  DEFAULT_THREADCOUNT = 10
+
+  serviceCall :RegisterHITType, :RegisterHITTypeResult, {
+                                                          :AssignmentDurationInSeconds => 60*60,
+                                                          :AutoApprovalDelayInSeconds => 60*60*24*7
+                                                        }
+  serviceCall :SetHITTypeNotification, :SetHITTypeNotificationResult
+
+
+  serviceCall :CreateHIT, :HIT, { :MaxAssignments => 1,
+                                  :AssignmentDurationInSeconds => 60*60,
+                                  :AutoApprovalDelayInSeconds => 60*60*24*7,
+                                  :LifetimeInSeconds => 60*60*24,
+                                }
+
+  serviceCall :DisableHIT, :DisableHITResult
+  serviceCall :DisposeHIT, :DisposeHITResult
+  serviceCall :ExtendHIT, :ExtendHITResult
+  serviceCall :ForceExpireHIT, :ForceExpireHITResult
+  serviceCall :GetHIT, :HIT, { :ResponseGroup => %w( Minimal HITDetail HITQuestion HITAssignmentSummary ) }
+  serviceCall :ChangeHITTypeOfHIT, :ChangeHITTypeOfHITResult
+
+  serviceCall :SearchHITs, :SearchHITsResult
+  serviceCall :GetReviewableHITs, :GetReviewableHITsResult
+  serviceCall :SetHITAsReviewing, :SetHITAsReviewingResult
+  serviceCall :GetAssignmentsForHIT, :GetAssignmentsForHITResult
+  serviceCall :GetReviewResultsForHIT, :GetReviewResultsForHITResult
+
+  paginate :SearchHITs, :HIT
+  paginate :GetReviewableHITs, :HIT
+  paginate :GetAssignmentsForHIT, :Assignment
+
+  serviceCall :GetAssignment, :GetAssignmentResult
+  serviceCall :ApproveAssignment, :ApproveAssignmentResult
+  serviceCall :RejectAssignment, :RejectAssignmentResult
+  serviceCall :ApproveRejectedAssignment, :ApproveRejectedAssignmentResult
+
+  serviceCall :GrantBonus, :GrantBonusResult
+  serviceCall :GetBonusPayments, :GetBonusPaymentsResult
+
+  serviceCall :CreateQualificationType, :QualificationType, { :QualificationTypeStatus => 'Active' }
+  serviceCall :GetQualificationType, :QualificationType
+  serviceCall :SearchQualificationTypes, :SearchQualificationTypesResult, { :MustBeRequestable => true }
+  serviceCall :UpdateQualificationType, :QualificationType
+  serviceCall :GetQualificationsForQualificationType, :GetQualificationsForQualificationTypeResult, { :Status => 'Granted' }
+  serviceCall :GetHITsForQualificationType, :GetHITsForQualificationTypeResult
+  serviceCall :DisposeQualficationType, :DisposeQualificationTypeResult
+
+  paginate :SearchQualificationTypes, :QualificationType
+  paginate :GetQualificationsForQualificationType, :Qualification
+
+  serviceCall :AssignQualification, :AssignQualificationResult
+  serviceCall :GetQualificationRequests, :GetQualificationRequestsResult
+  serviceCall :GrantQualification, :GrantQualificationResult
+  serviceCall :RejectQualificationRequest, :RejectQualificationRequestResult
+  serviceCall :GetQualificationScore, :Qualification
+  serviceCall :UpdateQualificationScore, :UpdateQualificationScoreResult
+  serviceCall :RevokeQualification, :RevokeQualificationResult
+
+  paginate :GetQualificationRequests, :QualificationRequest
+
+  serviceCall :GetBlockedWorkers, :GetBlockedWorkersResult
+  serviceCall :BlockWorker, :BlockWorkerResult
+  serviceCall :UnblockWorker, :UnblockWorkerResult
+
+  paginate :GetBlockedWorkers, :WorkerBlock
+
+  serviceCall :GetFileUploadURL, :GetFileUploadURLResult
+  serviceCall :GetAccountBalance, :GetAccountBalanceResult
+  serviceCall :GetRequesterStatistic, :GetStatisticResult, { :Count => 1 }
+  serviceCall :GetRequesterWorkerStatistic, :GetStatisticResult, { :Count => 1 }
+
+  serviceCall :NotifyWorkers, :NotifyWorkersResult
+
+  def initialize(args={})
+    newargs = args.dup
+    unless args[:Config].nil?
+      if args[:Config] == :Rails
+        rails_config = Amazon::Util::DataReader.load( File.join(::RAILS_ROOT,'config','mturk.yml'), :YAML )
+        newargs = args.merge rails_config[::RAILS_ENV].inject({})  {|a,b| a[b[0].to_sym] = b[1] ; a }
+      else
+        loaded = Amazon::Util::DataReader.load( args[:Config], :YAML )
+        newargs = args.merge loaded.inject({}) {|a,b| a[b[0].to_sym] = b[1] ; a }
+      end
+    end
+    @threadcount = args[:ThreadCount].to_i
+    @threadcount = DEFAULT_THREADCOUNT unless @threadcount >= 1
+    raise "Cannot override WSDL version ( #{WSDL_VERSION} )" unless args[:Version].nil? or args[:Version].equals? WSDL_VERSION
+    super newargs.merge( :Name => :AWSMechanicalTurkRequester,
+                         :ServiceClass => Amazon::WebServices::MechanicalTurk,
+                         :Version => WSDL_VERSION )
+  end
+
+  # Create a series of similar HITs, sharing common parameters.  Utilizes HITType
+  # * hit_template is the array of parameters to pass to createHIT.
+  # * question_template will be passed as a template into ERB to generate the :Question parameter
+  # * the RequesterAnnotation parameter of hit_template will also be passed through ERB
+  # * hit_data_set should consist of an array of hashes defining unique instance variables utilized by question_template
+  def createHITs( hit_template, question_template, hit_data_set )
+    hit_template = hit_template.dup
+    lifetime = hit_template[:LifetimeInSeconds]
+    numassignments_template = hit_template[:MaxAssignments]
+    annotation_template = hit_template[:RequesterAnnotation]
+    hit_template.delete :LifetimeInSeconds
+    hit_template.delete :MaxAssignments
+    hit_template.delete :RequesterAnnotation
+
+    ht = hit_template[:HITTypeId] || registerHITType( hit_template )[:HITTypeId]
+
+    tp = Amazon::Util::ThreadPool.new @threadcount
+
+    created = [].extend(MonitorMixin)
+    failed = [].extend(MonitorMixin)
+    hit_data_set.each do |hd|
+      tp.addWork(hd) do |hit_data|
+        begin
+          b = Amazon::Util::Binder.new( hit_data )
+          annotation = annotation_template.nil? ? nil : b.erb_eval( annotation_template )
+          numassignments = numassignments_template.nil? ? nil : b.erb_eval( numassignments_template.to_s ).to_i
+          question = b.erb_eval( question_template )
+          result = self.createHIT( :HITTypeId => ht,
+                                   :LifetimeInSeconds => lifetime,
+                                   :MaxAssignments => ( hit_data[:MaxAssignments] || numassignments || 1 ),
+                                   :Question => question,
+                                   :RequesterAnnotation => ( hit_data[:RequesterAnnotation] || annotation || "")
+                                 )
+          created.synchronize do
+            created << result
+          end
+        rescue => e
+          failed.synchronize do
+            failed << hit_data.merge( :Error => e.message )
+          end
+        end
+      end # tp.addWork
+    end # hit_data_set.each
+    tp.finish
+
+    return :Created => created, :Failed => failed
+  end
+
+  # Update a series of HITs to belong to a new HITType
+  # * hit_template is the array of parameters to pass to registerHITType
+  # * hit_ids is a list of HITIds (strings)
+  def updateHITs( hit_template, hit_ids )
+    hit_template = hit_template.dup
+    hit_template.delete :LifetimeInSeconds
+    hit_template.delete :RequesterAnnotation
+
+    hit_type_id = registerHITType( hit_template )[:HITTypeId]
+
+    tp = Amazon::Util::ThreadPool.new @threadcount
+
+    updated = [].extend(MonitorMixin)
+    failed = [].extend(MonitorMixin)
+    hit_ids.each do |hid|
+      tp.addWork(hid) do |hit_id|
+        begin
+          changeHITTypeOfHIT( :HITId => hit_id, :HITTypeId => hit_type_id )
+          updated.synchronize do
+            updated << hit_id
+          end
+        rescue => e
+          failed.synchronize do
+            failed << { :HITId => hit_id, :Error => e.message }
+          end
+        end
+      end # tp.addWork
+    end # hit_ids.each
+    tp.finish
+
+    return :Updated => updated, :Failed => failed
+  end
+
+
+  # Update a HIT with new properties.
+  # hit_id:: Id of the HIT to update
+  # hit_template:: hash ( parameter => value ) of parameters to update
+  #
+  # Acceptable attributes:
+  # * Title
+  # * Description
+  # * Keywords
+  # * Reward
+  # * QualificationRequirement
+  # * AutoApprovalDelayInSeconds
+  # * AssignmentDurationInSeconds
+  #
+  # Behind the scenes, this function retrieves the HIT, merges the HITs
+  # current attributes with any you specify, and registers a new HIT
+  # Template.  It then uses the new ChangeHITTypeOfHIT function to move
+  # your HIT to the newly-created HIT Template.
+  def updateHIT( hit_id, hit_template )
+    hit_template = hit_template.dup
+
+    hit = getHIT( :HITId => hit_id )
+
+    props = %w( Title Description Keywords Reward QualificationRequirement
+                AutoApprovalDelayInSeconds AssignmentDurationInSeconds
+              ).collect {|str| str.to_sym }
+
+    props.each do |p|
+      hit_template[p] = hit[p] if hit_template[p].nil?
+    end
+
+    hit_type_id = registerHITType( hit_template )[:HITTypeId]
+
+    changeHITTypeOfHIT( :HITId => hit_id, :HITTypeId => hit_type_id )
+  end
+
+
+  def getHITResults( list )
+    results = [].extend(MonitorMixin)
+    tp = Amazon::Util::ThreadPool.new @threadcount
+    list.each do |line|
+      tp.addWork(line) do |h|
+        hit = getHIT( :HITId => h[:HITId] )
+        getAssignmentsForHITAll( :HITId => h[:HITId] ).each {|assignment|
+          results.synchronize do
+            results << ( hit.merge( assignment ) )
+          end
+        }
+      end
+    end
+    tp.finish
+    results.flatten
+  end
+
+  # Returns available funds in USD
+  # Calls getAccountBalance and parses out the correct amount
+  def availableFunds
+    return getAccountBalance[:AvailableBalance][:Amount]
+  end
+
+  # helper function to simplify answer XML
+  def simplifyAnswer( answerXML )
+    answerHash = Amazon::WebServices::Util::XMLSimplifier.simplify REXML::Document.new(answerXML)
+    list = [answerHash[:Answer]].flatten
+    list.inject({}) { |answers, answer|
+      id = answer[:QuestionIdentifier]
+      result = answer[:FreeText] || answer[:SelectionIdentifier] || answer[:UploadedFileKey]
+      answers[id] = result
+      answers
+    }
+  end
+
+end # MechanicalTurkRequester
+
+end # Amazon::WebServices
+end # Amazon

data/lib/amazon/webservices/mturk/mechanical_turk_error_handler.rb

@@ -0,0 +1,150 @@
+# Copyright:: Copyright (c) 2007 Amazon Technologies, Inc.
+# License::   Apache License, Version 2.0
+
+require 'amazon/util/logging'
+require 'amazon/webservices/util/validation_exception'
+require 'amazon/webservices/util/unknown_result_exception'
+
+module Amazon
+module WebServices
+module MTurk
+
+class MechanicalTurkErrorHandler
+  include Amazon::Util::Logging
+
+  REQUIRED_PARAMETERS = [:Relay]
+
+  # Commands with these prefixes can be retried if we are unsure of success
+  RETRY_PRE = %w( search get register update disable assign set dispose )
+
+  # Max number of times to retry a call
+  MAX_RETRY = 6
+
+  # Base used in Exponential Backoff retry delay
+  BACKOFF_BASE = 2
+  # Scale factor for Exponential Backoff retry delay
+  BACKOFF_INITIAL = 0.1
+
+  # Matching pattern to find a 'Results' element in the Response
+  RESULT_PATTERN = /Result/
+  # Additional elements to be considered a 'Result' despite not matching RESULT_PATTERN
+  ACCEPTABLE_RESULTS = %w( HIT Qualification QualificationType QualificationRequest Information )
+
+  def initialize( args )
+    missing_parameters = REQUIRED_PARAMETERS - args.keys
+    raise "Missing paramters: #{missing_parameters.join(',')}" unless missing_parameters.empty?
+    @relay = args[:Relay]
+  end
+
+  def dispatch(method, *args)
+    try = 0
+    begin
+      try += 1
+      log "Dispatching call to #{method} (try #{try})"
+      response = @relay.send(method,*args)
+      validateResponse( response )
+      return response
+    rescue Exception => error
+      case handleError( error,method )
+      when :RetryWithBackoff
+        retry if doBackoff( try )
+      when :RetryImmediate
+        retry if canRetry( try )
+      when :Ignore
+        return :IgnoredError => error
+      when :Unknown
+        raise Util::UnknownResultException.new( error, method, args )
+      when :Fail
+        raise error
+      else
+        raise "Unknown error handling method: #{handleError( error,method )}"
+      end
+      raise error
+    end
+  end
+
+  def methodRetryable( method )
+    RETRY_PRE.each do |pre|
+      return true if method.to_s =~ /^#{pre}/i
+    end
+    return false
+  end
+
+  def handleError( error, method )
+    log "Handling error: #{error.inspect}"
+    case error.class.to_s
+    when 'Timeout::Error','SOAP::HTTPStreamError'
+      if methodRetryable( method )
+        return :RetryImmediate
+      else
+        return :Unknown
+      end
+    when 'SOAP::FaultError'
+      case error.faultcode.data
+      when "aws:Server.ServiceUnavailable"
+        return :RetryWithBackoff
+      else
+        return :Unknown
+      end
+    when 'Amazon::WebServices::Util::ValidationException'
+      case error.message
+      when 'AWS.ServiceUnavailable'
+        return :RetryWithBackoff if methodRetryable( method )
+      end
+      return :Fail
+    when 'RuntimeError'
+      case error.message
+      when 'Throttled'
+        return :RetryWithBackoff
+      else
+        return :RetryImmediate
+      end
+    else
+      return :Unknown
+    end
+  end
+
+  def canRetry( try )
+    try <= MAX_RETRY
+  end
+
+  def doBackoff( try )
+    return false unless canRetry(try)
+    delay = BACKOFF_INITIAL * ( BACKOFF_BASE ** try )
+    sleep delay
+    return true
+  end
+
+  def isResultTag( tag )
+    tag.to_s =~ RESULT_PATTERN or ACCEPTABLE_RESULTS.include?( tag.to_s )
+  end
+
+  def validateResponse(response)
+    log "Validating response: #{response.inspect}"
+    if response[:Errors] and response[:Errors][:Error]
+      if response[:Errors][:Error][:Code] == "ServiceUnavailable"
+        raise 'Throttled'
+      else
+        raise Util::ValidationException.new(response)
+      end
+    end
+    if response[:OperationRequest] and response[:OperationRequest][:Errors]
+      raise Util::ValidationException.new(response)
+    end
+    resultTags = response.keys.find_all {|r| isResultTag( r ) }
+    raise Util::ValidationException.new(response, "Didn't get back an acceptable result tag (got back #{response.keys.join(',')})") if resultTags.empty?
+    resultTags.each do |resultTag|
+      log "using result tag <#{resultTag}>"
+      result = response[resultTag]
+      if result[:Request] and result[:Request][:Errors]
+        raise Util::ValidationException.new(result)
+      end
+    end
+    response
+  end
+
+end # MechanicalTurkErrorHandler
+
+end # Amazon::WebServices::MTurk
+end # Amazon::WebServices
+end # Amazon