Index: third_party/gsutil/boto/boto/services/servicedef.py |
diff --git a/third_party/gsutil/boto/boto/services/servicedef.py b/third_party/gsutil/boto/boto/services/servicedef.py |
new file mode 100644 |
index 0000000000000000000000000000000000000000..1cb01aa75480e93fdae3ffd5d7a050cdd8b2e7e2 |
--- /dev/null |
+++ b/third_party/gsutil/boto/boto/services/servicedef.py |
@@ -0,0 +1,91 @@ |
+# 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. |
+ |
+from boto.pyami.config import Config |
+from boto.services.message import ServiceMessage |
+import boto |
+ |
+class ServiceDef(Config): |
+ |
+ def __init__(self, config_file, aws_access_key_id=None, aws_secret_access_key=None): |
+ Config.__init__(self, config_file) |
+ self.aws_access_key_id = aws_access_key_id |
+ self.aws_secret_access_key = aws_secret_access_key |
+ script = Config.get(self, 'Pyami', 'scripts') |
+ if script: |
+ self.name = script.split('.')[-1] |
+ else: |
+ self.name = None |
+ |
+ |
+ def get(self, name, default=None): |
+ return Config.get(self, self.name, name, default) |
+ |
+ def has_option(self, option): |
+ return Config.has_option(self, self.name, option) |
+ |
+ def getint(self, option, default=0): |
+ try: |
+ val = Config.get(self, self.name, option) |
+ val = int(val) |
+ except: |
+ val = int(default) |
+ return val |
+ |
+ def getbool(self, option, default=False): |
+ try: |
+ val = Config.get(self, self.name, option) |
+ if val.lower() == 'true': |
+ val = True |
+ else: |
+ val = False |
+ except: |
+ val = default |
+ return val |
+ |
+ def get_obj(self, name): |
+ """ |
+ Returns the AWS object associated with a given option. |
+ |
+ The heuristics used are a bit lame. If the option name contains |
+ the word 'bucket' it is assumed to be an S3 bucket, if the name |
+ contains the word 'queue' it is assumed to be an SQS queue and |
+ if it contains the word 'domain' it is assumed to be a SimpleDB |
+ domain. If the option name specified does not exist in the |
+ config file or if the AWS object cannot be retrieved this |
+ returns None. |
+ """ |
+ val = self.get(name) |
+ if not val: |
+ return None |
+ if name.find('queue') >= 0: |
+ obj = boto.lookup('sqs', val) |
+ if obj: |
+ obj.set_message_class(ServiceMessage) |
+ elif name.find('bucket') >= 0: |
+ obj = boto.lookup('s3', val) |
+ elif name.find('domain') >= 0: |
+ obj = boto.lookup('sdb', val) |
+ else: |
+ obj = None |
+ return obj |
+ |
+ |