catalin/moto
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Added basic support for SQS MessageAttributes.

Browse source
This commit is contained in:
Ralfas 2014-09-28 20:59:14 +01:00
commit 76aa9a8b22
5 changed files with 134 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

38
moto/sqs/utils.py
View file

@ -3,6 +3,8 @@ import datetime
import random
import string
from .exceptions import MessageAttributesInvalid
def generate_receipt_handle():
# http://docs.aws.amazon.com/AWSSimpleQueueService/latest/SQSDeveloperGuide/ImportantIdentifiers.html#ImportantIdentifiers-receipt-handles
@ -19,3 +21,39 @@ def unix_time(dt=None):
def unix_time_millis(dt=None):
return unix_time(dt) * 1000.0
def parse_message_attributes(querystring, base='', value_namespace='Value.'):
message_attributes = {}
index = 1
while True:
# Loop through looking for message attributes
name_key = base + 'MessageAttribute.{0}.Name'.format(index)
name = querystring.get(name_key)
if not name:
# Found all attributes
break
data_type_key = base + 'MessageAttribute.{0}.{1}DataType'.format(index, value_namespace)
data_type = querystring.get(data_type_key)
if not data_type:
raise MessageAttributesInvalid("The message attribute '{0}' must contain non-empty message attribute value.".format(name[0]))
data_type_parts = data_type[0].split('.')
if len(data_type_parts) > 2 or data_type_parts[0] not in ['String', 'Binary', 'Number']:
raise MessageAttributesInvalid("The message attribute '{0}' has an invalid message attribute type, the set of supported type prefixes is Binary, Number, and String.".format(name[0]))
type_prefix = 'String'
if data_type_parts[0] == 'Binary':
type_prefix = 'Binary'
value_key = base + 'MessageAttribute.{0}.{1}{2}Value'.format(index, value_namespace, type_prefix)
value = querystring.get(value_key)
if not value:
raise MessageAttributesInvalid("The message attribute '{0}' must contain non-empty message attribute value for message attribute type '{1}'.".format(name[0], data_type[0]))
message_attributes[name[0]] = {'data_type' : data_type[0], type_prefix.lower() + '_value' : value[0]}
index += 1
return message_attributes