~junaidali/charms/trusty/plumgrid-director/pg-restart

« back to all changes in this revision

Viewing changes to hooks/charmhelpers-pl/contrib/openstack/amulet/deployment.py

  • Committer: bbaqar at plumgrid
  • Date: 2015-05-19 21:05:20 UTC
  • Revision ID: bbaqar@plumgrid.com-20150519210520-8nymqswwgdu7qygg
PLUMgrid director initial charm

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
# Copyright 2014-2015 Canonical Limited.
 
2
#
 
3
# This file is part of charm-helpers.
 
4
#
 
5
# charm-helpers is free software: you can redistribute it and/or modify
 
6
# it under the terms of the GNU Lesser General Public License version 3 as
 
7
# published by the Free Software Foundation.
 
8
#
 
9
# charm-helpers is distributed in the hope that it will be useful,
 
10
# but WITHOUT ANY WARRANTY; without even the implied warranty of
 
11
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
12
# GNU Lesser General Public License for more details.
 
13
#
 
14
# You should have received a copy of the GNU Lesser General Public License
 
15
# along with charm-helpers.  If not, see <http://www.gnu.org/licenses/>.
 
16
 
 
17
import six
 
18
from collections import OrderedDict
 
19
from charmhelpers.contrib.amulet.deployment import (
 
20
    AmuletDeployment
 
21
)
 
22
 
 
23
 
 
24
class OpenStackAmuletDeployment(AmuletDeployment):
 
25
    """OpenStack amulet deployment.
 
26
 
 
27
       This class inherits from AmuletDeployment and has additional support
 
28
       that is specifically for use by OpenStack charms.
 
29
       """
 
30
 
 
31
    def __init__(self, series=None, openstack=None, source=None, stable=True):
 
32
        """Initialize the deployment environment."""
 
33
        super(OpenStackAmuletDeployment, self).__init__(series)
 
34
        self.openstack = openstack
 
35
        self.source = source
 
36
        self.stable = stable
 
37
        # Note(coreycb): this needs to be changed when new next branches come
 
38
        # out.
 
39
        self.current_next = "trusty"
 
40
 
 
41
    def _determine_branch_locations(self, other_services):
 
42
        """Determine the branch locations for the other services.
 
43
 
 
44
           Determine if the local branch being tested is derived from its
 
45
           stable or next (dev) branch, and based on this, use the corresonding
 
46
           stable or next branches for the other_services."""
 
47
        base_charms = ['mysql', 'mongodb']
 
48
 
 
49
        if self.series in ['precise', 'trusty']:
 
50
            base_series = self.series
 
51
        else:
 
52
            base_series = self.current_next
 
53
 
 
54
        if self.stable:
 
55
            for svc in other_services:
 
56
                temp = 'lp:charms/{}/{}'
 
57
                svc['location'] = temp.format(base_series,
 
58
                                              svc['name'])
 
59
        else:
 
60
            for svc in other_services:
 
61
                if svc['name'] in base_charms:
 
62
                    temp = 'lp:charms/{}/{}'
 
63
                    svc['location'] = temp.format(base_series,
 
64
                                                  svc['name'])
 
65
                else:
 
66
                    temp = 'lp:~openstack-charmers/charms/{}/{}/next'
 
67
                    svc['location'] = temp.format(self.current_next,
 
68
                                                  svc['name'])
 
69
        return other_services
 
70
 
 
71
    def _add_services(self, this_service, other_services):
 
72
        """Add services to the deployment and set openstack-origin/source."""
 
73
        other_services = self._determine_branch_locations(other_services)
 
74
 
 
75
        super(OpenStackAmuletDeployment, self)._add_services(this_service,
 
76
                                                             other_services)
 
77
 
 
78
        services = other_services
 
79
        services.append(this_service)
 
80
        use_source = ['mysql', 'mongodb', 'rabbitmq-server', 'ceph',
 
81
                      'ceph-osd', 'ceph-radosgw']
 
82
        # Openstack subordinate charms do not expose an origin option as that
 
83
        # is controlled by the principle
 
84
        ignore = ['neutron-openvswitch']
 
85
 
 
86
        if self.openstack:
 
87
            for svc in services:
 
88
                if svc['name'] not in use_source + ignore:
 
89
                    config = {'openstack-origin': self.openstack}
 
90
                    self.d.configure(svc['name'], config)
 
91
 
 
92
        if self.source:
 
93
            for svc in services:
 
94
                if svc['name'] in use_source and svc['name'] not in ignore:
 
95
                    config = {'source': self.source}
 
96
                    self.d.configure(svc['name'], config)
 
97
 
 
98
    def _configure_services(self, configs):
 
99
        """Configure all of the services."""
 
100
        for service, config in six.iteritems(configs):
 
101
            self.d.configure(service, config)
 
102
 
 
103
    def _get_openstack_release(self):
 
104
        """Get openstack release.
 
105
 
 
106
           Return an integer representing the enum value of the openstack
 
107
           release.
 
108
           """
 
109
        # Must be ordered by OpenStack release (not by Ubuntu release):
 
110
        (self.precise_essex, self.precise_folsom, self.precise_grizzly,
 
111
         self.precise_havana, self.precise_icehouse,
 
112
         self.trusty_icehouse, self.trusty_juno, self.utopic_juno,
 
113
         self.trusty_kilo, self.vivid_kilo) = range(10)
 
114
 
 
115
        releases = {
 
116
            ('precise', None): self.precise_essex,
 
117
            ('precise', 'cloud:precise-folsom'): self.precise_folsom,
 
118
            ('precise', 'cloud:precise-grizzly'): self.precise_grizzly,
 
119
            ('precise', 'cloud:precise-havana'): self.precise_havana,
 
120
            ('precise', 'cloud:precise-icehouse'): self.precise_icehouse,
 
121
            ('trusty', None): self.trusty_icehouse,
 
122
            ('trusty', 'cloud:trusty-juno'): self.trusty_juno,
 
123
            ('trusty', 'cloud:trusty-kilo'): self.trusty_kilo,
 
124
            ('utopic', None): self.utopic_juno,
 
125
            ('vivid', None): self.vivid_kilo}
 
126
        return releases[(self.series, self.openstack)]
 
127
 
 
128
    def _get_openstack_release_string(self):
 
129
        """Get openstack release string.
 
130
 
 
131
           Return a string representing the openstack release.
 
132
           """
 
133
        releases = OrderedDict([
 
134
            ('precise', 'essex'),
 
135
            ('quantal', 'folsom'),
 
136
            ('raring', 'grizzly'),
 
137
            ('saucy', 'havana'),
 
138
            ('trusty', 'icehouse'),
 
139
            ('utopic', 'juno'),
 
140
            ('vivid', 'kilo'),
 
141
        ])
 
142
        if self.openstack:
 
143
            os_origin = self.openstack.split(':')[1]
 
144
            return os_origin.split('%s-' % self.series)[1].split('/')[0]
 
145
        else:
 
146
            return releases[self.series]