| Index: tools/telemetry/third_party/gsutilz/third_party/boto/boto/contrib/ymlmessage.py
|
| diff --git a/tools/telemetry/third_party/gsutilz/third_party/gcs-oauth2-boto-plugin/gcs_oauth2_boto_plugin/__init__.py b/tools/telemetry/third_party/gsutilz/third_party/boto/boto/contrib/ymlmessage.py
|
| similarity index 55%
|
| copy from tools/telemetry/third_party/gsutilz/third_party/gcs-oauth2-boto-plugin/gcs_oauth2_boto_plugin/__init__.py
|
| copy to tools/telemetry/third_party/gsutilz/third_party/boto/boto/contrib/ymlmessage.py
|
| index a70e6515e01f13e404e77ef814accaa5ba5c157e..ae6aea488e5114111343c75f6c6ead7dd5beda37 100644
|
| --- a/tools/telemetry/third_party/gsutilz/third_party/gcs-oauth2-boto-plugin/gcs_oauth2_boto_plugin/__init__.py
|
| +++ b/tools/telemetry/third_party/gsutilz/third_party/boto/boto/contrib/ymlmessage.py
|
| @@ -1,4 +1,4 @@
|
| -# Copyright 2014 Google Inc. All Rights Reserved.
|
| +# Copyright (c) 2006,2007 Chris Moyer
|
| #
|
| # Permission is hereby granted, free of charge, to any person obtaining a
|
| # copy of this software and associated documentation files (the
|
| @@ -19,12 +19,35 @@
|
| # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
|
| # IN THE SOFTWARE.
|
|
|
| -from __future__ import absolute_import
|
| +"""
|
| +This module was contributed by Chris Moyer. It provides a subclass of the
|
| +SQS Message class that supports YAML as the body of the message.
|
|
|
| -# Import this module so that users can simply import the top-level module. Once
|
| -# imported, boto will pick up our subclasses of AuthHandler.
|
| -from gcs_oauth2_boto_plugin import oauth2_plugin
|
| +This module requires the yaml module.
|
| +"""
|
| +from boto.sqs.message import Message
|
| +import yaml
|
|
|
| -from gcs_oauth2_boto_plugin.oauth2_helper import SetFallbackClientIdAndSecret
|
| -from gcs_oauth2_boto_plugin.oauth2_helper import SetLock
|
|
|
| +class YAMLMessage(Message):
|
| + """
|
| + The YAMLMessage class provides a YAML compatible message. Encoding and
|
| + decoding are handled automaticaly.
|
| +
|
| + Access this message data like such:
|
| +
|
| + m.data = [ 1, 2, 3]
|
| + m.data[0] # Returns 1
|
| +
|
| + This depends on the PyYAML package
|
| + """
|
| +
|
| + def __init__(self, queue=None, body='', xml_attrs=None):
|
| + self.data = None
|
| + super(YAMLMessage, self).__init__(queue, body)
|
| +
|
| + def set_body(self, body):
|
| + self.data = yaml.load(body)
|
| +
|
| + def get_body(self):
|
| + return yaml.dump(self.data)
|
|
|