| Index: third_party/gsutil/boto/dynamodb/types.py
|
| diff --git a/third_party/gsutil/boto/dynamodb/types.py b/third_party/gsutil/boto/dynamodb/types.py
|
| new file mode 100644
|
| index 0000000000000000000000000000000000000000..58fb89cae5ece378c0869c0cb06588cb25e315d4
|
| --- /dev/null
|
| +++ b/third_party/gsutil/boto/dynamodb/types.py
|
| @@ -0,0 +1,161 @@
|
| +# Copyright (c) 2011 Mitch Garnaat http://garnaat.org/
|
| +# Copyright (c) 2011 Amazon.com, Inc. or its affiliates. All Rights Reserved
|
| +#
|
| +# 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.
|
| +#
|
| +"""
|
| +Some utility functions to deal with mapping Amazon DynamoDB types to
|
| +Python types and vice-versa.
|
| +"""
|
| +import base64
|
| +
|
| +
|
| +def is_num(n):
|
| + types = (int, long, float, bool)
|
| + return isinstance(n, types) or n in types
|
| +
|
| +
|
| +def is_str(n):
|
| + return isinstance(n, basestring) or (isinstance(n, type) and
|
| + issubclass(n, basestring))
|
| +
|
| +
|
| +def is_binary(n):
|
| + return isinstance(n, Binary)
|
| +
|
| +
|
| +def convert_num(s):
|
| + if '.' in s:
|
| + n = float(s)
|
| + else:
|
| + n = int(s)
|
| + return n
|
| +
|
| +
|
| +def convert_binary(n):
|
| + return Binary(base64.b64decode(n))
|
| +
|
| +
|
| +def get_dynamodb_type(val):
|
| + """
|
| + Take a scalar Python value and return a string representing
|
| + the corresponding Amazon DynamoDB type. If the value passed in is
|
| + not a supported type, raise a TypeError.
|
| + """
|
| + dynamodb_type = None
|
| + if is_num(val):
|
| + dynamodb_type = 'N'
|
| + elif is_str(val):
|
| + dynamodb_type = 'S'
|
| + elif isinstance(val, (set, frozenset)):
|
| + if False not in map(is_num, val):
|
| + dynamodb_type = 'NS'
|
| + elif False not in map(is_str, val):
|
| + dynamodb_type = 'SS'
|
| + elif False not in map(is_binary, val):
|
| + dynamodb_type = 'BS'
|
| + elif isinstance(val, Binary):
|
| + dynamodb_type = 'B'
|
| + if dynamodb_type is None:
|
| + msg = 'Unsupported type "%s" for value "%s"' % (type(val), val)
|
| + raise TypeError(msg)
|
| + return dynamodb_type
|
| +
|
| +
|
| +def dynamize_value(val):
|
| + """
|
| + Take a scalar Python value and return a dict consisting
|
| + of the Amazon DynamoDB type specification and the value that
|
| + needs to be sent to Amazon DynamoDB. If the type of the value
|
| + is not supported, raise a TypeError
|
| + """
|
| + def _str(val):
|
| + """
|
| + DynamoDB stores booleans as numbers. True is 1, False is 0.
|
| + This function converts Python booleans into DynamoDB friendly
|
| + representation.
|
| + """
|
| + if isinstance(val, bool):
|
| + return str(int(val))
|
| + return str(val)
|
| +
|
| + dynamodb_type = get_dynamodb_type(val)
|
| + if dynamodb_type == 'N':
|
| + val = {dynamodb_type: _str(val)}
|
| + elif dynamodb_type == 'S':
|
| + val = {dynamodb_type: val}
|
| + elif dynamodb_type == 'NS':
|
| + val = {dynamodb_type: [str(n) for n in val]}
|
| + elif dynamodb_type == 'SS':
|
| + val = {dynamodb_type: [n for n in val]}
|
| + elif dynamodb_type == 'B':
|
| + val = {dynamodb_type: val.encode()}
|
| + elif dynamodb_type == 'BS':
|
| + val = {dynamodb_type: [n.encode() for n in val]}
|
| + return val
|
| +
|
| +
|
| +class Binary(object):
|
| + def __init__(self, value):
|
| + self.value = value
|
| +
|
| + def encode(self):
|
| + return base64.b64encode(self.value)
|
| +
|
| + def __eq__(self, other):
|
| + if isinstance(other, Binary):
|
| + return self.value == other.value
|
| + else:
|
| + return self.value == other
|
| +
|
| + def __ne__(self, other):
|
| + return not self.__eq__(other)
|
| +
|
| + def __repr__(self):
|
| + return 'Binary(%s)' % self.value
|
| +
|
| + def __str__(self):
|
| + return self.value
|
| +
|
| + def __hash__(self):
|
| + return hash(self.value)
|
| +
|
| +
|
| +def item_object_hook(dct):
|
| + """
|
| + A custom object hook for use when decoding JSON item bodys.
|
| + This hook will transform Amazon DynamoDB JSON responses to something
|
| + that maps directly to native Python types.
|
| + """
|
| + if len(dct.keys()) > 1:
|
| + return dct
|
| + if 'S' in dct:
|
| + return dct['S']
|
| + if 'N' in dct:
|
| + return convert_num(dct['N'])
|
| + if 'SS' in dct:
|
| + return set(dct['SS'])
|
| + if 'NS' in dct:
|
| + return set(map(convert_num, dct['NS']))
|
| + if 'B' in dct:
|
| + return convert_binary(dct['B'])
|
| + if 'BS' in dct:
|
| + return set(map(convert_binary, dct['BS']))
|
| + return dct
|
|
|