Browse Source

Initial commit

changes/13/749213/1
Liam Young 2 years ago
commit
518b4fbdc4
  1. 79
      ops_openstack.py

79
ops_openstack.py

@ -0,0 +1,79 @@
from ops.charm import CharmBase
from ops.framework import (
StoredState,
)
from charmhelpers.fetch import (
apt_install,
apt_update,
add_source,
)
from ops.model import (
ActiveStatus,
MaintenanceStatus,
WaitingStatus,
)
import charmhelpers.contrib.openstack.utils as os_utils
import logging
logger = logging.getLogger()
class OSBaseCharm(CharmBase):
state = StoredState()
PACKAGES = []
RESTART_MAP = {}
def __init__(self, framework, key):
super().__init__(framework, key)
self.state.set_default(is_started=False)
self.state.set_default(is_paused=False)
self.framework.observe(self.on.install, self)
self.framework.observe(self.on.update_status, self)
self.framework.observe(self.on.pause_action, self)
self.framework.observe(self.on.resume_action, self)
def on_install(self, event):
logging.info("Installing packages")
if self.framework.model.config.get('source'):
add_source(
self.framework.model.config['source'],
self.framework.model.config.get('key'))
apt_update(fatal=True)
apt_install(self.PACKAGES, fatal=True)
def update_status(self):
logging.info("Updating status")
if self.state.is_paused:
self.model.unit.status = MaintenanceStatus(
"Paused. Use 'resume' action to resume normal service.")
if self.state.is_started:
self.model.unit.status = ActiveStatus('Unit is ready')
else:
self.model.unit.status = WaitingStatus('Not ready for reasons')
logging.info("Status updated")
def on_update_status(self, event):
self.update_status()
def services(self):
_svcs = []
for svc in self.RESTART_MAP.values():
_svcs.extend(svc)
return list(set(_svcs))
def on_pause_action(self, event):
_, messages = os_utils.manage_payload_services(
'pause',
services=self.services(),
charm_func=None)
self.state.is_paused = True
self.update_status()
def on_resume_action(self, event):
_, messages = os_utils.manage_payload_services(
'resume',
services=self.services(),
charm_func=None)
self.state.is_paused = False
self.update_status()
Loading…
Cancel
Save