descriptor-packages/tools/charm-generator/generator/ansible-charm/templates/ansible_lib.py.j2

   1 {#-
   2 # Copyright 2019 Whitestack, LLC
   3 #
   4 # Licensed under the Apache License, Version 2.0 (the "License"); you may
   5 # not use this file except in compliance with the License. You may obtain
   6 # a copy of the License at
   7 #
   8 #         http://www.apache.org/licenses/LICENSE-2.0
   9 #
  10 # Unless required by applicable law or agreed to in writing, software
  11 # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
  12 # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
  13 # License for the specific language governing permissions and limitations
  14 # under the License.
  15 #
  16 # For those usages not covered by the Apache License, Version 2.0 please
  17 # contact: esousa@whitestack.com or glavado@whitestack.com
  18 -#}
  19 {%- if license is defined -%}
  20 # Copyright {{ license.year }} {{ license.company }}
  21 #
  22 # Licensed under the Apache License, Version 2.0 (the "License"); you may
  23 # not use this file except in compliance with the License. You may obtain
  24 # a copy of the License at
  25 #
  26 #         http://www.apache.org/licenses/LICENSE-2.0
  27 #
  28 # Unless required by applicable law or agreed to in writing, software
  29 # distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
  30 # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
  31 # License for the specific language governing permissions and limitations
  32 # under the License.
  33 #
  34 # For those usages not covered by the Apache License, Version 2.0 please
  35 # contact: {{ license.email }}
  36 {%- endif %}
  37
  38 import fnmatch
  39 import os
  40 import yaml
  41 import subprocess
  42
  43 from charmhelpers.core.hookenv import config
  44
  45
  46 def create_hosts(cfg, hosts):
  47     inventory_path = '/etc/ansible/hosts'
  48
  49     with open(inventory_path, 'w') as f:
  50         f.write('[{}]\n'.format(hosts))
  51         h1 = '{0} ansible_connection=ssh ansible_ssh_user={1} ansible_ssh_pass={2} ' \
  52              'ansible_python_interpreter=/usr/bin/python3\n'.format(cfg['ssh-hostname'], cfg['ssh-username'],
  53                                                                     cfg['ssh-password'])
  54         f.write(h1)
  55
  56
  57 def create_ansible_cfg():
  58     ansible_config_path = '/etc/ansible/ansible.cfg'
  59
  60     with open(ansible_config_path, 'w') as f:
  61         f.write('[defaults]\n')
  62         f.write('host_key_checking = False\n')
  63         # logs playbook execution attempts to the specified path
  64         f.write('log_path = /var/log/ansible.log\n')
  65
  66 # Function to find the playbook path
  67 def find(pattern, path):
  68     result = ''
  69     for root, dirs, files in os.walk(path):
  70         for name in files:
  71             if fnmatch.fnmatch(name, pattern):
  72                 result = os.path.join(root, name)
  73     return result
  74
  75
  76 def execute_playbook(playbook_file, vars_dict=None):
  77     playbook_path = find(playbook_file, '/var/lib/juju/agents/')
  78
  79     cfg = config()
  80
  81     with open(playbook_path, 'r') as f:
  82         playbook_data = yaml.load(f)
  83
  84     hosts = 'all'
  85     if 'hosts' in playbook_data[0].keys() and playbook_data[0]['hosts']:
  86         hosts = playbook_data[0]['hosts']
  87
  88     create_ansible_cfg()
  89     create_hosts(cfg, hosts)
  90
  91     call = 'ansible-playbook %s ' % playbook_path
  92
  93     if vars_dict and isinstance(vars_dict, dict) and len(vars_dict) > 0:
  94         call += '--extra-vars '
  95
  96         string_var = ''
  97         for v in vars_dict.items():
  98             string_var += '%s=%s ' % v
  99
 100         string_var = string_var.strip()
 101         call += '"%s"' % string_var
 102
 103     call = call.strip()
 104     result = subprocess.check_output(call, shell=True)
 105
 106     return result