Charmed-Kubernetes/kubernetes-control-plane/hooks/relations/vsphere-integration/requires.py

142 lines
4.5 KiB
Python

"""
This is the requires side of the interface layer, for use in charms that wish
to request integration with vSphere native features. The integration will be
provided by the vSphere integration charm, which allows the requiring charm
to not require cloud credentials itself and not have a lot of vSphere
specific API code.
The flags that are set by the requires side of this interface are:
* **`endpoint.{endpoint_name}.joined`** This flag is set when the relation
has been joined, and the charm should then use the methods documented below
to request specific vSphere features. This flag is automatically removed
if the relation is broken. It should not be removed by the charm.
* **`endpoint.{endpoint_name}.ready`** This flag is set once the requested
features have been enabled for the vSphere instance on which the charm is
running. This flag is automatically removed if new integration features are
requested. It should not be removed by the charm.
* **`endpoint.{endpoint_name}.ready.changed`** This flag is set if the data
changes after the ready flag was set. This flag should be removed by the
charm once handled.
"""
from charms.reactive import Endpoint
from charms.reactive import when, when_not
from charms.reactive import clear_flag, is_flag_set, set_flag, toggle_flag
from charms.reactive import data_changed
class VsphereIntegrationRequires(Endpoint):
"""
Interface to request integration access.
Note that due to resource limits and permissions granularity, policies are
limited to being applied at the charm level. That means that, if any
permissions are requested (i.e., any of the enable methods are called),
what is granted will be the sum of those ever requested by any instance of
the charm on this cloud.
Labels, on the other hand, will be instance specific.
Example usage:
```python
from charms.reactive import when, endpoint_from_flag
@when('endpoint.vsphere.ready')
def vsphere_integration_ready():
vsphere = endpoint_from_flag('endpoint.vsphere.joined')
update_config_enable_vsphere(vsphere.vsphere_ip,
vsphere.user,
vsphere.password,
vsphere.datacenter,
vsphere.datastore,
vsphere.folder,
vsphere.respool_path)
```
"""
@property
def _received(self):
"""
Helper to streamline access to received data.
"""
return self.all_joined_units.received
@when('endpoint.{endpoint_name}.changed')
def check_ready(self):
"""
Manage flags to signal when the endpoint is ready as well as noting
if changes have been made since it became ready.
"""
was_ready = is_flag_set(self.expand_name('ready'))
toggle_flag(self.expand_name('ready'), self.is_ready)
if self.is_ready and was_ready and self.is_changed:
set_flag(self.expand_name('ready.changed'))
clear_flag(self.expand_name('changed'))
@when_not('endpoint.{endpoint_name}.joined')
def remove_ready(self):
clear_flag(self.expand_name('ready'))
@property
def is_ready(self):
"""
Whether or not the request for this instance has been completed.
"""
return all(field is not None for field in [
self.vsphere_ip,
self.user,
self.password,
self.datacenter,
self.datastore,
self.folder,
self.respool_path,
])
@property
def is_changed(self):
"""
Whether or not the request for this instance has changed.
"""
return data_changed(self.expand_name('all-data'), [
self.vsphere_ip,
self.user,
self.password,
self.datacenter,
self.datastore,
self.folder,
self.respool_path,
])
@property
def vsphere_ip(self):
return self._received['vsphere_ip']
@property
def user(self):
return self._received['user']
@property
def password(self):
return self._received['password']
@property
def datacenter(self):
return self._received['datacenter']
@property
def datastore(self):
return self._received['datastore']
@property
def folder(self):
return self._received['folder']
@property
def respool_path(self):
return self._received['respool_path']