← Back to branch summary

~soren/nova/iptables-security-groups

« back to all changes in this revision

Viewing changes to vendor/boto/boto/sdb/persist/__init__.py

  • Committer: Jesse Andrews
  • Date: 2010-05-28 06:05:26 UTC
  • Revision ID: git-v1:bf6e6e718cdc7488e2da87b21e258ccc065fe499
initial commit

Show diffs side-by-side

added added

removed removed

Lines of Context:
vendor/boto/boto/sdb/persist/__init__.py
 
1
# Copyright (c) 2006,2007,2008 Mitch Garnaat http://garnaat.org/
 
2
#
 
3
# Permission is hereby granted, free of charge, to any person obtaining a
 
4
# copy of this software and associated documentation files (the
 
5
# "Software"), to deal in the Software without restriction, including
 
6
# without limitation the rights to use, copy, modify, merge, publish, dis-
 
7
# tribute, sublicense, and/or sell copies of the Software, and to permit
 
8
# persons to whom the Software is furnished to do so, subject to the fol-
 
9
# lowing conditions:
 
10
#
 
11
# The above copyright notice and this permission notice shall be included
 
12
# in all copies or substantial portions of the Software.
 
13
#
 
14
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
 
15
# OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABIL-
 
16
# ITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT
 
17
# SHALL THE AUTHOR BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, 
 
18
# WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 
19
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
 
20
# IN THE SOFTWARE.
 
21
 
 
22
import boto
 
23
from boto.utils import find_class
 
24
 
 
25
class Manager(object):
 
26
 
 
27
    DefaultDomainName = boto.config.get('Persist', 'default_domain', None)
 
28
 
 
29
    def __init__(self, domain_name=None, aws_access_key_id=None, aws_secret_access_key=None, debug=0):
 
30
        self.domain_name = domain_name
 
31
        self.aws_access_key_id = aws_access_key_id
 
32
        self.aws_secret_access_key = aws_secret_access_key
 
33
        self.domain = None
 
34
        self.sdb = None
 
35
        self.s3 = None
 
36
        if not self.domain_name:
 
37
            self.domain_name = self.DefaultDomainName
 
38
            if self.domain_name:
 
39
                boto.log.info('No SimpleDB domain set, using default_domain: %s' % self.domain_name)
 
40
            else:
 
41
                boto.log.warning('No SimpleDB domain set, persistance is disabled')
 
42
        if self.domain_name:
 
43
            self.sdb = boto.connect_sdb(aws_access_key_id=self.aws_access_key_id,
 
44
                                        aws_secret_access_key=self.aws_secret_access_key,
 
45
                                        debug=debug)
 
46
            self.domain = self.sdb.lookup(self.domain_name)
 
47
            if not self.domain:
 
48
                self.domain = self.sdb.create_domain(self.domain_name)
 
49
 
 
50
    def get_s3_connection(self):
 
51
        if not self.s3:
 
52
            self.s3 = boto.connect_s3(self.aws_access_key_id, self.aws_secret_access_key)
 
53
        return self.s3
 
54
 
 
55
def get_manager(domain_name=None, aws_access_key_id=None, aws_secret_access_key=None, debug=0):
 
56
    return Manager(domain_name, aws_access_key_id, aws_secret_access_key, debug=debug)
 
57
 
 
58
def set_domain(domain_name):
 
59
    Manager.DefaultDomainName = domain_name
 
60
 
 
61
def get_domain():
 
62
    return Manager.DefaultDomainName
 
63
 
 
64
def revive_object_from_id(id, manager):
 
65
    if not manager.domain:
 
66
        return None
 
67
    attrs = manager.domain.get_attributes(id, ['__module__', '__type__', '__lineage__'])
 
68
    try:
 
69
        cls = find_class(attrs['__module__'], attrs['__type__'])
 
70
        return cls(id, manager=manager)
 
71
    except ImportError:
 
72
        return None
 
73
 
 
74
def object_lister(cls, query_lister, manager):
 
75
    for item in query_lister:
 
76
        if cls:
 
77
            yield cls(item.name)
 
78
        else:
 
79
            o = revive_object_from_id(item.name, manager)
 
80
            if o:
 
81
                yield o
 
82
                
 
83