Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(150)

Unified Diff: third_party/gsutil/third_party/boto/boto/mturk/connection.py

Issue 1377933002: [catapult] - Copy Telemetry's gsutilz over to third_party. (Closed) Base URL: https://github.com/catapult-project/catapult.git@master
Patch Set: Rename to gsutil. Created 5 years, 3 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: third_party/gsutil/third_party/boto/boto/mturk/connection.py
diff --git a/third_party/gsutil/third_party/boto/boto/mturk/connection.py b/third_party/gsutil/third_party/boto/boto/mturk/connection.py
new file mode 100644
index 0000000000000000000000000000000000000000..4f2a23faf1a8a8f7e8d78fbcf8cc4302b5bb0666
--- /dev/null
+++ b/third_party/gsutil/third_party/boto/boto/mturk/connection.py
@@ -0,0 +1,1052 @@
+# Copyright (c) 2006,2007 Mitch Garnaat http://garnaat.org/
+#
+# Permission is hereby granted, free of charge, to any person obtaining a
+# copy of this software and associated documentation files (the
+# "Software"), to deal in the Software without restriction, including
+# without limitation the rights to use, copy, modify, merge, publish, dis-
+# tribute, sublicense, and/or sell copies of the Software, and to permit
+# persons to whom the Software is furnished to do so, subject to the fol-
+# lowing conditions:
+#
+# The above copyright notice and this permission notice shall be included
+# in all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
+# OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABIL-
+# ITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT
+# SHALL THE AUTHOR BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
+# WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
+# IN THE SOFTWARE.
+import xml.sax
+import datetime
+import itertools
+
+from boto import handler
+from boto import config
+from boto.mturk.price import Price
+import boto.mturk.notification
+from boto.connection import AWSQueryConnection
+from boto.exception import EC2ResponseError
+from boto.resultset import ResultSet
+from boto.mturk.question import QuestionForm, ExternalQuestion, HTMLQuestion
+
+
+class MTurkRequestError(EC2ResponseError):
+ "Error for MTurk Requests"
+ # todo: subclass from an abstract parent of EC2ResponseError
+
+
+class MTurkConnection(AWSQueryConnection):
+
+ APIVersion = '2012-03-25'
+
+ def __init__(self, aws_access_key_id=None, aws_secret_access_key=None,
+ is_secure=True, port=None, proxy=None, proxy_port=None,
+ proxy_user=None, proxy_pass=None,
+ host=None, debug=0,
+ https_connection_factory=None, security_token=None,
+ profile_name=None):
+ if not host:
+ if config.has_option('MTurk', 'sandbox') and config.get('MTurk', 'sandbox') == 'True':
+ host = 'mechanicalturk.sandbox.amazonaws.com'
+ else:
+ host = 'mechanicalturk.amazonaws.com'
+ self.debug = debug
+
+ super(MTurkConnection, self).__init__(aws_access_key_id,
+ aws_secret_access_key,
+ is_secure, port, proxy, proxy_port,
+ proxy_user, proxy_pass, host, debug,
+ https_connection_factory,
+ security_token=security_token,
+ profile_name=profile_name)
+
+ def _required_auth_capability(self):
+ return ['mturk']
+
+ def get_account_balance(self):
+ """
+ """
+ params = {}
+ return self._process_request('GetAccountBalance', params,
+ [('AvailableBalance', Price),
+ ('OnHoldBalance', Price)])
+
+ def register_hit_type(self, title, description, reward, duration,
+ keywords=None, approval_delay=None, qual_req=None):
+ """
+ Register a new HIT Type
+ title, description are strings
+ reward is a Price object
+ duration can be a timedelta, or an object castable to an int
+ """
+ params = dict(
+ Title=title,
+ Description=description,
+ AssignmentDurationInSeconds=self.duration_as_seconds(duration),
+ )
+ params.update(MTurkConnection.get_price_as_price(reward).get_as_params('Reward'))
+
+ if keywords:
+ params['Keywords'] = self.get_keywords_as_string(keywords)
+
+ if approval_delay is not None:
+ d = self.duration_as_seconds(approval_delay)
+ params['AutoApprovalDelayInSeconds'] = d
+
+ if qual_req is not None:
+ params.update(qual_req.get_as_params())
+
+ return self._process_request('RegisterHITType', params,
+ [('HITTypeId', HITTypeId)])
+
+ def set_email_notification(self, hit_type, email, event_types=None):
+ """
+ Performs a SetHITTypeNotification operation to set email
+ notification for a specified HIT type
+ """
+ return self._set_notification(hit_type, 'Email', email,
+ 'SetHITTypeNotification', event_types)
+
+ def set_rest_notification(self, hit_type, url, event_types=None):
+ """
+ Performs a SetHITTypeNotification operation to set REST notification
+ for a specified HIT type
+ """
+ return self._set_notification(hit_type, 'REST', url,
+ 'SetHITTypeNotification', event_types)
+
+ def set_sqs_notification(self, hit_type, queue_url, event_types=None):
+ """
+ Performs a SetHITTypeNotification operation so set SQS notification
+ for a specified HIT type. Queue URL is of form:
+ https://queue.amazonaws.com/<CUSTOMER_ID>/<QUEUE_NAME> and can be
+ found when looking at the details for a Queue in the AWS Console
+ """
+ return self._set_notification(hit_type, "SQS", queue_url,
+ 'SetHITTypeNotification', event_types)
+
+ def send_test_event_notification(self, hit_type, url,
+ event_types=None,
+ test_event_type='Ping'):
+ """
+ Performs a SendTestEventNotification operation with REST notification
+ for a specified HIT type
+ """
+ return self._set_notification(hit_type, 'REST', url,
+ 'SendTestEventNotification',
+ event_types, test_event_type)
+
+ def _set_notification(self, hit_type, transport,
+ destination, request_type,
+ event_types=None, test_event_type=None):
+ """
+ Common operation to set notification or send a test event
+ notification for a specified HIT type
+ """
+ params = {'HITTypeId': hit_type}
+
+ # from the Developer Guide:
+ # The 'Active' parameter is optional. If omitted, the active status of
+ # the HIT type's notification specification is unchanged. All HIT types
+ # begin with their notification specifications in the "inactive" status.
+ notification_params = {'Destination': destination,
+ 'Transport': transport,
+ 'Version': boto.mturk.notification.NotificationMessage.NOTIFICATION_VERSION,
+ 'Active': True,
+ }
+
+ # add specific event types if required
+ if event_types:
+ self.build_list_params(notification_params, event_types,
+ 'EventType')
+
+ # Set up dict of 'Notification.1.Transport' etc. values
+ notification_rest_params = {}
+ num = 1
+ for key in notification_params:
+ notification_rest_params['Notification.%d.%s' % (num, key)] = notification_params[key]
+
+ # Update main params dict
+ params.update(notification_rest_params)
+
+ # If test notification, specify the notification type to be tested
+ if test_event_type:
+ params.update({'TestEventType': test_event_type})
+
+ # Execute operation
+ return self._process_request(request_type, params)
+
+ def create_hit(self, hit_type=None, question=None, hit_layout=None,
+ lifetime=datetime.timedelta(days=7),
+ max_assignments=1,
+ title=None, description=None, keywords=None,
+ reward=None, duration=datetime.timedelta(days=7),
+ approval_delay=None, annotation=None,
+ questions=None, qualifications=None,
+ layout_params=None, response_groups=None):
+ """
+ Creates a new HIT.
+ Returns a ResultSet
+ See: http://docs.amazonwebservices.com/AWSMechTurk/2012-03-25/AWSMturkAPI/ApiReference_CreateHITOperation.html
+ """
+
+ # Handle basic required arguments and set up params dict
+ params = {'LifetimeInSeconds':
+ self.duration_as_seconds(lifetime),
+ 'MaxAssignments': max_assignments,
+ }
+
+ # handle single or multiple questions or layouts
+ neither = question is None and questions is None
+ if hit_layout is None:
+ both = question is not None and questions is not None
+ if neither or both:
+ raise ValueError("Must specify question (single Question instance) or questions (list or QuestionForm instance), but not both")
+ if question:
+ questions = [question]
+ question_param = QuestionForm(questions)
+ if isinstance(question, QuestionForm):
+ question_param = question
+ elif isinstance(question, ExternalQuestion):
+ question_param = question
+ elif isinstance(question, HTMLQuestion):
+ question_param = question
+ params['Question'] = question_param.get_as_xml()
+ else:
+ if not neither:
+ raise ValueError("Must not specify question (single Question instance) or questions (list or QuestionForm instance) when specifying hit_layout")
+ params['HITLayoutId'] = hit_layout
+ if layout_params:
+ params.update(layout_params.get_as_params())
+
+ # if hit type specified then add it
+ # else add the additional required parameters
+ if hit_type:
+ params['HITTypeId'] = hit_type
+ else:
+ # Handle keywords
+ final_keywords = MTurkConnection.get_keywords_as_string(keywords)
+
+ # Handle price argument
+ final_price = MTurkConnection.get_price_as_price(reward)
+
+ final_duration = self.duration_as_seconds(duration)
+
+ additional_params = dict(
+ Title=title,
+ Description=description,
+ Keywords=final_keywords,
+ AssignmentDurationInSeconds=final_duration,
+ )
+ additional_params.update(final_price.get_as_params('Reward'))
+
+ if approval_delay is not None:
+ d = self.duration_as_seconds(approval_delay)
+ additional_params['AutoApprovalDelayInSeconds'] = d
+
+ # add these params to the others
+ params.update(additional_params)
+
+ # add the annotation if specified
+ if annotation is not None:
+ params['RequesterAnnotation'] = annotation
+
+ # Add the Qualifications if specified
+ if qualifications is not None:
+ params.update(qualifications.get_as_params())
+
+ # Handle optional response groups argument
+ if response_groups:
+ self.build_list_params(params, response_groups, 'ResponseGroup')
+
+ # Submit
+ return self._process_request('CreateHIT', params, [('HIT', HIT)])
+
+ def change_hit_type_of_hit(self, hit_id, hit_type):
+ """
+ Change the HIT type of an existing HIT. Note that the reward associated
+ with the new HIT type must match the reward of the current HIT type in
+ order for the operation to be valid.
+
+ :type hit_id: str
+ :type hit_type: str
+ """
+ params = {'HITId': hit_id,
+ 'HITTypeId': hit_type}
+
+ return self._process_request('ChangeHITTypeOfHIT', params)
+
+ def get_reviewable_hits(self, hit_type=None, status='Reviewable',
+ sort_by='Expiration', sort_direction='Ascending',
+ page_size=10, page_number=1):
+ """
+ Retrieve the HITs that have a status of Reviewable, or HITs that
+ have a status of Reviewing, and that belong to the Requester
+ calling the operation.
+ """
+ params = {'Status': status,
+ 'SortProperty': sort_by,
+ 'SortDirection': sort_direction,
+ 'PageSize': page_size,
+ 'PageNumber': page_number}
+
+ # Handle optional hit_type argument
+ if hit_type is not None:
+ params.update({'HITTypeId': hit_type})
+
+ return self._process_request('GetReviewableHITs', params,
+ [('HIT', HIT)])
+
+ @staticmethod
+ def _get_pages(page_size, total_records):
+ """
+ Given a page size (records per page) and a total number of
+ records, return the page numbers to be retrieved.
+ """
+ pages = total_records / page_size + bool(total_records % page_size)
+ return list(range(1, pages + 1))
+
+ def get_all_hits(self):
+ """
+ Return all of a Requester's HITs
+
+ Despite what search_hits says, it does not return all hits, but
+ instead returns a page of hits. This method will pull the hits
+ from the server 100 at a time, but will yield the results
+ iteratively, so subsequent requests are made on demand.
+ """
+ page_size = 100
+ search_rs = self.search_hits(page_size=page_size)
+ total_records = int(search_rs.TotalNumResults)
+ get_page_hits = lambda page: self.search_hits(page_size=page_size, page_number=page)
+ page_nums = self._get_pages(page_size, total_records)
+ hit_sets = itertools.imap(get_page_hits, page_nums)
+ return itertools.chain.from_iterable(hit_sets)
+
+ def search_hits(self, sort_by='CreationTime', sort_direction='Ascending',
+ page_size=10, page_number=1, response_groups=None):
+ """
+ Return a page of a Requester's HITs, on behalf of the Requester.
+ The operation returns HITs of any status, except for HITs that
+ have been disposed with the DisposeHIT operation.
+ Note:
+ The SearchHITs operation does not accept any search parameters
+ that filter the results.
+ """
+ params = {'SortProperty': sort_by,
+ 'SortDirection': sort_direction,
+ 'PageSize': page_size,
+ 'PageNumber': page_number}
+ # Handle optional response groups argument
+ if response_groups:
+ self.build_list_params(params, response_groups, 'ResponseGroup')
+
+ return self._process_request('SearchHITs', params, [('HIT', HIT)])
+
+ def get_assignment(self, assignment_id, response_groups=None):
+ """
+ Retrieves an assignment using the assignment's ID. Requesters can only
+ retrieve their own assignments, and only assignments whose related HIT
+ has not been disposed.
+
+ The returned ResultSet will have the following attributes:
+
+ Request
+ This element is present only if the Request ResponseGroup
+ is specified.
+ Assignment
+ The assignment. The response includes one Assignment object.
+ HIT
+ The HIT associated with this assignment. The response
+ includes one HIT object.
+
+ """
+
+ params = {'AssignmentId': assignment_id}
+
+ # Handle optional response groups argument
+ if response_groups:
+ self.build_list_params(params, response_groups, 'ResponseGroup')
+
+ return self._process_request('GetAssignment', params,
+ [('Assignment', Assignment),
+ ('HIT', HIT)])
+
+ def get_assignments(self, hit_id, status=None,
+ sort_by='SubmitTime', sort_direction='Ascending',
+ page_size=10, page_number=1, response_groups=None):
+ """
+ Retrieves completed assignments for a HIT.
+ Use this operation to retrieve the results for a HIT.
+
+ The returned ResultSet will have the following attributes:
+
+ NumResults
+ The number of assignments on the page in the filtered results
+ list, equivalent to the number of assignments being returned
+ by this call.
+ A non-negative integer, as a string.
+ PageNumber
+ The number of the page in the filtered results list being
+ returned.
+ A positive integer, as a string.
+ TotalNumResults
+ The total number of HITs in the filtered results list based
+ on this call.
+ A non-negative integer, as a string.
+
+ The ResultSet will contain zero or more Assignment objects
+
+ """
+ params = {'HITId': hit_id,
+ 'SortProperty': sort_by,
+ 'SortDirection': sort_direction,
+ 'PageSize': page_size,
+ 'PageNumber': page_number}
+
+ if status is not None:
+ params['AssignmentStatus'] = status
+
+ # Handle optional response groups argument
+ if response_groups:
+ self.build_list_params(params, response_groups, 'ResponseGroup')
+
+ return self._process_request('GetAssignmentsForHIT', params,
+ [('Assignment', Assignment)])
+
+ def approve_assignment(self, assignment_id, feedback=None):
+ """
+ """
+ params = {'AssignmentId': assignment_id}
+ if feedback:
+ params['RequesterFeedback'] = feedback
+ return self._process_request('ApproveAssignment', params)
+
+ def reject_assignment(self, assignment_id, feedback=None):
+ """
+ """
+ params = {'AssignmentId': assignment_id}
+ if feedback:
+ params['RequesterFeedback'] = feedback
+ return self._process_request('RejectAssignment', params)
+
+ def approve_rejected_assignment(self, assignment_id, feedback=None):
+ """
+ """
+ params = {'AssignmentId': assignment_id}
+ if feedback:
+ params['RequesterFeedback'] = feedback
+ return self._process_request('ApproveRejectedAssignment', params)
+
+ def get_file_upload_url(self, assignment_id, question_identifier):
+ """
+ Generates and returns a temporary URL to an uploaded file. The
+ temporary URL is used to retrieve the file as an answer to a
+ FileUploadAnswer question, it is valid for 60 seconds.
+
+ Will have a FileUploadURL attribute as per the API Reference.
+ """
+
+ params = {'AssignmentId': assignment_id,
+ 'QuestionIdentifier': question_identifier}
+
+ return self._process_request('GetFileUploadURL', params,
+ [('FileUploadURL', FileUploadURL)])
+
+ def get_hit(self, hit_id, response_groups=None):
+ """
+ """
+ params = {'HITId': hit_id}
+ # Handle optional response groups argument
+ if response_groups:
+ self.build_list_params(params, response_groups, 'ResponseGroup')
+
+ return self._process_request('GetHIT', params, [('HIT', HIT)])
+
+ def set_reviewing(self, hit_id, revert=None):
+ """
+ Update a HIT with a status of Reviewable to have a status of Reviewing,
+ or reverts a Reviewing HIT back to the Reviewable status.
+
+ Only HITs with a status of Reviewable can be updated with a status of
+ Reviewing. Similarly, only Reviewing HITs can be reverted back to a
+ status of Reviewable.
+ """
+ params = {'HITId': hit_id}
+ if revert:
+ params['Revert'] = revert
+ return self._process_request('SetHITAsReviewing', params)
+
+ def disable_hit(self, hit_id, response_groups=None):
+ """
+ Remove a HIT from the Mechanical Turk marketplace, approves all
+ submitted assignments that have not already been approved or rejected,
+ and disposes of the HIT and all assignment data.
+
+ Assignments for the HIT that have already been submitted, but not yet
+ approved or rejected, will be automatically approved. Assignments in
+ progress at the time of the call to DisableHIT will be approved once
+ the assignments are submitted. You will be charged for approval of
+ these assignments. DisableHIT completely disposes of the HIT and
+ all submitted assignment data. Assignment results data cannot be
+ retrieved for a HIT that has been disposed.
+
+ It is not possible to re-enable a HIT once it has been disabled.
+ To make the work from a disabled HIT available again, create a new HIT.
+ """
+ params = {'HITId': hit_id}
+ # Handle optional response groups argument
+ if response_groups:
+ self.build_list_params(params, response_groups, 'ResponseGroup')
+
+ return self._process_request('DisableHIT', params)
+
+ def dispose_hit(self, hit_id):
+ """
+ Dispose of a HIT that is no longer needed.
+
+ Only HITs in the "reviewable" state, with all submitted
+ assignments approved or rejected, can be disposed. A Requester
+ can call GetReviewableHITs to determine which HITs are
+ reviewable, then call GetAssignmentsForHIT to retrieve the
+ assignments. Disposing of a HIT removes the HIT from the
+ results of a call to GetReviewableHITs. """
+ params = {'HITId': hit_id}
+ return self._process_request('DisposeHIT', params)
+
+ def expire_hit(self, hit_id):
+
+ """
+ Expire a HIT that is no longer needed.
+
+ The effect is identical to the HIT expiring on its own. The
+ HIT no longer appears on the Mechanical Turk web site, and no
+ new Workers are allowed to accept the HIT. Workers who have
+ accepted the HIT prior to expiration are allowed to complete
+ it or return it, or allow the assignment duration to elapse
+ (abandon the HIT). Once all remaining assignments have been
+ submitted, the expired HIT becomes"reviewable", and will be
+ returned by a call to GetReviewableHITs.
+ """
+ params = {'HITId': hit_id}
+ return self._process_request('ForceExpireHIT', params)
+
+ def extend_hit(self, hit_id, assignments_increment=None,
+ expiration_increment=None):
+ """
+ Increase the maximum number of assignments, or extend the
+ expiration date, of an existing HIT.
+
+ NOTE: If a HIT has a status of Reviewable and the HIT is
+ extended to make it Available, the HIT will not be returned by
+ GetReviewableHITs, and its submitted assignments will not be
+ returned by GetAssignmentsForHIT, until the HIT is Reviewable
+ again. Assignment auto-approval will still happen on its
+ original schedule, even if the HIT has been extended. Be sure
+ to retrieve and approve (or reject) submitted assignments
+ before extending the HIT, if so desired.
+ """
+ # must provide assignment *or* expiration increment
+ if (assignments_increment is None and expiration_increment is None) or \
+ (assignments_increment is not None and expiration_increment is not None):
+ raise ValueError("Must specify either assignments_increment or expiration_increment, but not both")
+
+ params = {'HITId': hit_id}
+ if assignments_increment:
+ params['MaxAssignmentsIncrement'] = assignments_increment
+ if expiration_increment:
+ params['ExpirationIncrementInSeconds'] = expiration_increment
+
+ return self._process_request('ExtendHIT', params)
+
+ def get_help(self, about, help_type='Operation'):
+ """
+ Return information about the Mechanical Turk Service
+ operations and response group NOTE - this is basically useless
+ as it just returns the URL of the documentation
+
+ help_type: either 'Operation' or 'ResponseGroup'
+ """
+ params = {'About': about, 'HelpType': help_type}
+ return self._process_request('Help', params)
+
+ def grant_bonus(self, worker_id, assignment_id, bonus_price, reason):
+ """
+ Issues a payment of money from your account to a Worker. To
+ be eligible for a bonus, the Worker must have submitted
+ results for one of your HITs, and have had those results
+ approved or rejected. This payment happens separately from the
+ reward you pay to the Worker when you approve the Worker's
+ assignment. The Bonus must be passed in as an instance of the
+ Price object.
+ """
+ params = bonus_price.get_as_params('BonusAmount', 1)
+ params['WorkerId'] = worker_id
+ params['AssignmentId'] = assignment_id
+ params['Reason'] = reason
+
+ return self._process_request('GrantBonus', params)
+
+ def block_worker(self, worker_id, reason):
+ """
+ Block a worker from working on my tasks.
+ """
+ params = {'WorkerId': worker_id, 'Reason': reason}
+
+ return self._process_request('BlockWorker', params)
+
+ def unblock_worker(self, worker_id, reason):
+ """
+ Unblock a worker from working on my tasks.
+ """
+ params = {'WorkerId': worker_id, 'Reason': reason}
+
+ return self._process_request('UnblockWorker', params)
+
+ def notify_workers(self, worker_ids, subject, message_text):
+ """
+ Send a text message to workers.
+ """
+ params = {'Subject': subject,
+ 'MessageText': message_text}
+ self.build_list_params(params, worker_ids, 'WorkerId')
+
+ return self._process_request('NotifyWorkers', params)
+
+ def create_qualification_type(self,
+ name,
+ description,
+ status,
+ keywords=None,
+ retry_delay=None,
+ test=None,
+ answer_key=None,
+ answer_key_xml=None,
+ test_duration=None,
+ auto_granted=False,
+ auto_granted_value=1):
+ """
+ Create a new Qualification Type.
+
+ name: This will be visible to workers and must be unique for a
+ given requester.
+
+ description: description shown to workers. Max 2000 characters.
+
+ status: 'Active' or 'Inactive'
+
+ keywords: list of keyword strings or comma separated string.
+ Max length of 1000 characters when concatenated with commas.
+
+ retry_delay: number of seconds after requesting a
+ qualification the worker must wait before they can ask again.
+ If not specified, workers can only request this qualification
+ once.
+
+ test: a QuestionForm
+
+ answer_key: an XML string of your answer key, for automatically
+ scored qualification tests.
+ (Consider implementing an AnswerKey class for this to support.)
+
+ test_duration: the number of seconds a worker has to complete the test.
+
+ auto_granted: if True, requests for the Qualification are granted
+ immediately. Can't coexist with a test.
+
+ auto_granted_value: auto_granted qualifications are given this value.
+
+ """
+
+ params = {'Name': name,
+ 'Description': description,
+ 'QualificationTypeStatus': status,
+ }
+ if retry_delay is not None:
+ params['RetryDelayInSeconds'] = retry_delay
+
+ if test is not None:
+ assert(isinstance(test, QuestionForm))
+ assert(test_duration is not None)
+ params['Test'] = test.get_as_xml()
+
+ if test_duration is not None:
+ params['TestDurationInSeconds'] = test_duration
+
+ if answer_key is not None:
+ if isinstance(answer_key, basestring):
+ params['AnswerKey'] = answer_key # xml
+ else:
+ raise TypeError
+ # Eventually someone will write an AnswerKey class.
+
+ if auto_granted:
+ assert(test is None)
+ params['AutoGranted'] = True
+ params['AutoGrantedValue'] = auto_granted_value
+
+ if keywords:
+ params['Keywords'] = self.get_keywords_as_string(keywords)
+
+ return self._process_request('CreateQualificationType', params,
+ [('QualificationType',
+ QualificationType)])
+
+ def get_qualification_type(self, qualification_type_id):
+ params = {'QualificationTypeId': qualification_type_id }
+ return self._process_request('GetQualificationType', params,
+ [('QualificationType', QualificationType)])
+
+ def get_all_qualifications_for_qual_type(self, qualification_type_id):
+ page_size = 100
+ search_qual = self.get_qualifications_for_qualification_type(qualification_type_id)
+ total_records = int(search_qual.TotalNumResults)
+ get_page_quals = lambda page: self.get_qualifications_for_qualification_type(qualification_type_id = qualification_type_id, page_size=page_size, page_number = page)
+ page_nums = self._get_pages(page_size, total_records)
+ qual_sets = itertools.imap(get_page_quals, page_nums)
+ return itertools.chain.from_iterable(qual_sets)
+
+ def get_qualifications_for_qualification_type(self, qualification_type_id, page_size=100, page_number = 1):
+ params = {'QualificationTypeId': qualification_type_id,
+ 'PageSize': page_size,
+ 'PageNumber': page_number}
+ return self._process_request('GetQualificationsForQualificationType', params,
+ [('Qualification', Qualification)])
+
+ def update_qualification_type(self, qualification_type_id,
+ description=None,
+ status=None,
+ retry_delay=None,
+ test=None,
+ answer_key=None,
+ test_duration=None,
+ auto_granted=None,
+ auto_granted_value=None):
+
+ params = {'QualificationTypeId': qualification_type_id}
+
+ if description is not None:
+ params['Description'] = description
+
+ if status is not None:
+ params['QualificationTypeStatus'] = status
+
+ if retry_delay is not None:
+ params['RetryDelayInSeconds'] = retry_delay
+
+ if test is not None:
+ assert(isinstance(test, QuestionForm))
+ params['Test'] = test.get_as_xml()
+
+ if test_duration is not None:
+ params['TestDurationInSeconds'] = test_duration
+
+ if answer_key is not None:
+ if isinstance(answer_key, basestring):
+ params['AnswerKey'] = answer_key # xml
+ else:
+ raise TypeError
+ # Eventually someone will write an AnswerKey class.
+
+ if auto_granted is not None:
+ params['AutoGranted'] = auto_granted
+
+ if auto_granted_value is not None:
+ params['AutoGrantedValue'] = auto_granted_value
+
+ return self._process_request('UpdateQualificationType', params,
+ [('QualificationType', QualificationType)])
+
+ def dispose_qualification_type(self, qualification_type_id):
+ """TODO: Document."""
+ params = {'QualificationTypeId': qualification_type_id}
+ return self._process_request('DisposeQualificationType', params)
+
+ def search_qualification_types(self, query=None, sort_by='Name',
+ sort_direction='Ascending', page_size=10,
+ page_number=1, must_be_requestable=True,
+ must_be_owned_by_caller=True):
+ """TODO: Document."""
+ params = {'Query': query,
+ 'SortProperty': sort_by,
+ 'SortDirection': sort_direction,
+ 'PageSize': page_size,
+ 'PageNumber': page_number,
+ 'MustBeRequestable': must_be_requestable,
+ 'MustBeOwnedByCaller': must_be_owned_by_caller}
+ return self._process_request('SearchQualificationTypes', params,
+ [('QualificationType', QualificationType)])
+
+ def get_qualification_requests(self, qualification_type_id,
+ sort_by='Expiration',
+ sort_direction='Ascending', page_size=10,
+ page_number=1):
+ """TODO: Document."""
+ params = {'QualificationTypeId': qualification_type_id,
+ 'SortProperty': sort_by,
+ 'SortDirection': sort_direction,
+ 'PageSize': page_size,
+ 'PageNumber': page_number}
+ return self._process_request('GetQualificationRequests', params,
+ [('QualificationRequest', QualificationRequest)])
+
+ def grant_qualification(self, qualification_request_id, integer_value=1):
+ """TODO: Document."""
+ params = {'QualificationRequestId': qualification_request_id,
+ 'IntegerValue': integer_value}
+ return self._process_request('GrantQualification', params)
+
+ def revoke_qualification(self, subject_id, qualification_type_id,
+ reason=None):
+ """TODO: Document."""
+ params = {'SubjectId': subject_id,
+ 'QualificationTypeId': qualification_type_id,
+ 'Reason': reason}
+ return self._process_request('RevokeQualification', params)
+
+ def assign_qualification(self, qualification_type_id, worker_id,
+ value=1, send_notification=True):
+ params = {'QualificationTypeId': qualification_type_id,
+ 'WorkerId' : worker_id,
+ 'IntegerValue' : value,
+ 'SendNotification' : send_notification}
+ return self._process_request('AssignQualification', params)
+
+ def get_qualification_score(self, qualification_type_id, worker_id):
+ """TODO: Document."""
+ params = {'QualificationTypeId' : qualification_type_id,
+ 'SubjectId' : worker_id}
+ return self._process_request('GetQualificationScore', params,
+ [('Qualification', Qualification)])
+
+ def update_qualification_score(self, qualification_type_id, worker_id,
+ value):
+ """TODO: Document."""
+ params = {'QualificationTypeId' : qualification_type_id,
+ 'SubjectId' : worker_id,
+ 'IntegerValue' : value}
+ return self._process_request('UpdateQualificationScore', params)
+
+ def _process_request(self, request_type, params, marker_elems=None):
+ """
+ Helper to process the xml response from AWS
+ """
+ params['Operation'] = request_type
+ response = self.make_request(None, params, verb='POST')
+ return self._process_response(response, marker_elems)
+
+ def _process_response(self, response, marker_elems=None):
+ """
+ Helper to process the xml response from AWS
+ """
+ body = response.read()
+ if self.debug == 2:
+ print(body)
+ if '<Errors>' not in body:
+ rs = ResultSet(marker_elems)
+ h = handler.XmlHandler(rs, self)
+ xml.sax.parseString(body, h)
+ return rs
+ else:
+ raise MTurkRequestError(response.status, response.reason, body)
+
+ @staticmethod
+ def get_keywords_as_string(keywords):
+ """
+ Returns a comma+space-separated string of keywords from either
+ a list or a string
+ """
+ if isinstance(keywords, list):
+ keywords = ', '.join(keywords)
+ if isinstance(keywords, str):
+ final_keywords = keywords
+ elif isinstance(keywords, unicode):
+ final_keywords = keywords.encode('utf-8')
+ elif keywords is None:
+ final_keywords = ""
+ else:
+ raise TypeError("keywords argument must be a string or a list of strings; got a %s" % type(keywords))
+ return final_keywords
+
+ @staticmethod
+ def get_price_as_price(reward):
+ """
+ Returns a Price data structure from either a float or a Price
+ """
+ if isinstance(reward, Price):
+ final_price = reward
+ else:
+ final_price = Price(reward)
+ return final_price
+
+ @staticmethod
+ def duration_as_seconds(duration):
+ if isinstance(duration, datetime.timedelta):
+ duration = duration.days * 86400 + duration.seconds
+ try:
+ duration = int(duration)
+ except TypeError:
+ raise TypeError("Duration must be a timedelta or int-castable, got %s" % type(duration))
+ return duration
+
+
+class BaseAutoResultElement(object):
+ """
+ Base class to automatically add attributes when parsing XML
+ """
+ def __init__(self, connection):
+ pass
+
+ def startElement(self, name, attrs, connection):
+ return None
+
+ def endElement(self, name, value, connection):
+ setattr(self, name, value)
+
+
+class HIT(BaseAutoResultElement):
+ """
+ Class to extract a HIT structure from a response (used in ResultSet)
+
+ Will have attributes named as per the Developer Guide,
+ e.g. HITId, HITTypeId, CreationTime
+ """
+
+ # property helper to determine if HIT has expired
+ def _has_expired(self):
+ """ Has this HIT expired yet? """
+ expired = False
+ if hasattr(self, 'Expiration'):
+ now = datetime.datetime.utcnow()
+ expiration = datetime.datetime.strptime(self.Expiration, '%Y-%m-%dT%H:%M:%SZ')
+ expired = (now >= expiration)
+ else:
+ raise ValueError("ERROR: Request for expired property, but no Expiration in HIT!")
+ return expired
+
+ # are we there yet?
+ expired = property(_has_expired)
+
+
+class FileUploadURL(BaseAutoResultElement):
+ """
+ Class to extract an FileUploadURL structure from a response
+ """
+
+ pass
+
+
+class HITTypeId(BaseAutoResultElement):
+ """
+ Class to extract an HITTypeId structure from a response
+ """
+
+ pass
+
+
+class Qualification(BaseAutoResultElement):
+ """
+ Class to extract an Qualification structure from a response (used in
+ ResultSet)
+
+ Will have attributes named as per the Developer Guide such as
+ QualificationTypeId, IntegerValue. Does not seem to contain GrantTime.
+ """
+
+ pass
+
+
+class QualificationType(BaseAutoResultElement):
+ """
+ Class to extract an QualificationType structure from a response (used in
+ ResultSet)
+
+ Will have attributes named as per the Developer Guide,
+ e.g. QualificationTypeId, CreationTime, Name, etc
+ """
+
+ pass
+
+
+class QualificationRequest(BaseAutoResultElement):
+ """
+ Class to extract an QualificationRequest structure from a response (used in
+ ResultSet)
+
+ Will have attributes named as per the Developer Guide,
+ e.g. QualificationRequestId, QualificationTypeId, SubjectId, etc
+ """
+
+ def __init__(self, connection):
+ super(QualificationRequest, self).__init__(connection)
+ self.answers = []
+
+ def endElement(self, name, value, connection):
+ # the answer consists of embedded XML, so it needs to be parsed independantly
+ if name == 'Answer':
+ answer_rs = ResultSet([('Answer', QuestionFormAnswer)])
+ h = handler.XmlHandler(answer_rs, connection)
+ value = connection.get_utf8_value(value)
+ xml.sax.parseString(value, h)
+ self.answers.append(answer_rs)
+ else:
+ super(QualificationRequest, self).endElement(name, value, connection)
+
+
+class Assignment(BaseAutoResultElement):
+ """
+ Class to extract an Assignment structure from a response (used in
+ ResultSet)
+
+ Will have attributes named as per the Developer Guide,
+ e.g. AssignmentId, WorkerId, HITId, Answer, etc
+ """
+
+ def __init__(self, connection):
+ super(Assignment, self).__init__(connection)
+ self.answers = []
+
+ def endElement(self, name, value, connection):
+ # the answer consists of embedded XML, so it needs to be parsed independantly
+ if name == 'Answer':
+ answer_rs = ResultSet([('Answer', QuestionFormAnswer)])
+ h = handler.XmlHandler(answer_rs, connection)
+ value = connection.get_utf8_value(value)
+ xml.sax.parseString(value, h)
+ self.answers.append(answer_rs)
+ else:
+ super(Assignment, self).endElement(name, value, connection)
+
+
+class QuestionFormAnswer(BaseAutoResultElement):
+ """
+ Class to extract Answers from inside the embedded XML
+ QuestionFormAnswers element inside the Answer element which is
+ part of the Assignment and QualificationRequest structures
+
+ A QuestionFormAnswers element contains an Answer element for each
+ question in the HIT or Qualification test for which the Worker
+ provided an answer. Each Answer contains a QuestionIdentifier
+ element whose value corresponds to the QuestionIdentifier of a
+ Question in the QuestionForm. See the QuestionForm data structure
+ for more information about questions and answer specifications.
+
+ If the question expects a free-text answer, the Answer element
+ contains a FreeText element. This element contains the Worker's
+ answer
+
+ *NOTE* - currently really only supports free-text and selection answers
+ """
+
+ def __init__(self, connection):
+ super(QuestionFormAnswer, self).__init__(connection)
+ self.fields = []
+ self.qid = None
+
+ def endElement(self, name, value, connection):
+ if name == 'QuestionIdentifier':
+ self.qid = value
+ elif name in ['FreeText', 'SelectionIdentifier', 'OtherSelectionText'] and self.qid:
+ self.fields.append(value)

Powered by Google App Engine
This is Rietveld 408576698