152 lines
6.0 KiB
Python
152 lines
6.0 KiB
Python
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
|
|
from urllib.parse import urlparse
|
|
import uuid
|
|
|
|
from charms.reactive import set_flag, clear_flag
|
|
from charms.reactive import Endpoint
|
|
from charms.reactive import when_not, when
|
|
|
|
|
|
class NovaComputeRequires(Endpoint):
|
|
|
|
@when('endpoint.{endpoint_name}.changed')
|
|
def data_changed(self):
|
|
"""Set flag to indicate to charm relation data has changed."""
|
|
set_flag(self.expand_name('{endpoint_name}.available'))
|
|
|
|
@when_not('endpoint.{endpoint_name}.joined')
|
|
def broken(self):
|
|
"""Remove flag to indicate to charm relation has gone.."""
|
|
clear_flag(self.expand_name('{endpoint_name}.available'))
|
|
|
|
@when('endpoint.{endpoint_name}.joined')
|
|
def joined(self):
|
|
"""Set flag to indicate to charm relation has been joined."""
|
|
set_flag(self.expand_name('{endpoint_name}.connected'))
|
|
|
|
def set_network_data(self, neutron_url, neutron_plugin=None,
|
|
network_manager=None, enable_security_groups=True):
|
|
"""Send compute nodes data relating to network setup.
|
|
|
|
:param neutron_url: URL for network API service
|
|
:type neutron_url: str
|
|
:param neutron_plugin: Neutron plugin to use
|
|
:type neutron_plugin: str
|
|
:param network_manager: Network Manager
|
|
:type network_manager: str
|
|
:param enable_security_groups: Whether to enable security groups
|
|
:type enable_security_group: bool
|
|
"""
|
|
o = urlparse(neutron_url)
|
|
if enable_security_groups:
|
|
security_groups = 'yes'
|
|
else:
|
|
security_groups = 'no'
|
|
for r in self.relations:
|
|
r.to_publish_raw['quantum_host'] = o.hostname
|
|
r.to_publish_raw['quantum_plugin'] = neutron_plugin or 'ovs'
|
|
r.to_publish_raw['quantum_port'] = o.port
|
|
r.to_publish_raw['quantum_security_groups'] = security_groups
|
|
r.to_publish_raw['quantum_url'] = neutron_url
|
|
r.to_publish_raw['network_manager'] = network_manager or 'neutron'
|
|
|
|
def set_console_data(self, serial_console_base_url, enable_serial_console):
|
|
"""Send compute nodes data relating to serial console access.
|
|
|
|
:param serial_console_base_url: URL for accessing the serial console.
|
|
:type serial_console_base_url: str
|
|
:param enable_serial_console: Whether to enable the serial console
|
|
:type enable_serial_console: bool
|
|
"""
|
|
for r in self.relations:
|
|
r.to_publish_raw[
|
|
'serial_console_base_url'] = serial_console_base_url
|
|
r.to_publish_raw['enable_serial_console'] = enable_serial_console
|
|
|
|
def trigger_remote_restart(self, restart_key=None):
|
|
"""Trigger a restart of services on the remote application.
|
|
|
|
:param restart_key: Key to send to remote service, restarts are
|
|
triggered when the key changes.
|
|
:type restart_key: str
|
|
"""
|
|
if not restart_key:
|
|
restart_key = uuid.uuid1()
|
|
for relation in self.relations:
|
|
relation.to_publish_raw['restart_trigger'] = restart_key
|
|
|
|
def set_region(self, region):
|
|
"""Send compute nodes region information.
|
|
|
|
:param region: Region compute nodes will belong to.
|
|
:type region: str
|
|
"""
|
|
for relation in self.relations:
|
|
relation.to_publish_raw['region'] = region
|
|
|
|
def set_volume_data(self, volume_service):
|
|
"""Send compute nodes volume information.
|
|
|
|
:param volume_service: Name of volume service to use, eg cinder
|
|
:type volume_service: str
|
|
"""
|
|
for relation in self.relations:
|
|
relation.to_publish_raw['volume_service'] = volume_service
|
|
|
|
def set_ec2_data(self, ec2_host):
|
|
"""Send compute nodes ec2 information.
|
|
|
|
:param ec2_host: Name of ec2_host.
|
|
:type ec2_host: str
|
|
"""
|
|
for relation in self.relations:
|
|
relation.to_publish_raw['ec2_host'] = ec2_host
|
|
|
|
def collect_ssh_keys(self, application_name=None):
|
|
"""Query related units and collect ssh artifacts.
|
|
|
|
:param application_name: Only return artifacts from units of this
|
|
applicationa.
|
|
:type application_name: str
|
|
:returns: {APP_NAME: {UNIT_NAME: {pupkey1:.., hostkey1:...}}}
|
|
:rtype: dict
|
|
"""
|
|
ssh_keys = {}
|
|
for rel in self.relations:
|
|
if application_name and application_name != rel.application_name:
|
|
continue
|
|
ssh_keys[rel.application_name] = {}
|
|
for unit in rel.units:
|
|
nova_ssh_pub_key = unit.received.get('nova_ssh_public_key')
|
|
ssh_pub_key = unit.received.get('ssh_public_key')
|
|
if nova_ssh_pub_key and ssh_pub_key:
|
|
ssh_keys[rel.application_name][unit.unit_name] = {
|
|
'nova_ssh_pub_key': nova_ssh_pub_key,
|
|
'hostname': unit.received.get('hostname'),
|
|
'private-address': unit.received.get(
|
|
'private-address'),
|
|
'ssh_pub_key': ssh_pub_key}
|
|
return ssh_keys
|
|
|
|
def send_ssh_keys(self, relation, settings):
|
|
"""Publish the provided ssh settings on the given relation
|
|
|
|
:param relation: Relation to publish settings on.
|
|
:type relation: charms.reactive.endpoints.Relation
|
|
:param settings: SSH settings to publish.
|
|
:type settings: dict
|
|
"""
|
|
for key, value in settings.items():
|
|
relation.to_publish_raw[key] = value
|