184 lines
3.8 KiB
Python
184 lines
3.8 KiB
Python
|
|
#
|
|
# Copyright 2012 Canonical Ltd.
|
|
#
|
|
# Authors:
|
|
# James Page <james.page@ubuntu.com>
|
|
# Paul Collins <paul.collins@canonical.com>
|
|
#
|
|
|
|
import os
|
|
import subprocess
|
|
import socket
|
|
import sys
|
|
|
|
|
|
def do_hooks(hooks):
|
|
hook = os.path.basename(sys.argv[0])
|
|
|
|
try:
|
|
hooks[hook]()
|
|
except KeyError:
|
|
juju_log('INFO',
|
|
"This charm doesn't know how to handle '{}'.".format(hook))
|
|
|
|
|
|
def install(*pkgs):
|
|
cmd = [
|
|
'apt-get',
|
|
'-y',
|
|
'install'
|
|
]
|
|
for pkg in pkgs:
|
|
cmd.append(pkg)
|
|
subprocess.check_call(cmd)
|
|
|
|
TEMPLATES_DIR = 'templates'
|
|
|
|
try:
|
|
import jinja2
|
|
except ImportError:
|
|
install('python-jinja2')
|
|
import jinja2
|
|
|
|
|
|
def render_template(template_name, context, template_dir=TEMPLATES_DIR):
|
|
templates = jinja2.Environment(
|
|
loader=jinja2.FileSystemLoader(template_dir)
|
|
)
|
|
template = templates.get_template(template_name)
|
|
return template.render(context)
|
|
|
|
|
|
CLOUD_ARCHIVE = \
|
|
""" # Ubuntu Cloud Archive
|
|
deb http://ubuntu-cloud.archive.canonical.com/ubuntu {} main
|
|
"""
|
|
|
|
|
|
def configure_source():
|
|
source = str(config_get('source'))
|
|
if not source:
|
|
return
|
|
if source.startswith('ppa:'):
|
|
cmd = [
|
|
'add-apt-repository',
|
|
source
|
|
]
|
|
subprocess.check_call(cmd)
|
|
if source.startswith('cloud:'):
|
|
install('ubuntu-cloud-keyring')
|
|
pocket = source.split(':')[1]
|
|
with open('/etc/apt/sources.list.d/cloud-archive.list', 'w') as apt:
|
|
apt.write(CLOUD_ARCHIVE.format(pocket))
|
|
if source.startswith('http:'):
|
|
with open('/etc/apt/sources.list.d/quantum.list', 'w') as apt:
|
|
apt.write("deb " + source + "\n")
|
|
key = config_get('key')
|
|
if key:
|
|
cmd = [
|
|
'apt-key',
|
|
'adv', '--keyserver keyserver.ubuntu.com',
|
|
'--recv-keys', key
|
|
]
|
|
subprocess.check_call(cmd)
|
|
cmd = [
|
|
'apt-get',
|
|
'update'
|
|
]
|
|
subprocess.check_call(cmd)
|
|
|
|
# Protocols
|
|
TCP = 'TCP'
|
|
UDP = 'UDP'
|
|
|
|
|
|
def expose(port, protocol='TCP'):
|
|
cmd = [
|
|
'open-port',
|
|
'{}/{}'.format(port, protocol)
|
|
]
|
|
subprocess.check_call(cmd)
|
|
|
|
|
|
def juju_log(severity, message):
|
|
cmd = [
|
|
'juju-log',
|
|
'--log-level', severity,
|
|
message
|
|
]
|
|
subprocess.check_call(cmd)
|
|
|
|
|
|
def relation_ids(relation):
|
|
cmd = [
|
|
'relation-ids',
|
|
relation
|
|
]
|
|
return subprocess.check_output(cmd).split() # IGNORE:E1103
|
|
|
|
|
|
def relation_list(rid):
|
|
cmd = [
|
|
'relation-list',
|
|
'-r', rid,
|
|
]
|
|
return subprocess.check_output(cmd).split() # IGNORE:E1103
|
|
|
|
|
|
def relation_get(attribute, unit=None, rid=None):
|
|
cmd = [
|
|
'relation-get',
|
|
]
|
|
if rid:
|
|
cmd.append('-r')
|
|
cmd.append(rid)
|
|
cmd.append(attribute)
|
|
if unit:
|
|
cmd.append(unit)
|
|
return subprocess.check_output(cmd).strip() # IGNORE:E1103
|
|
|
|
|
|
def relation_set(**kwargs):
|
|
cmd = [
|
|
'relation-set'
|
|
]
|
|
args = []
|
|
for k, v in kwargs.items():
|
|
if k == 'rid':
|
|
cmd.append('-r')
|
|
cmd.append(v)
|
|
else:
|
|
args.append('{}={}'.format(k, v))
|
|
cmd += args
|
|
subprocess.check_call(cmd)
|
|
|
|
|
|
def unit_get(attribute):
|
|
cmd = [
|
|
'unit-get',
|
|
attribute
|
|
]
|
|
return subprocess.check_output(cmd).strip() # IGNORE:E1103
|
|
|
|
|
|
def config_get(attribute):
|
|
cmd = [
|
|
'config-get',
|
|
attribute
|
|
]
|
|
return subprocess.check_output(cmd).strip() # IGNORE:E1103
|
|
|
|
|
|
def get_unit_hostname():
|
|
return socket.gethostname()
|
|
|
|
|
|
def get_host_ip(hostname=unit_get('private-address')):
|
|
cmd = [
|
|
'dig',
|
|
'+short',
|
|
hostname
|
|
]
|
|
return subprocess.check_output(cmd).strip() # IGNORE:E1103
|