1006 lines
138 KiB
HTML
1006 lines
138 KiB
HTML
<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Transitional//EN"
|
|
"http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd">
|
|
|
|
|
|
<html xmlns="http://www.w3.org/1999/xhtml">
|
|
<head>
|
|
<meta http-equiv="Content-Type" content="text/html; charset=utf-8" />
|
|
|
|
<title>kollacli.common.inventory — kollacli 1.0.0.dev401 documentation</title>
|
|
|
|
<link rel="stylesheet" href="../../../_static/default.css" type="text/css" />
|
|
<link rel="stylesheet" href="../../../_static/pygments.css" type="text/css" />
|
|
|
|
<script type="text/javascript">
|
|
var DOCUMENTATION_OPTIONS = {
|
|
URL_ROOT: '../../../',
|
|
VERSION: '1.0.0.dev401',
|
|
COLLAPSE_INDEX: false,
|
|
FILE_SUFFIX: '.html',
|
|
HAS_SOURCE: true
|
|
};
|
|
</script>
|
|
<script type="text/javascript" src="../../../_static/jquery.js"></script>
|
|
<script type="text/javascript" src="../../../_static/underscore.js"></script>
|
|
<script type="text/javascript" src="../../../_static/doctools.js"></script>
|
|
<link rel="top" title="kollacli 1.0.0.dev401 documentation" href="../../../index.html" />
|
|
<link rel="up" title="Module code" href="../../index.html" />
|
|
</head>
|
|
<body>
|
|
<div class="related">
|
|
<h3>Navigation</h3>
|
|
<ul>
|
|
<li class="right" style="margin-right: 10px">
|
|
<a href="../../../genindex.html" title="General Index"
|
|
accesskey="I">index</a></li>
|
|
<li class="right" >
|
|
<a href="../../../py-modindex.html" title="Python Module Index"
|
|
>modules</a> |</li>
|
|
<li><a href="../../../index.html">kollacli 1.0.0.dev401 documentation</a> »</li>
|
|
<li><a href="../../index.html" accesskey="U">Module code</a> »</li>
|
|
</ul>
|
|
</div>
|
|
|
|
<div class="document">
|
|
<div class="documentwrapper">
|
|
<div class="bodywrapper">
|
|
<div class="body">
|
|
|
|
<h1>Source code for kollacli.common.inventory</h1><div class="highlight"><pre>
|
|
<span></span><span class="c1"># Copyright(c) 2016, Oracle and/or its affiliates. All Rights Reserved.</span>
|
|
<span class="c1">#</span>
|
|
<span class="c1"># Licensed under the Apache License, Version 2.0 (the "License"); you may</span>
|
|
<span class="c1"># not use this file except in compliance with the License. You may obtain</span>
|
|
<span class="c1"># a copy of the License at</span>
|
|
<span class="c1">#</span>
|
|
<span class="c1"># http://www.apache.org/licenses/LICENSE-2.0</span>
|
|
<span class="c1">#</span>
|
|
<span class="c1"># Unless required by applicable law or agreed to in writing, software</span>
|
|
<span class="c1"># distributed under the License is distributed on an "AS IS" BASIS, WITHOUT</span>
|
|
<span class="c1"># WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the</span>
|
|
<span class="c1"># License for the specific language governing permissions and limitations</span>
|
|
<span class="c1"># under the License.</span>
|
|
<span class="kn">import</span> <span class="nn">json</span>
|
|
<span class="kn">import</span> <span class="nn">jsonpickle</span>
|
|
<span class="kn">import</span> <span class="nn">logging</span>
|
|
<span class="kn">import</span> <span class="nn">os</span>
|
|
<span class="kn">import</span> <span class="nn">tempfile</span>
|
|
<span class="kn">import</span> <span class="nn">traceback</span>
|
|
<span class="kn">import</span> <span class="nn">uuid</span>
|
|
|
|
<span class="kn">import</span> <span class="nn">kollacli.i18n</span> <span class="kn">as</span> <span class="nn">u</span>
|
|
|
|
<span class="kn">from</span> <span class="nn">kollacli.api.exceptions</span> <span class="kn">import</span> <span class="n">FailedOperation</span>
|
|
<span class="kn">from</span> <span class="nn">kollacli.api.exceptions</span> <span class="kn">import</span> <span class="n">HostError</span>
|
|
<span class="kn">from</span> <span class="nn">kollacli.api.exceptions</span> <span class="kn">import</span> <span class="n">InvalidArgument</span>
|
|
<span class="kn">from</span> <span class="nn">kollacli.api.exceptions</span> <span class="kn">import</span> <span class="n">InvalidConfiguration</span>
|
|
<span class="kn">from</span> <span class="nn">kollacli.api.exceptions</span> <span class="kn">import</span> <span class="n">MissingArgument</span>
|
|
<span class="kn">from</span> <span class="nn">kollacli.api.exceptions</span> <span class="kn">import</span> <span class="n">NotInInventory</span>
|
|
<span class="kn">from</span> <span class="nn">kollacli.common.sshutils</span> <span class="kn">import</span> <span class="n">ssh_setup_host</span>
|
|
<span class="kn">from</span> <span class="nn">kollacli.common.utils</span> <span class="kn">import</span> <span class="n">get_admin_user</span>
|
|
<span class="kn">from</span> <span class="nn">kollacli.common.utils</span> <span class="kn">import</span> <span class="n">get_ansible_command</span>
|
|
<span class="kn">from</span> <span class="nn">kollacli.common.utils</span> <span class="kn">import</span> <span class="n">get_group_vars_dir</span>
|
|
<span class="kn">from</span> <span class="nn">kollacli.common.utils</span> <span class="kn">import</span> <span class="n">get_host_vars_dir</span>
|
|
<span class="kn">from</span> <span class="nn">kollacli.common.utils</span> <span class="kn">import</span> <span class="n">get_kollacli_etc</span>
|
|
<span class="kn">from</span> <span class="nn">kollacli.common.utils</span> <span class="kn">import</span> <span class="n">run_cmd</span>
|
|
<span class="kn">from</span> <span class="nn">kollacli.common.utils</span> <span class="kn">import</span> <span class="n">sync_read_file</span>
|
|
<span class="kn">from</span> <span class="nn">kollacli.common.utils</span> <span class="kn">import</span> <span class="n">sync_write_file</span>
|
|
|
|
<span class="n">ANSIBLE_SSH_USER</span> <span class="o">=</span> <span class="s1">'ansible_ssh_user'</span>
|
|
<span class="n">ANSIBLE_CONNECTION</span> <span class="o">=</span> <span class="s1">'ansible_connection'</span>
|
|
<span class="n">ANSIBLE_BECOME</span> <span class="o">=</span> <span class="s1">'ansible_become'</span>
|
|
|
|
<span class="n">INVENTORY_PATH</span> <span class="o">=</span> <span class="s1">'ansible/inventory.json'</span>
|
|
|
|
<span class="n">COMPUTE_GRP_NAME</span> <span class="o">=</span> <span class="s1">'compute'</span>
|
|
<span class="n">CONTROL_GRP_NAME</span> <span class="o">=</span> <span class="s1">'control'</span>
|
|
<span class="n">NETWORK_GRP_NAME</span> <span class="o">=</span> <span class="s1">'network'</span>
|
|
<span class="n">STORAGE_GRP_NAME</span> <span class="o">=</span> <span class="s1">'storage'</span>
|
|
<span class="n">DATABASE_GRP_NAME</span> <span class="o">=</span> <span class="s1">'database'</span>
|
|
|
|
<span class="n">DEPLOY_GROUPS</span> <span class="o">=</span> <span class="p">[</span>
|
|
<span class="n">COMPUTE_GRP_NAME</span><span class="p">,</span>
|
|
<span class="n">CONTROL_GRP_NAME</span><span class="p">,</span>
|
|
<span class="n">NETWORK_GRP_NAME</span><span class="p">,</span>
|
|
<span class="n">STORAGE_GRP_NAME</span><span class="p">,</span>
|
|
<span class="n">DATABASE_GRP_NAME</span><span class="p">,</span>
|
|
<span class="p">]</span>
|
|
|
|
<span class="n">SERVICES</span> <span class="o">=</span> <span class="p">{</span>
|
|
<span class="s1">'ceilometer'</span><span class="p">:</span> <span class="p">[</span><span class="s1">'ceilometer-alarm-evaluator'</span><span class="p">,</span> <span class="s1">'ceilometer-alarm-notifier'</span><span class="p">,</span>
|
|
<span class="s1">'ceilometer-api'</span><span class="p">,</span> <span class="s1">'ceilometer-central'</span><span class="p">,</span>
|
|
<span class="s1">'ceilometer-collector'</span><span class="p">,</span> <span class="s1">'ceilometer-notification'</span><span class="p">],</span>
|
|
<span class="s1">'cinder'</span><span class="p">:</span> <span class="p">[</span><span class="s1">'cinder-api'</span><span class="p">,</span> <span class="s1">'cinder-scheduler'</span><span class="p">,</span> <span class="s1">'cinder-backup'</span><span class="p">,</span>
|
|
<span class="s1">'cinder-volume'</span><span class="p">],</span>
|
|
<span class="s1">'glance'</span><span class="p">:</span> <span class="p">[</span><span class="s1">'glance-api'</span><span class="p">,</span> <span class="s1">'glance-registry'</span><span class="p">],</span>
|
|
<span class="s1">'haproxy'</span><span class="p">:</span> <span class="p">[],</span>
|
|
<span class="s1">'heat'</span><span class="p">:</span> <span class="p">[</span><span class="s1">'heat-api'</span><span class="p">,</span> <span class="s1">'heat-api-cfn'</span><span class="p">,</span> <span class="s1">'heat-engine'</span><span class="p">],</span>
|
|
<span class="s1">'horizon'</span><span class="p">:</span> <span class="p">[],</span>
|
|
<span class="s1">'keystone'</span><span class="p">:</span> <span class="p">[],</span>
|
|
<span class="s1">'memcached'</span><span class="p">:</span> <span class="p">[],</span>
|
|
<span class="s1">'murano'</span><span class="p">:</span> <span class="p">[</span><span class="s1">'murano-api'</span><span class="p">,</span> <span class="s1">'murano-engine'</span><span class="p">],</span>
|
|
<span class="s1">'mysqlcluster'</span><span class="p">:</span> <span class="p">[</span><span class="s1">'mysqlcluster-api'</span><span class="p">,</span> <span class="s1">'mysqlcluster-mgmt'</span><span class="p">,</span>
|
|
<span class="s1">'mysqlcluster-ndb'</span><span class="p">],</span>
|
|
<span class="s1">'neutron'</span><span class="p">:</span> <span class="p">[</span><span class="s1">'neutron-server'</span><span class="p">,</span> <span class="s1">'neutron-agents'</span><span class="p">],</span>
|
|
<span class="s1">'nova'</span><span class="p">:</span> <span class="p">[</span><span class="s1">'nova-api'</span><span class="p">,</span> <span class="s1">'nova-conductor'</span><span class="p">,</span> <span class="s1">'nova-consoleauth'</span><span class="p">,</span>
|
|
<span class="s1">'nova-novncproxy'</span><span class="p">,</span> <span class="s1">'nova-scheduler'</span><span class="p">],</span>
|
|
<span class="s1">'rabbitmq'</span><span class="p">:</span> <span class="p">[],</span>
|
|
<span class="s1">'swift'</span><span class="p">:</span> <span class="p">[</span><span class="s1">'swift-proxy-server'</span><span class="p">,</span> <span class="s1">'swift-account-server'</span><span class="p">,</span>
|
|
<span class="s1">'swift-container-server'</span><span class="p">,</span> <span class="s1">'swift-object-server'</span><span class="p">],</span>
|
|
<span class="p">}</span>
|
|
|
|
<span class="n">DEFAULT_GROUPS</span> <span class="o">=</span> <span class="p">{</span>
|
|
<span class="s1">'ceilometer'</span><span class="p">:</span> <span class="n">CONTROL_GRP_NAME</span><span class="p">,</span>
|
|
<span class="s1">'cinder'</span><span class="p">:</span> <span class="n">CONTROL_GRP_NAME</span><span class="p">,</span>
|
|
<span class="s1">'glance'</span><span class="p">:</span> <span class="n">CONTROL_GRP_NAME</span><span class="p">,</span>
|
|
<span class="s1">'haproxy'</span><span class="p">:</span> <span class="n">CONTROL_GRP_NAME</span><span class="p">,</span>
|
|
<span class="s1">'heat'</span><span class="p">:</span> <span class="n">CONTROL_GRP_NAME</span><span class="p">,</span>
|
|
<span class="s1">'horizon'</span><span class="p">:</span> <span class="n">CONTROL_GRP_NAME</span><span class="p">,</span>
|
|
<span class="s1">'keystone'</span><span class="p">:</span> <span class="n">CONTROL_GRP_NAME</span><span class="p">,</span>
|
|
<span class="s1">'memcached'</span><span class="p">:</span> <span class="n">CONTROL_GRP_NAME</span><span class="p">,</span>
|
|
<span class="s1">'murano'</span><span class="p">:</span> <span class="n">CONTROL_GRP_NAME</span><span class="p">,</span>
|
|
<span class="s1">'mysqlcluster'</span><span class="p">:</span> <span class="n">CONTROL_GRP_NAME</span><span class="p">,</span>
|
|
<span class="s1">'neutron'</span><span class="p">:</span> <span class="n">NETWORK_GRP_NAME</span><span class="p">,</span>
|
|
<span class="s1">'nova'</span><span class="p">:</span> <span class="n">CONTROL_GRP_NAME</span><span class="p">,</span>
|
|
<span class="s1">'rabbitmq'</span><span class="p">:</span> <span class="n">CONTROL_GRP_NAME</span><span class="p">,</span>
|
|
<span class="s1">'swift'</span><span class="p">:</span> <span class="n">CONTROL_GRP_NAME</span><span class="p">,</span>
|
|
<span class="p">}</span>
|
|
|
|
<span class="n">DEFAULT_OVERRIDES</span> <span class="o">=</span> <span class="p">{</span>
|
|
<span class="s1">'cinder-backup'</span><span class="p">:</span> <span class="n">STORAGE_GRP_NAME</span><span class="p">,</span>
|
|
<span class="s1">'cinder-volume'</span><span class="p">:</span> <span class="n">STORAGE_GRP_NAME</span><span class="p">,</span>
|
|
<span class="s1">'mysqlcluster-ndb'</span><span class="p">:</span> <span class="n">DATABASE_GRP_NAME</span><span class="p">,</span>
|
|
<span class="s1">'neutron-server'</span><span class="p">:</span> <span class="n">CONTROL_GRP_NAME</span><span class="p">,</span>
|
|
<span class="s1">'swift-account-server'</span><span class="p">:</span> <span class="n">STORAGE_GRP_NAME</span><span class="p">,</span>
|
|
<span class="s1">'swift-container-server'</span><span class="p">:</span> <span class="n">STORAGE_GRP_NAME</span><span class="p">,</span>
|
|
<span class="s1">'swift-object-server'</span><span class="p">:</span> <span class="n">STORAGE_GRP_NAME</span><span class="p">,</span>
|
|
<span class="p">}</span>
|
|
|
|
|
|
<span class="c1"># these groups cannot be deleted, they are required by kolla</span>
|
|
<span class="n">PROTECTED_GROUPS</span> <span class="o">=</span> <span class="p">[</span><span class="n">COMPUTE_GRP_NAME</span><span class="p">]</span>
|
|
|
|
<span class="n">LOG</span> <span class="o">=</span> <span class="n">logging</span><span class="o">.</span><span class="n">getLogger</span><span class="p">(</span><span class="n">__name__</span><span class="p">)</span>
|
|
|
|
|
|
<div class="viewcode-block" id="remove_temp_inventory"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.remove_temp_inventory">[docs]</a><span class="k">def</span> <span class="nf">remove_temp_inventory</span><span class="p">(</span><span class="n">path</span><span class="p">):</span>
|
|
<span class="sd">"""remove temp inventory file and its parent directory"""</span>
|
|
<span class="k">if</span> <span class="n">path</span><span class="p">:</span>
|
|
<span class="k">if</span> <span class="n">os</span><span class="o">.</span><span class="n">path</span><span class="o">.</span><span class="n">exists</span><span class="p">(</span><span class="n">path</span><span class="p">):</span>
|
|
<span class="n">os</span><span class="o">.</span><span class="n">remove</span><span class="p">(</span><span class="n">path</span><span class="p">)</span>
|
|
<span class="n">dirpath</span> <span class="o">=</span> <span class="n">os</span><span class="o">.</span><span class="n">path</span><span class="o">.</span><span class="n">dirname</span><span class="p">(</span><span class="n">path</span><span class="p">)</span>
|
|
<span class="k">if</span> <span class="n">os</span><span class="o">.</span><span class="n">path</span><span class="o">.</span><span class="n">exists</span><span class="p">(</span><span class="n">dirpath</span><span class="p">):</span>
|
|
<span class="n">os</span><span class="o">.</span><span class="n">rmdir</span><span class="p">(</span><span class="n">dirpath</span><span class="p">)</span>
|
|
|
|
</div>
|
|
<div class="viewcode-block" id="Host"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Host">[docs]</a><span class="k">class</span> <span class="nc">Host</span><span class="p">(</span><span class="nb">object</span><span class="p">):</span>
|
|
<span class="n">class_version</span> <span class="o">=</span> <span class="mi">1</span>
|
|
|
|
<span class="k">def</span> <span class="nf">__init__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">hostname</span><span class="p">):</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">name</span> <span class="o">=</span> <span class="n">hostname</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">alias</span> <span class="o">=</span> <span class="s1">''</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">is_mgmt</span> <span class="o">=</span> <span class="bp">False</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">hypervisor</span> <span class="o">=</span> <span class="s1">''</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">vars</span> <span class="o">=</span> <span class="p">{}</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">version</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">__class__</span><span class="o">.</span><span class="n">class_version</span>
|
|
|
|
<div class="viewcode-block" id="Host.get_vars"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Host.get_vars">[docs]</a> <span class="k">def</span> <span class="nf">get_vars</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
|
|
<span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">vars</span><span class="o">.</span><span class="n">copy</span><span class="p">()</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Host.set_var"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Host.set_var">[docs]</a> <span class="k">def</span> <span class="nf">set_var</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">name</span><span class="p">,</span> <span class="n">value</span><span class="p">):</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">vars</span><span class="p">[</span><span class="n">name</span><span class="p">]</span> <span class="o">=</span> <span class="n">value</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Host.upgrade"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Host.upgrade">[docs]</a> <span class="k">def</span> <span class="nf">upgrade</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
|
|
<span class="k">pass</span>
|
|
|
|
</div></div>
|
|
<div class="viewcode-block" id="HostGroup"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.HostGroup">[docs]</a><span class="k">class</span> <span class="nc">HostGroup</span><span class="p">(</span><span class="nb">object</span><span class="p">):</span>
|
|
<span class="n">class_version</span> <span class="o">=</span> <span class="mi">1</span>
|
|
|
|
<span class="k">def</span> <span class="nf">__init__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">name</span><span class="p">):</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">name</span> <span class="o">=</span> <span class="n">name</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">hostnames</span> <span class="o">=</span> <span class="p">[]</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">vars</span> <span class="o">=</span> <span class="p">{}</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">version</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">__class__</span><span class="o">.</span><span class="n">class_version</span>
|
|
|
|
<div class="viewcode-block" id="HostGroup.upgrade"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.HostGroup.upgrade">[docs]</a> <span class="k">def</span> <span class="nf">upgrade</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
|
|
<span class="k">pass</span>
|
|
</div>
|
|
<div class="viewcode-block" id="HostGroup.add_host"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.HostGroup.add_host">[docs]</a> <span class="k">def</span> <span class="nf">add_host</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">host</span><span class="p">):</span>
|
|
<span class="k">if</span> <span class="n">host</span><span class="o">.</span><span class="n">name</span> <span class="ow">not</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">hostnames</span><span class="p">:</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">hostnames</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">host</span><span class="o">.</span><span class="n">name</span><span class="p">)</span>
|
|
</div>
|
|
<div class="viewcode-block" id="HostGroup.remove_host"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.HostGroup.remove_host">[docs]</a> <span class="k">def</span> <span class="nf">remove_host</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">host</span><span class="p">):</span>
|
|
<span class="k">if</span> <span class="n">host</span><span class="o">.</span><span class="n">name</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">hostnames</span><span class="p">:</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">hostnames</span><span class="o">.</span><span class="n">remove</span><span class="p">(</span><span class="n">host</span><span class="o">.</span><span class="n">name</span><span class="p">)</span>
|
|
</div>
|
|
<div class="viewcode-block" id="HostGroup.get_hostnames"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.HostGroup.get_hostnames">[docs]</a> <span class="k">def</span> <span class="nf">get_hostnames</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
|
|
<span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">hostnames</span>
|
|
</div>
|
|
<div class="viewcode-block" id="HostGroup.get_vars"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.HostGroup.get_vars">[docs]</a> <span class="k">def</span> <span class="nf">get_vars</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
|
|
<span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">vars</span><span class="o">.</span><span class="n">copy</span><span class="p">()</span>
|
|
</div>
|
|
<div class="viewcode-block" id="HostGroup.set_var"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.HostGroup.set_var">[docs]</a> <span class="k">def</span> <span class="nf">set_var</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">name</span><span class="p">,</span> <span class="n">value</span><span class="p">):</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">vars</span><span class="p">[</span><span class="n">name</span><span class="p">]</span> <span class="o">=</span> <span class="n">value</span>
|
|
</div>
|
|
<div class="viewcode-block" id="HostGroup.clear_var"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.HostGroup.clear_var">[docs]</a> <span class="k">def</span> <span class="nf">clear_var</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">name</span><span class="p">):</span>
|
|
<span class="k">if</span> <span class="n">name</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">vars</span><span class="p">:</span>
|
|
<span class="k">del</span> <span class="bp">self</span><span class="o">.</span><span class="n">vars</span><span class="p">[</span><span class="n">name</span><span class="p">]</span>
|
|
</div>
|
|
<div class="viewcode-block" id="HostGroup.set_remote"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.HostGroup.set_remote">[docs]</a> <span class="k">def</span> <span class="nf">set_remote</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">remote_flag</span><span class="p">):</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">set_var</span><span class="p">(</span><span class="n">ANSIBLE_BECOME</span><span class="p">,</span> <span class="s1">'yes'</span><span class="p">)</span>
|
|
<span class="k">if</span> <span class="n">remote_flag</span><span class="p">:</span>
|
|
<span class="c1"># set the ssh info for all the servers in the group</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">set_var</span><span class="p">(</span><span class="n">ANSIBLE_SSH_USER</span><span class="p">,</span> <span class="n">get_admin_user</span><span class="p">())</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">clear_var</span><span class="p">(</span><span class="n">ANSIBLE_CONNECTION</span><span class="p">)</span>
|
|
<span class="k">else</span><span class="p">:</span>
|
|
<span class="c1"># remove ssh info, add local connection type</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">set_var</span><span class="p">(</span><span class="n">ANSIBLE_CONNECTION</span><span class="p">,</span> <span class="s1">'local'</span><span class="p">)</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">clear_var</span><span class="p">(</span><span class="n">ANSIBLE_SSH_USER</span><span class="p">)</span>
|
|
|
|
</div></div>
|
|
<div class="viewcode-block" id="Service"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Service">[docs]</a><span class="k">class</span> <span class="nc">Service</span><span class="p">(</span><span class="nb">object</span><span class="p">):</span>
|
|
<span class="n">class_version</span> <span class="o">=</span> <span class="mi">1</span>
|
|
|
|
<span class="k">def</span> <span class="nf">__init__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">name</span><span class="p">):</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">name</span> <span class="o">=</span> <span class="n">name</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">_sub_servicenames</span> <span class="o">=</span> <span class="p">[]</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">_groupnames</span> <span class="o">=</span> <span class="p">[]</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">_vars</span> <span class="o">=</span> <span class="p">{}</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">version</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">__class__</span><span class="o">.</span><span class="n">class_version</span>
|
|
|
|
<div class="viewcode-block" id="Service.upgrade"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Service.upgrade">[docs]</a> <span class="k">def</span> <span class="nf">upgrade</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
|
|
<span class="k">pass</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Service.add_groupname"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Service.add_groupname">[docs]</a> <span class="k">def</span> <span class="nf">add_groupname</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">groupname</span><span class="p">):</span>
|
|
<span class="k">if</span> <span class="n">groupname</span> <span class="ow">is</span> <span class="ow">not</span> <span class="bp">None</span> <span class="ow">and</span> <span class="n">groupname</span> <span class="ow">not</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">_groupnames</span><span class="p">:</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">_groupnames</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">groupname</span><span class="p">)</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Service.remove_groupname"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Service.remove_groupname">[docs]</a> <span class="k">def</span> <span class="nf">remove_groupname</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">groupname</span><span class="p">):</span>
|
|
<span class="k">if</span> <span class="n">groupname</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">_groupnames</span><span class="p">:</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">_groupnames</span><span class="o">.</span><span class="n">remove</span><span class="p">(</span><span class="n">groupname</span><span class="p">)</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Service.get_groupnames"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Service.get_groupnames">[docs]</a> <span class="k">def</span> <span class="nf">get_groupnames</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
|
|
<span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_groupnames</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Service.get_sub_servicenames"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Service.get_sub_servicenames">[docs]</a> <span class="k">def</span> <span class="nf">get_sub_servicenames</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
|
|
<span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_sub_servicenames</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Service.add_sub_servicename"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Service.add_sub_servicename">[docs]</a> <span class="k">def</span> <span class="nf">add_sub_servicename</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">sub_servicename</span><span class="p">):</span>
|
|
<span class="k">if</span> <span class="n">sub_servicename</span> <span class="ow">not</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">_sub_servicenames</span><span class="p">:</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">_sub_servicenames</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">sub_servicename</span><span class="p">)</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Service.get_vars"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Service.get_vars">[docs]</a> <span class="k">def</span> <span class="nf">get_vars</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
|
|
<span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_vars</span><span class="o">.</span><span class="n">copy</span><span class="p">()</span>
|
|
|
|
</div></div>
|
|
<div class="viewcode-block" id="SubService"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.SubService">[docs]</a><span class="k">class</span> <span class="nc">SubService</span><span class="p">(</span><span class="nb">object</span><span class="p">):</span>
|
|
<span class="n">class_version</span> <span class="o">=</span> <span class="mi">1</span>
|
|
|
|
<span class="k">def</span> <span class="nf">__init__</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">name</span><span class="p">):</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">name</span> <span class="o">=</span> <span class="n">name</span>
|
|
|
|
<span class="c1"># groups and parent services are mutually exclusive</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">_groupnames</span> <span class="o">=</span> <span class="p">[]</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">_parent_servicename</span> <span class="o">=</span> <span class="bp">None</span>
|
|
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">_vars</span> <span class="o">=</span> <span class="p">{}</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">version</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">__class__</span><span class="o">.</span><span class="n">class_version</span>
|
|
|
|
<div class="viewcode-block" id="SubService.upgrade"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.SubService.upgrade">[docs]</a> <span class="k">def</span> <span class="nf">upgrade</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
|
|
<span class="k">pass</span>
|
|
</div>
|
|
<div class="viewcode-block" id="SubService.add_groupname"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.SubService.add_groupname">[docs]</a> <span class="k">def</span> <span class="nf">add_groupname</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">groupname</span><span class="p">):</span>
|
|
<span class="k">if</span> <span class="n">groupname</span> <span class="ow">not</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">_groupnames</span><span class="p">:</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">_groupnames</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">groupname</span><span class="p">)</span>
|
|
</div>
|
|
<div class="viewcode-block" id="SubService.remove_groupname"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.SubService.remove_groupname">[docs]</a> <span class="k">def</span> <span class="nf">remove_groupname</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">groupname</span><span class="p">):</span>
|
|
<span class="k">if</span> <span class="n">groupname</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">_groupnames</span><span class="p">:</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">_groupnames</span><span class="o">.</span><span class="n">remove</span><span class="p">(</span><span class="n">groupname</span><span class="p">)</span>
|
|
<span class="k">if</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">_groupnames</span><span class="p">:</span>
|
|
<span class="c1"># no groups left, re-associate to the parent</span>
|
|
<span class="k">for</span> <span class="n">servicename</span> <span class="ow">in</span> <span class="n">SERVICES</span><span class="p">:</span>
|
|
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">name</span> <span class="ow">in</span> <span class="n">SERVICES</span><span class="p">[</span><span class="n">servicename</span><span class="p">]:</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">set_parent_servicename</span><span class="p">(</span><span class="n">servicename</span><span class="p">)</span>
|
|
<span class="k">break</span>
|
|
</div>
|
|
<div class="viewcode-block" id="SubService.get_groupnames"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.SubService.get_groupnames">[docs]</a> <span class="k">def</span> <span class="nf">get_groupnames</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
|
|
<span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_groupnames</span>
|
|
</div>
|
|
<div class="viewcode-block" id="SubService.set_parent_servicename"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.SubService.set_parent_servicename">[docs]</a> <span class="k">def</span> <span class="nf">set_parent_servicename</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">parent_svc_name</span><span class="p">):</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">_parent_servicename</span> <span class="o">=</span> <span class="n">parent_svc_name</span>
|
|
</div>
|
|
<div class="viewcode-block" id="SubService.get_parent_servicename"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.SubService.get_parent_servicename">[docs]</a> <span class="k">def</span> <span class="nf">get_parent_servicename</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
|
|
<span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_parent_servicename</span>
|
|
</div>
|
|
<div class="viewcode-block" id="SubService.get_vars"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.SubService.get_vars">[docs]</a> <span class="k">def</span> <span class="nf">get_vars</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
|
|
<span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">vars</span><span class="o">.</span><span class="n">copy</span><span class="p">()</span>
|
|
|
|
</div></div>
|
|
<div class="viewcode-block" id="Inventory"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Inventory">[docs]</a><span class="k">class</span> <span class="nc">Inventory</span><span class="p">(</span><span class="nb">object</span><span class="p">):</span>
|
|
<span class="n">class_version</span> <span class="o">=</span> <span class="mi">3</span>
|
|
|
|
<span class="sd">"""class version history</span>
|
|
|
|
<span class="sd"> 1: initial release</span>
|
|
<span class="sd"> """</span>
|
|
<span class="k">def</span> <span class="nf">__init__</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">_groups</span> <span class="o">=</span> <span class="p">{}</span> <span class="c1"># kv = name:object</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">_hosts</span> <span class="o">=</span> <span class="p">{}</span> <span class="c1"># kv = name:object</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">_services</span> <span class="o">=</span> <span class="p">{}</span> <span class="c1"># kv = name:object</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">_sub_services</span> <span class="o">=</span> <span class="p">{}</span> <span class="c1"># kv = name:object</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">vars</span> <span class="o">=</span> <span class="p">{}</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">version</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">__class__</span><span class="o">.</span><span class="n">class_version</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">remote_mode</span> <span class="o">=</span> <span class="bp">True</span>
|
|
|
|
<span class="c1"># initialize the inventory to its defaults</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">_create_default_inventory</span><span class="p">()</span>
|
|
|
|
<div class="viewcode-block" id="Inventory.upgrade"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Inventory.upgrade">[docs]</a> <span class="k">def</span> <span class="nf">upgrade</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
|
|
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">version</span> <span class="o"><=</span> <span class="mi">1</span><span class="p">:</span>
|
|
<span class="c1"># upgrade from inventory v1</span>
|
|
|
|
<span class="c1"># add ceilometer to inventory</span>
|
|
<span class="n">svc_name</span> <span class="o">=</span> <span class="s1">'ceilometer'</span>
|
|
<span class="n">svc</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">create_service</span><span class="p">(</span><span class="n">svc_name</span><span class="p">)</span>
|
|
|
|
<span class="c1"># associate ceilometer with all groups that heat is in.</span>
|
|
<span class="n">clone_svc</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_service</span><span class="p">(</span><span class="s1">'heat'</span><span class="p">)</span>
|
|
<span class="n">groups</span> <span class="o">=</span> <span class="n">clone_svc</span><span class="o">.</span><span class="n">get_groupnames</span><span class="p">()</span>
|
|
<span class="k">for</span> <span class="n">group</span> <span class="ow">in</span> <span class="n">groups</span><span class="p">:</span>
|
|
<span class="n">svc</span><span class="o">.</span><span class="n">add_groupname</span><span class="p">(</span><span class="n">group</span><span class="p">)</span>
|
|
|
|
<span class="c1"># stitch sub-service to service and set override</span>
|
|
<span class="c1"># groups</span>
|
|
<span class="k">for</span> <span class="n">sub_svc_name</span> <span class="ow">in</span> <span class="n">SERVICES</span><span class="p">[</span><span class="n">svc_name</span><span class="p">]:</span>
|
|
<span class="n">sub_svc</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">create_sub_service</span><span class="p">(</span><span class="n">sub_svc_name</span><span class="p">)</span>
|
|
<span class="n">sub_svc</span><span class="o">.</span><span class="n">set_parent_servicename</span><span class="p">(</span><span class="n">svc_name</span><span class="p">)</span>
|
|
<span class="n">svc</span><span class="o">.</span><span class="n">add_sub_servicename</span><span class="p">(</span><span class="n">sub_svc_name</span><span class="p">)</span>
|
|
|
|
<span class="k">if</span> <span class="bp">self</span><span class="o">.</span><span class="n">version</span> <span class="o"><=</span> <span class="mi">2</span><span class="p">:</span>
|
|
<span class="c1"># upgrade from inventory v2</span>
|
|
|
|
<span class="c1"># some sub-services may be missing their parent associations.</span>
|
|
<span class="c1"># they are now needed in v3.</span>
|
|
<span class="k">for</span> <span class="n">svc</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_services</span><span class="p">():</span>
|
|
<span class="k">for</span> <span class="n">sub_svcname</span> <span class="ow">in</span> <span class="n">svc</span><span class="o">.</span><span class="n">get_sub_servicenames</span><span class="p">():</span>
|
|
<span class="n">sub_svc</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_sub_service</span><span class="p">(</span><span class="n">sub_svcname</span><span class="p">)</span>
|
|
<span class="k">if</span> <span class="ow">not</span> <span class="n">sub_svc</span><span class="o">.</span><span class="n">get_parent_servicename</span><span class="p">():</span>
|
|
<span class="n">sub_svc</span><span class="o">.</span><span class="n">set_parent_servicename</span><span class="p">(</span><span class="n">svc</span><span class="o">.</span><span class="n">name</span><span class="p">)</span>
|
|
|
|
<span class="c1"># update the version and save upgraded inventory file</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">version</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">__class__</span><span class="o">.</span><span class="n">class_version</span>
|
|
<span class="n">Inventory</span><span class="o">.</span><span class="n">save</span><span class="p">(</span><span class="bp">self</span><span class="p">)</span>
|
|
</div>
|
|
<span class="nd">@staticmethod</span>
|
|
<div class="viewcode-block" id="Inventory.load"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Inventory.load">[docs]</a> <span class="k">def</span> <span class="nf">load</span><span class="p">():</span>
|
|
<span class="sd">"""load the inventory from a pickle file"""</span>
|
|
<span class="n">inventory_path</span> <span class="o">=</span> <span class="n">os</span><span class="o">.</span><span class="n">path</span><span class="o">.</span><span class="n">join</span><span class="p">(</span><span class="n">get_kollacli_etc</span><span class="p">(),</span> <span class="n">INVENTORY_PATH</span><span class="p">)</span>
|
|
<span class="n">data</span> <span class="o">=</span> <span class="s1">''</span>
|
|
<span class="k">try</span><span class="p">:</span>
|
|
<span class="k">if</span> <span class="n">os</span><span class="o">.</span><span class="n">path</span><span class="o">.</span><span class="n">exists</span><span class="p">(</span><span class="n">inventory_path</span><span class="p">):</span>
|
|
<span class="n">data</span> <span class="o">=</span> <span class="n">sync_read_file</span><span class="p">(</span><span class="n">inventory_path</span><span class="p">)</span>
|
|
|
|
<span class="c1"># The inventory path changed between v1 and v2. Need to change</span>
|
|
<span class="c1"># path throughout the inventory. This has to be done before</span>
|
|
<span class="c1"># the pickle decode.</span>
|
|
<span class="k">if</span> <span class="s1">'kollacli.common.inventory'</span> <span class="ow">not</span> <span class="ow">in</span> <span class="n">data</span><span class="p">:</span>
|
|
<span class="n">data</span> <span class="o">=</span> <span class="n">data</span><span class="o">.</span><span class="n">replace</span><span class="p">(</span>
|
|
<span class="s1">'"py/object": "kollacli.ansible.inventory.'</span><span class="p">,</span>
|
|
<span class="s1">'"py/object": "kollacli.common.inventory.'</span><span class="p">)</span>
|
|
|
|
<span class="k">if</span> <span class="n">data</span><span class="o">.</span><span class="n">strip</span><span class="p">():</span>
|
|
<span class="n">inventory</span> <span class="o">=</span> <span class="n">jsonpickle</span><span class="o">.</span><span class="n">decode</span><span class="p">(</span><span class="n">data</span><span class="p">)</span>
|
|
|
|
<span class="c1"># upgrade version handling</span>
|
|
<span class="k">if</span> <span class="n">inventory</span><span class="o">.</span><span class="n">version</span> <span class="o">!=</span> <span class="n">inventory</span><span class="o">.</span><span class="n">class_version</span><span class="p">:</span>
|
|
<span class="n">inventory</span><span class="o">.</span><span class="n">upgrade</span><span class="p">()</span>
|
|
<span class="k">else</span><span class="p">:</span>
|
|
<span class="n">inventory</span> <span class="o">=</span> <span class="n">Inventory</span><span class="p">()</span>
|
|
<span class="k">except</span> <span class="ne">Exception</span><span class="p">:</span>
|
|
<span class="k">raise</span> <span class="n">FailedOperation</span><span class="p">(</span>
|
|
<span class="n">u</span><span class="o">.</span><span class="n">_</span><span class="p">(</span><span class="s1">'Loading inventory failed. : {error}'</span><span class="p">)</span>
|
|
<span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="n">error</span><span class="o">=</span><span class="n">traceback</span><span class="o">.</span><span class="n">format_exc</span><span class="p">()))</span>
|
|
<span class="k">return</span> <span class="n">inventory</span>
|
|
</div>
|
|
<span class="nd">@staticmethod</span>
|
|
<div class="viewcode-block" id="Inventory.save"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Inventory.save">[docs]</a> <span class="k">def</span> <span class="nf">save</span><span class="p">(</span><span class="n">inventory</span><span class="p">):</span>
|
|
<span class="sd">"""Save the inventory in a pickle file"""</span>
|
|
<span class="n">inventory_path</span> <span class="o">=</span> <span class="n">os</span><span class="o">.</span><span class="n">path</span><span class="o">.</span><span class="n">join</span><span class="p">(</span><span class="n">get_kollacli_etc</span><span class="p">(),</span> <span class="n">INVENTORY_PATH</span><span class="p">)</span>
|
|
<span class="k">try</span><span class="p">:</span>
|
|
<span class="c1"># multiple trips thru json to render a readable inventory file</span>
|
|
<span class="n">data</span> <span class="o">=</span> <span class="n">jsonpickle</span><span class="o">.</span><span class="n">encode</span><span class="p">(</span><span class="n">inventory</span><span class="p">)</span>
|
|
<span class="n">data_str</span> <span class="o">=</span> <span class="n">json</span><span class="o">.</span><span class="n">loads</span><span class="p">(</span><span class="n">data</span><span class="p">)</span>
|
|
<span class="n">pretty_data</span> <span class="o">=</span> <span class="n">json</span><span class="o">.</span><span class="n">dumps</span><span class="p">(</span><span class="n">data_str</span><span class="p">,</span> <span class="n">indent</span><span class="o">=</span><span class="mi">4</span><span class="p">)</span>
|
|
<span class="n">sync_write_file</span><span class="p">(</span><span class="n">inventory_path</span><span class="p">,</span> <span class="n">pretty_data</span><span class="p">)</span>
|
|
|
|
<span class="k">except</span> <span class="ne">Exception</span> <span class="k">as</span> <span class="n">e</span><span class="p">:</span>
|
|
<span class="k">raise</span> <span class="n">FailedOperation</span><span class="p">(</span>
|
|
<span class="n">u</span><span class="o">.</span><span class="n">_</span><span class="p">(</span><span class="s1">'Saving inventory failed. : {error}'</span><span class="p">)</span>
|
|
<span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="n">error</span><span class="o">=</span><span class="nb">str</span><span class="p">(</span><span class="n">e</span><span class="p">)))</span>
|
|
</div>
|
|
<span class="k">def</span> <span class="nf">_create_default_inventory</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
|
|
|
|
<span class="c1"># create the default groups</span>
|
|
<span class="k">for</span> <span class="n">groupname</span> <span class="ow">in</span> <span class="n">DEPLOY_GROUPS</span><span class="p">:</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">add_group</span><span class="p">(</span><span class="n">groupname</span><span class="p">)</span>
|
|
|
|
<span class="c1"># create the default services/sub_services & their default groups</span>
|
|
<span class="k">for</span> <span class="n">svcname</span> <span class="ow">in</span> <span class="n">SERVICES</span><span class="p">:</span>
|
|
<span class="n">svc</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">create_service</span><span class="p">(</span><span class="n">svcname</span><span class="p">)</span>
|
|
<span class="n">default_grpname</span> <span class="o">=</span> <span class="n">DEFAULT_GROUPS</span><span class="p">[</span><span class="n">svcname</span><span class="p">]</span>
|
|
<span class="n">svc</span><span class="o">.</span><span class="n">add_groupname</span><span class="p">(</span><span class="n">default_grpname</span><span class="p">)</span>
|
|
<span class="n">sub_svcnames</span> <span class="o">=</span> <span class="n">SERVICES</span><span class="p">[</span><span class="n">svcname</span><span class="p">]</span>
|
|
<span class="k">if</span> <span class="n">sub_svcnames</span><span class="p">:</span>
|
|
<span class="k">for</span> <span class="n">sub_svcname</span> <span class="ow">in</span> <span class="n">sub_svcnames</span><span class="p">:</span>
|
|
<span class="c1"># create a subservice</span>
|
|
<span class="n">svc</span><span class="o">.</span><span class="n">add_sub_servicename</span><span class="p">(</span><span class="n">sub_svcname</span><span class="p">)</span>
|
|
<span class="n">sub_svc</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">create_sub_service</span><span class="p">(</span><span class="n">sub_svcname</span><span class="p">)</span>
|
|
<span class="n">sub_svc</span><span class="o">.</span><span class="n">set_parent_servicename</span><span class="p">(</span><span class="n">svc</span><span class="o">.</span><span class="n">name</span><span class="p">)</span>
|
|
<span class="k">if</span> <span class="n">sub_svc</span><span class="o">.</span><span class="n">name</span> <span class="ow">in</span> <span class="n">DEFAULT_OVERRIDES</span><span class="p">:</span>
|
|
<span class="n">sub_svc</span><span class="o">.</span><span class="n">add_groupname</span><span class="p">(</span><span class="n">DEFAULT_OVERRIDES</span><span class="p">[</span><span class="n">sub_svc</span><span class="o">.</span><span class="n">name</span><span class="p">])</span>
|
|
|
|
<div class="viewcode-block" id="Inventory.get_hosts"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Inventory.get_hosts">[docs]</a> <span class="k">def</span> <span class="nf">get_hosts</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
|
|
<span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_hosts</span><span class="o">.</span><span class="n">values</span><span class="p">()</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Inventory.get_hostnames"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Inventory.get_hostnames">[docs]</a> <span class="k">def</span> <span class="nf">get_hostnames</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
|
|
<span class="k">return</span> <span class="nb">list</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">_hosts</span><span class="o">.</span><span class="n">keys</span><span class="p">())</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Inventory.get_host"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Inventory.get_host">[docs]</a> <span class="k">def</span> <span class="nf">get_host</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">hostname</span><span class="p">):</span>
|
|
<span class="n">host</span> <span class="o">=</span> <span class="bp">None</span>
|
|
<span class="k">if</span> <span class="n">hostname</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">_hosts</span><span class="p">:</span>
|
|
<span class="n">host</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_hosts</span><span class="p">[</span><span class="n">hostname</span><span class="p">]</span>
|
|
<span class="k">return</span> <span class="n">host</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Inventory.add_host"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Inventory.add_host">[docs]</a> <span class="k">def</span> <span class="nf">add_host</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">hostname</span><span class="p">,</span> <span class="n">groupname</span><span class="o">=</span><span class="bp">None</span><span class="p">):</span>
|
|
<span class="sd">"""add host</span>
|
|
|
|
<span class="sd"> if groupname is none, create a new host</span>
|
|
<span class="sd"> if group name is not none, add host to group</span>
|
|
<span class="sd"> """</span>
|
|
<span class="k">if</span> <span class="n">groupname</span> <span class="ow">and</span> <span class="n">groupname</span> <span class="ow">not</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">_groups</span><span class="p">:</span>
|
|
<span class="k">raise</span> <span class="n">NotInInventory</span><span class="p">(</span><span class="n">u</span><span class="o">.</span><span class="n">_</span><span class="p">(</span><span class="s1">'Group'</span><span class="p">),</span> <span class="n">groupname</span><span class="p">)</span>
|
|
|
|
<span class="k">if</span> <span class="n">groupname</span> <span class="ow">and</span> <span class="n">hostname</span> <span class="ow">not</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">_hosts</span><span class="p">:</span>
|
|
<span class="c1"># if a groupname is specified, the host must already exist</span>
|
|
<span class="k">raise</span> <span class="n">NotInInventory</span><span class="p">(</span><span class="n">u</span><span class="o">.</span><span class="n">_</span><span class="p">(</span><span class="s1">'Host'</span><span class="p">),</span> <span class="n">hostname</span><span class="p">)</span>
|
|
|
|
<span class="k">if</span> <span class="ow">not</span> <span class="n">groupname</span> <span class="ow">and</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">remote_mode</span> <span class="ow">and</span> <span class="nb">len</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">_hosts</span><span class="p">)</span> <span class="o">>=</span> <span class="mi">1</span><span class="p">:</span>
|
|
<span class="k">raise</span> <span class="n">InvalidConfiguration</span><span class="p">(</span>
|
|
<span class="n">u</span><span class="o">.</span><span class="n">_</span><span class="p">(</span><span class="s1">'Cannot have more than one host when in local deploy '</span>
|
|
<span class="s1">'mode.'</span><span class="p">))</span>
|
|
|
|
<span class="n">changed</span> <span class="o">=</span> <span class="bp">False</span>
|
|
<span class="c1"># create new host if it doesn't exist</span>
|
|
<span class="n">host</span> <span class="o">=</span> <span class="n">Host</span><span class="p">(</span><span class="n">hostname</span><span class="p">)</span>
|
|
<span class="k">if</span> <span class="n">hostname</span> <span class="ow">not</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_hostnames</span><span class="p">():</span>
|
|
<span class="c1"># a new host is being added to the inventory</span>
|
|
<span class="n">changed</span> <span class="o">=</span> <span class="bp">True</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">_hosts</span><span class="p">[</span><span class="n">hostname</span><span class="p">]</span> <span class="o">=</span> <span class="n">host</span>
|
|
|
|
<span class="c1"># a host is to be added to an existing group</span>
|
|
<span class="k">elif</span> <span class="n">groupname</span><span class="p">:</span>
|
|
<span class="n">group</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_groups</span><span class="p">[</span><span class="n">groupname</span><span class="p">]</span>
|
|
<span class="k">if</span> <span class="n">hostname</span> <span class="ow">not</span> <span class="ow">in</span> <span class="n">group</span><span class="o">.</span><span class="n">get_hostnames</span><span class="p">():</span>
|
|
<span class="n">changed</span> <span class="o">=</span> <span class="bp">True</span>
|
|
<span class="n">group</span><span class="o">.</span><span class="n">add_host</span><span class="p">(</span><span class="n">host</span><span class="p">)</span>
|
|
<span class="k">return</span> <span class="n">changed</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Inventory.remove_all_hosts"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Inventory.remove_all_hosts">[docs]</a> <span class="k">def</span> <span class="nf">remove_all_hosts</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
|
|
<span class="sd">"""remove all hosts."""</span>
|
|
<span class="n">hostnamess</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_hostnames</span><span class="p">()</span>
|
|
<span class="k">for</span> <span class="n">hostname</span> <span class="ow">in</span> <span class="n">hostnamess</span><span class="p">:</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">remove_host</span><span class="p">(</span><span class="n">hostname</span><span class="p">)</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Inventory.remove_host"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Inventory.remove_host">[docs]</a> <span class="k">def</span> <span class="nf">remove_host</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">hostname</span><span class="p">,</span> <span class="n">groupname</span><span class="o">=</span><span class="bp">None</span><span class="p">):</span>
|
|
<span class="sd">"""remove host</span>
|
|
|
|
<span class="sd"> if groupname is none, delete host</span>
|
|
<span class="sd"> if group name is not none, remove host from group</span>
|
|
<span class="sd"> """</span>
|
|
<span class="n">changed</span> <span class="o">=</span> <span class="bp">False</span>
|
|
<span class="k">if</span> <span class="n">groupname</span> <span class="ow">and</span> <span class="n">groupname</span> <span class="ow">not</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">_groups</span><span class="p">:</span>
|
|
<span class="k">raise</span> <span class="n">NotInInventory</span><span class="p">(</span><span class="n">u</span><span class="o">.</span><span class="n">_</span><span class="p">(</span><span class="s1">'Group'</span><span class="p">),</span> <span class="n">groupname</span><span class="p">)</span>
|
|
|
|
<span class="k">if</span> <span class="n">hostname</span> <span class="ow">not</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">_hosts</span><span class="p">:</span>
|
|
<span class="k">return</span> <span class="n">changed</span>
|
|
|
|
<span class="n">changed</span> <span class="o">=</span> <span class="bp">True</span>
|
|
<span class="n">host</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_hosts</span><span class="p">[</span><span class="n">hostname</span><span class="p">]</span>
|
|
<span class="n">groups</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_groups</span><span class="p">(</span><span class="n">host</span><span class="p">)</span>
|
|
<span class="k">for</span> <span class="n">group</span> <span class="ow">in</span> <span class="n">groups</span><span class="p">:</span>
|
|
<span class="k">if</span> <span class="ow">not</span> <span class="n">groupname</span> <span class="ow">or</span> <span class="n">groupname</span> <span class="o">==</span> <span class="n">group</span><span class="o">.</span><span class="n">name</span><span class="p">:</span>
|
|
<span class="n">group</span><span class="o">.</span><span class="n">remove_host</span><span class="p">(</span><span class="n">host</span><span class="p">)</span>
|
|
|
|
<span class="n">host_vars</span> <span class="o">=</span> <span class="n">os</span><span class="o">.</span><span class="n">path</span><span class="o">.</span><span class="n">join</span><span class="p">(</span><span class="n">get_host_vars_dir</span><span class="p">(),</span> <span class="n">hostname</span><span class="p">)</span>
|
|
<span class="k">if</span> <span class="n">os</span><span class="o">.</span><span class="n">path</span><span class="o">.</span><span class="n">exists</span><span class="p">(</span><span class="n">host_vars</span><span class="p">):</span>
|
|
<span class="n">os</span><span class="o">.</span><span class="n">remove</span><span class="p">(</span><span class="n">host_vars</span><span class="p">)</span>
|
|
|
|
<span class="k">if</span> <span class="ow">not</span> <span class="n">groupname</span><span class="p">:</span>
|
|
<span class="k">del</span> <span class="bp">self</span><span class="o">.</span><span class="n">_hosts</span><span class="p">[</span><span class="n">hostname</span><span class="p">]</span>
|
|
<span class="k">return</span> <span class="n">changed</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Inventory.setup_hosts"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Inventory.setup_hosts">[docs]</a> <span class="k">def</span> <span class="nf">setup_hosts</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">hosts_info</span><span class="p">):</span>
|
|
<span class="sd">"""setup multiple hosts</span>
|
|
|
|
<span class="sd"> hosts_info is a dict of format:</span>
|
|
<span class="sd"> {'hostname1': {</span>
|
|
<span class="sd"> 'password': password</span>
|
|
<span class="sd"> 'uname': user_name</span>
|
|
<span class="sd"> }</span>
|
|
<span class="sd"> }</span>
|
|
<span class="sd"> The uname entry is optional.</span>
|
|
<span class="sd"> """</span>
|
|
<span class="n">failed_hosts</span> <span class="o">=</span> <span class="p">{}</span>
|
|
<span class="k">for</span> <span class="n">hostname</span><span class="p">,</span> <span class="n">host_info</span> <span class="ow">in</span> <span class="n">hosts_info</span><span class="o">.</span><span class="n">items</span><span class="p">():</span>
|
|
<span class="n">host</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_host</span><span class="p">(</span><span class="n">hostname</span><span class="p">)</span>
|
|
<span class="k">if</span> <span class="ow">not</span> <span class="n">host</span><span class="p">:</span>
|
|
<span class="n">failed_hosts</span><span class="p">[</span><span class="n">hostname</span><span class="p">]</span> <span class="o">=</span> <span class="n">u</span><span class="o">.</span><span class="n">_</span><span class="p">(</span><span class="s2">"Host doesn't exist."</span><span class="p">)</span>
|
|
<span class="k">continue</span>
|
|
<span class="k">if</span> <span class="ow">not</span> <span class="n">host_info</span> <span class="ow">or</span> <span class="s1">'password'</span> <span class="ow">not</span> <span class="ow">in</span> <span class="n">host_info</span><span class="p">:</span>
|
|
<span class="n">failed_hosts</span><span class="p">[</span><span class="n">hostname</span><span class="p">]</span> <span class="o">=</span> <span class="n">u</span><span class="o">.</span><span class="n">_</span><span class="p">(</span><span class="s1">'No password in yml file.'</span><span class="p">)</span>
|
|
<span class="k">continue</span>
|
|
<span class="n">passwd</span> <span class="o">=</span> <span class="n">host_info</span><span class="p">[</span><span class="s1">'password'</span><span class="p">]</span>
|
|
<span class="n">uname</span> <span class="o">=</span> <span class="bp">None</span>
|
|
<span class="k">if</span> <span class="s1">'uname'</span> <span class="ow">in</span> <span class="n">host_info</span><span class="p">:</span>
|
|
<span class="n">uname</span> <span class="o">=</span> <span class="n">host_info</span><span class="p">[</span><span class="s1">'uname'</span><span class="p">]</span>
|
|
<span class="k">try</span><span class="p">:</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">setup_host</span><span class="p">(</span><span class="n">hostname</span><span class="p">,</span> <span class="n">passwd</span><span class="p">,</span> <span class="n">uname</span><span class="p">)</span>
|
|
<span class="k">except</span> <span class="ne">Exception</span> <span class="k">as</span> <span class="n">e</span><span class="p">:</span>
|
|
<span class="n">failed_hosts</span><span class="p">[</span><span class="n">hostname</span><span class="p">]</span> <span class="o">=</span> <span class="s1">'</span><span class="si">%s</span><span class="s1">'</span> <span class="o">%</span> <span class="n">e</span>
|
|
<span class="k">if</span> <span class="n">failed_hosts</span><span class="p">:</span>
|
|
<span class="n">summary</span> <span class="o">=</span> <span class="s1">'</span><span class="se">\n</span><span class="s1">'</span>
|
|
<span class="k">for</span> <span class="n">hostname</span><span class="p">,</span> <span class="n">err</span> <span class="ow">in</span> <span class="n">failed_hosts</span><span class="o">.</span><span class="n">items</span><span class="p">():</span>
|
|
<span class="n">summary</span> <span class="o">=</span> <span class="n">summary</span> <span class="o">+</span> <span class="s1">'- </span><span class="si">%s</span><span class="s1">: </span><span class="si">%s</span><span class="se">\n</span><span class="s1">'</span> <span class="o">%</span> <span class="p">(</span><span class="n">hostname</span><span class="p">,</span> <span class="n">err</span><span class="p">)</span>
|
|
<span class="k">raise</span> <span class="n">HostError</span><span class="p">(</span>
|
|
<span class="n">u</span><span class="o">.</span><span class="n">_</span><span class="p">(</span><span class="s1">'Not all hosts were set up. : {reasons}'</span><span class="p">)</span>
|
|
<span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="n">reasons</span><span class="o">=</span><span class="n">summary</span><span class="p">))</span>
|
|
<span class="k">else</span><span class="p">:</span>
|
|
<span class="n">LOG</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="n">u</span><span class="o">.</span><span class="n">_LI</span><span class="p">(</span><span class="s1">'All hosts were successfully set up.'</span><span class="p">))</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Inventory.setup_host"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Inventory.setup_host">[docs]</a> <span class="k">def</span> <span class="nf">setup_host</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">hostname</span><span class="p">,</span> <span class="n">password</span><span class="p">,</span> <span class="n">uname</span><span class="o">=</span><span class="bp">None</span><span class="p">):</span>
|
|
<span class="k">try</span><span class="p">:</span>
|
|
<span class="n">LOG</span><span class="o">.</span><span class="n">info</span><span class="p">(</span>
|
|
<span class="n">u</span><span class="o">.</span><span class="n">_LI</span><span class="p">(</span><span class="s1">'Starting setup of host ({host}).'</span><span class="p">)</span>
|
|
<span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="n">host</span><span class="o">=</span><span class="n">hostname</span><span class="p">))</span>
|
|
<span class="n">ssh_setup_host</span><span class="p">(</span><span class="n">hostname</span><span class="p">,</span> <span class="n">password</span><span class="p">,</span> <span class="n">uname</span><span class="p">)</span>
|
|
<span class="n">check_ok</span><span class="p">,</span> <span class="n">msg</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">ssh_check_host</span><span class="p">(</span><span class="n">hostname</span><span class="p">)</span>
|
|
<span class="k">if</span> <span class="ow">not</span> <span class="n">check_ok</span><span class="p">:</span>
|
|
<span class="k">raise</span> <span class="ne">Exception</span><span class="p">(</span><span class="n">u</span><span class="o">.</span><span class="n">_</span><span class="p">(</span><span class="s1">'Post-setup ssh check failed. {err}'</span><span class="p">)</span>
|
|
<span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="n">err</span><span class="o">=</span><span class="n">msg</span><span class="p">))</span>
|
|
<span class="n">LOG</span><span class="o">.</span><span class="n">info</span><span class="p">(</span><span class="n">u</span><span class="o">.</span><span class="n">_LI</span><span class="p">(</span><span class="s1">'Host ({host}) setup succeeded.'</span><span class="p">)</span>
|
|
<span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="n">host</span><span class="o">=</span><span class="n">hostname</span><span class="p">))</span>
|
|
<span class="k">except</span> <span class="ne">Exception</span> <span class="k">as</span> <span class="n">e</span><span class="p">:</span>
|
|
<span class="k">raise</span> <span class="n">HostError</span><span class="p">(</span>
|
|
<span class="n">u</span><span class="o">.</span><span class="n">_</span><span class="p">(</span><span class="s1">'Host ({host}) setup failed : {error}'</span><span class="p">)</span>
|
|
<span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="n">host</span><span class="o">=</span><span class="n">hostname</span><span class="p">,</span> <span class="n">error</span><span class="o">=</span><span class="nb">str</span><span class="p">(</span><span class="n">e</span><span class="p">)))</span>
|
|
<span class="k">return</span> <span class="bp">True</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Inventory.ssh_check_hosts"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Inventory.ssh_check_hosts">[docs]</a> <span class="k">def</span> <span class="nf">ssh_check_hosts</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">hostnames</span><span class="p">):</span>
|
|
<span class="sd">"""ssh check for hosts</span>
|
|
|
|
<span class="sd"> return {hostname: {'success': True|False,</span>
|
|
<span class="sd"> 'msg': message}}</span>
|
|
<span class="sd"> """</span>
|
|
<span class="n">summary</span> <span class="o">=</span> <span class="p">{}</span>
|
|
<span class="k">for</span> <span class="n">hostname</span> <span class="ow">in</span> <span class="n">hostnames</span><span class="p">:</span>
|
|
<span class="n">is_ok</span><span class="p">,</span> <span class="n">msg</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">ssh_check_host</span><span class="p">(</span><span class="n">hostname</span><span class="p">)</span>
|
|
<span class="n">summary</span><span class="p">[</span><span class="n">hostname</span><span class="p">]</span> <span class="o">=</span> <span class="p">{}</span>
|
|
<span class="n">summary</span><span class="p">[</span><span class="n">hostname</span><span class="p">][</span><span class="s1">'success'</span><span class="p">]</span> <span class="o">=</span> <span class="n">is_ok</span>
|
|
<span class="n">summary</span><span class="p">[</span><span class="n">hostname</span><span class="p">][</span><span class="s1">'msg'</span><span class="p">]</span> <span class="o">=</span> <span class="n">msg</span>
|
|
<span class="k">return</span> <span class="n">summary</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Inventory.ssh_check_host"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Inventory.ssh_check_host">[docs]</a> <span class="k">def</span> <span class="nf">ssh_check_host</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">hostname</span><span class="p">):</span>
|
|
<span class="n">err_msg</span><span class="p">,</span> <span class="n">output</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">run_ansible_command</span><span class="p">(</span><span class="s1">'-m ping'</span><span class="p">,</span> <span class="n">hostname</span><span class="p">)</span>
|
|
<span class="n">is_ok</span> <span class="o">=</span> <span class="bp">True</span>
|
|
<span class="k">if</span> <span class="n">err_msg</span><span class="p">:</span>
|
|
<span class="n">is_ok</span> <span class="o">=</span> <span class="bp">False</span>
|
|
<span class="n">msg</span> <span class="o">=</span> <span class="p">(</span>
|
|
<span class="n">u</span><span class="o">.</span><span class="n">_</span><span class="p">(</span><span class="s1">'Host ({host}) ssh check failed. : {error} {message}'</span><span class="p">)</span>
|
|
<span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="n">host</span><span class="o">=</span><span class="n">hostname</span><span class="p">,</span> <span class="n">error</span><span class="o">=</span><span class="n">err_msg</span><span class="p">,</span> <span class="n">message</span><span class="o">=</span><span class="n">output</span><span class="p">))</span>
|
|
<span class="k">else</span><span class="p">:</span>
|
|
<span class="n">msg</span> <span class="o">=</span> <span class="p">(</span><span class="n">u</span><span class="o">.</span><span class="n">_LI</span><span class="p">(</span><span class="s1">'Host ({host}) ssh check succeeded.'</span><span class="p">)</span>
|
|
<span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="n">host</span><span class="o">=</span><span class="n">hostname</span><span class="p">))</span>
|
|
<span class="k">return</span> <span class="n">is_ok</span><span class="p">,</span> <span class="n">msg</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Inventory.run_ansible_command"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Inventory.run_ansible_command">[docs]</a> <span class="k">def</span> <span class="nf">run_ansible_command</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">ansible_command</span><span class="p">,</span> <span class="n">hostname</span><span class="p">):</span>
|
|
<span class="n">err_msg</span> <span class="o">=</span> <span class="bp">None</span>
|
|
<span class="n">command_string</span> <span class="o">=</span> <span class="s1">'/usr/bin/sudo -u </span><span class="si">%s</span><span class="s1"> </span><span class="si">%s</span><span class="s1"> -vvv'</span> <span class="o">%</span> \
|
|
<span class="p">(</span><span class="n">get_admin_user</span><span class="p">(),</span> <span class="n">get_ansible_command</span><span class="p">())</span>
|
|
<span class="n">gen_file_path</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">create_json_gen_file</span><span class="p">()</span>
|
|
<span class="n">cmd</span> <span class="o">=</span> <span class="s1">'</span><span class="si">%s</span><span class="s1"> </span><span class="si">%s</span><span class="s1"> -i </span><span class="si">%s</span><span class="s1"> </span><span class="si">%s</span><span class="s1">'</span> <span class="o">%</span> <span class="p">(</span><span class="n">command_string</span><span class="p">,</span> <span class="n">hostname</span><span class="p">,</span> <span class="n">gen_file_path</span><span class="p">,</span>
|
|
<span class="n">ansible_command</span><span class="p">)</span>
|
|
<span class="k">try</span><span class="p">:</span>
|
|
<span class="n">err_msg</span><span class="p">,</span> <span class="n">output</span> <span class="o">=</span> <span class="n">run_cmd</span><span class="p">(</span><span class="n">cmd</span><span class="p">,</span> <span class="bp">False</span><span class="p">)</span>
|
|
<span class="k">except</span> <span class="ne">Exception</span> <span class="k">as</span> <span class="n">e</span><span class="p">:</span>
|
|
<span class="n">err_msg</span> <span class="o">=</span> <span class="nb">str</span><span class="p">(</span><span class="n">e</span><span class="p">)</span>
|
|
<span class="k">finally</span><span class="p">:</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">remove_json_gen_file</span><span class="p">(</span><span class="n">gen_file_path</span><span class="p">)</span>
|
|
<span class="k">return</span> <span class="n">err_msg</span><span class="p">,</span> <span class="n">output</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Inventory.add_group"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Inventory.add_group">[docs]</a> <span class="k">def</span> <span class="nf">add_group</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">groupname</span><span class="p">):</span>
|
|
|
|
<span class="c1"># Group names cannot overlap with service names:</span>
|
|
<span class="k">if</span> <span class="n">groupname</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">_services</span> <span class="ow">or</span> <span class="n">groupname</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">_sub_services</span><span class="p">:</span>
|
|
<span class="k">raise</span> <span class="n">InvalidArgument</span><span class="p">(</span>
|
|
<span class="n">u</span><span class="o">.</span><span class="n">_</span><span class="p">(</span><span class="s1">'Invalid group name. A service name '</span>
|
|
<span class="s1">'cannot be used for a group name.'</span><span class="p">))</span>
|
|
|
|
<span class="k">if</span> <span class="n">groupname</span> <span class="ow">not</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">_groups</span><span class="p">:</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">_groups</span><span class="p">[</span><span class="n">groupname</span><span class="p">]</span> <span class="o">=</span> <span class="n">HostGroup</span><span class="p">(</span><span class="n">groupname</span><span class="p">)</span>
|
|
|
|
<span class="n">group</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_groups</span><span class="p">[</span><span class="n">groupname</span><span class="p">]</span>
|
|
|
|
<span class="n">group</span><span class="o">.</span><span class="n">set_remote</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">remote_mode</span><span class="p">)</span>
|
|
|
|
<span class="k">return</span> <span class="n">group</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Inventory.remove_group"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Inventory.remove_group">[docs]</a> <span class="k">def</span> <span class="nf">remove_group</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">groupname</span><span class="p">):</span>
|
|
<span class="k">if</span> <span class="n">groupname</span> <span class="ow">in</span> <span class="n">PROTECTED_GROUPS</span><span class="p">:</span>
|
|
<span class="k">raise</span> <span class="n">InvalidArgument</span><span class="p">(</span>
|
|
<span class="n">u</span><span class="o">.</span><span class="n">_</span><span class="p">(</span><span class="s1">'Cannot remove {group} group. It is required by kolla.'</span><span class="p">)</span>
|
|
<span class="o">.</span><span class="n">format</span><span class="p">(</span><span class="n">group</span><span class="o">=</span><span class="n">groupname</span><span class="p">))</span>
|
|
|
|
<span class="c1"># remove group from services & subservices</span>
|
|
<span class="k">for</span> <span class="n">service</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">_services</span><span class="o">.</span><span class="n">values</span><span class="p">():</span>
|
|
<span class="n">service</span><span class="o">.</span><span class="n">remove_groupname</span><span class="p">(</span><span class="n">groupname</span><span class="p">)</span>
|
|
|
|
<span class="k">for</span> <span class="n">subservice</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">_sub_services</span><span class="o">.</span><span class="n">values</span><span class="p">():</span>
|
|
<span class="n">subservice</span><span class="o">.</span><span class="n">remove_groupname</span><span class="p">(</span><span class="n">groupname</span><span class="p">)</span>
|
|
|
|
<span class="n">group_vars</span> <span class="o">=</span> <span class="n">os</span><span class="o">.</span><span class="n">path</span><span class="o">.</span><span class="n">join</span><span class="p">(</span><span class="n">get_group_vars_dir</span><span class="p">(),</span> <span class="n">groupname</span><span class="p">)</span>
|
|
<span class="k">if</span> <span class="n">os</span><span class="o">.</span><span class="n">path</span><span class="o">.</span><span class="n">exists</span><span class="p">(</span><span class="n">group_vars</span><span class="p">)</span> <span class="ow">and</span> <span class="n">groupname</span> <span class="o">!=</span> <span class="s1">'__GLOBAL__'</span><span class="p">:</span>
|
|
<span class="n">os</span><span class="o">.</span><span class="n">remove</span><span class="p">(</span><span class="n">group_vars</span><span class="p">)</span>
|
|
|
|
<span class="k">if</span> <span class="n">groupname</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">_groups</span><span class="p">:</span>
|
|
<span class="k">del</span> <span class="bp">self</span><span class="o">.</span><span class="n">_groups</span><span class="p">[</span><span class="n">groupname</span><span class="p">]</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Inventory.get_group"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Inventory.get_group">[docs]</a> <span class="k">def</span> <span class="nf">get_group</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">groupname</span><span class="p">):</span>
|
|
<span class="n">group</span> <span class="o">=</span> <span class="bp">None</span>
|
|
<span class="k">if</span> <span class="n">groupname</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">_groups</span><span class="p">:</span>
|
|
<span class="n">group</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_groups</span><span class="p">[</span><span class="n">groupname</span><span class="p">]</span>
|
|
<span class="k">return</span> <span class="n">group</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Inventory.get_groupnames"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Inventory.get_groupnames">[docs]</a> <span class="k">def</span> <span class="nf">get_groupnames</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
|
|
<span class="k">return</span> <span class="nb">list</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">_groups</span><span class="o">.</span><span class="n">keys</span><span class="p">())</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Inventory.get_groups"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Inventory.get_groups">[docs]</a> <span class="k">def</span> <span class="nf">get_groups</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">host</span><span class="o">=</span><span class="bp">None</span><span class="p">):</span>
|
|
<span class="sd">"""return all groups containing host</span>
|
|
|
|
<span class="sd"> if hosts is none, return all groups in inventory</span>
|
|
<span class="sd"> """</span>
|
|
<span class="n">groups</span> <span class="o">=</span> <span class="p">[]</span>
|
|
<span class="k">if</span> <span class="ow">not</span> <span class="n">host</span><span class="p">:</span>
|
|
<span class="n">groups</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_groups</span><span class="o">.</span><span class="n">values</span><span class="p">()</span>
|
|
|
|
<span class="k">else</span><span class="p">:</span>
|
|
<span class="k">for</span> <span class="n">group</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">_groups</span><span class="o">.</span><span class="n">values</span><span class="p">():</span>
|
|
<span class="k">if</span> <span class="n">host</span><span class="o">.</span><span class="n">name</span> <span class="ow">in</span> <span class="n">group</span><span class="o">.</span><span class="n">get_hostnames</span><span class="p">():</span>
|
|
<span class="n">groups</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">group</span><span class="p">)</span>
|
|
<span class="k">return</span> <span class="n">groups</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Inventory.get_host_groups"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Inventory.get_host_groups">[docs]</a> <span class="k">def</span> <span class="nf">get_host_groups</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
|
|
<span class="sd">"""return { hostname : [groupnames] }"""</span>
|
|
|
|
<span class="n">host_groups</span> <span class="o">=</span> <span class="p">{}</span>
|
|
<span class="k">for</span> <span class="n">host</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">_hosts</span><span class="o">.</span><span class="n">values</span><span class="p">():</span>
|
|
<span class="n">host_groups</span><span class="p">[</span><span class="n">host</span><span class="o">.</span><span class="n">name</span><span class="p">]</span> <span class="o">=</span> <span class="p">[]</span>
|
|
<span class="n">groups</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_groups</span><span class="p">(</span><span class="n">host</span><span class="p">)</span>
|
|
<span class="k">for</span> <span class="n">group</span> <span class="ow">in</span> <span class="n">groups</span><span class="p">:</span>
|
|
<span class="n">host_groups</span><span class="p">[</span><span class="n">host</span><span class="o">.</span><span class="n">name</span><span class="p">]</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">group</span><span class="o">.</span><span class="n">name</span><span class="p">)</span>
|
|
<span class="k">return</span> <span class="n">host_groups</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Inventory.get_group_services"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Inventory.get_group_services">[docs]</a> <span class="k">def</span> <span class="nf">get_group_services</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
|
|
<span class="sd">"""get groups and their services</span>
|
|
|
|
<span class="sd"> return { groupname: [servicenames] }</span>
|
|
<span class="sd"> """</span>
|
|
|
|
<span class="n">group_services</span> <span class="o">=</span> <span class="p">{}</span>
|
|
|
|
<span class="k">for</span> <span class="n">group</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_groups</span><span class="p">():</span>
|
|
<span class="n">group_services</span><span class="p">[</span><span class="n">group</span><span class="o">.</span><span class="n">name</span><span class="p">]</span> <span class="o">=</span> <span class="p">[]</span>
|
|
|
|
<span class="k">for</span> <span class="n">svc</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_services</span><span class="p">():</span>
|
|
<span class="k">for</span> <span class="n">groupname</span> <span class="ow">in</span> <span class="n">svc</span><span class="o">.</span><span class="n">get_groupnames</span><span class="p">():</span>
|
|
<span class="n">group_services</span><span class="p">[</span><span class="n">groupname</span><span class="p">]</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">svc</span><span class="o">.</span><span class="n">name</span><span class="p">)</span>
|
|
<span class="k">for</span> <span class="n">sub_svc</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_sub_services</span><span class="p">():</span>
|
|
<span class="k">for</span> <span class="n">groupname</span> <span class="ow">in</span> <span class="n">sub_svc</span><span class="o">.</span><span class="n">get_groupnames</span><span class="p">():</span>
|
|
<span class="n">group_services</span><span class="p">[</span><span class="n">groupname</span><span class="p">]</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">sub_svc</span><span class="o">.</span><span class="n">name</span><span class="p">)</span>
|
|
<span class="k">return</span> <span class="n">group_services</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Inventory.get_group_hosts"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Inventory.get_group_hosts">[docs]</a> <span class="k">def</span> <span class="nf">get_group_hosts</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
|
|
<span class="sd">"""return { groupname : [hostnames] }"""</span>
|
|
<span class="n">group_hosts</span> <span class="o">=</span> <span class="p">{}</span>
|
|
<span class="k">for</span> <span class="n">group</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_groups</span><span class="p">():</span>
|
|
<span class="n">group_hosts</span><span class="p">[</span><span class="n">group</span><span class="o">.</span><span class="n">name</span><span class="p">]</span> <span class="o">=</span> <span class="p">[]</span>
|
|
<span class="k">for</span> <span class="n">hostname</span> <span class="ow">in</span> <span class="n">group</span><span class="o">.</span><span class="n">get_hostnames</span><span class="p">():</span>
|
|
<span class="n">group_hosts</span><span class="p">[</span><span class="n">group</span><span class="o">.</span><span class="n">name</span><span class="p">]</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">hostname</span><span class="p">)</span>
|
|
<span class="k">return</span> <span class="n">group_hosts</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Inventory.create_service"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Inventory.create_service">[docs]</a> <span class="k">def</span> <span class="nf">create_service</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">servicename</span><span class="p">):</span>
|
|
<span class="k">if</span> <span class="n">servicename</span> <span class="ow">not</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">_services</span><span class="p">:</span>
|
|
<span class="n">service</span> <span class="o">=</span> <span class="n">Service</span><span class="p">(</span><span class="n">servicename</span><span class="p">)</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">_services</span><span class="p">[</span><span class="n">servicename</span><span class="p">]</span> <span class="o">=</span> <span class="n">service</span>
|
|
<span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_services</span><span class="p">[</span><span class="n">servicename</span><span class="p">]</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Inventory.delete_service"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Inventory.delete_service">[docs]</a> <span class="k">def</span> <span class="nf">delete_service</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">servicename</span><span class="p">):</span>
|
|
<span class="k">if</span> <span class="n">servicename</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">_services</span><span class="p">:</span>
|
|
<span class="k">del</span> <span class="bp">self</span><span class="o">.</span><span class="n">_services</span><span class="p">[</span><span class="n">servicename</span><span class="p">]</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Inventory.get_services"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Inventory.get_services">[docs]</a> <span class="k">def</span> <span class="nf">get_services</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
|
|
<span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_services</span><span class="o">.</span><span class="n">values</span><span class="p">()</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Inventory.get_service"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Inventory.get_service">[docs]</a> <span class="k">def</span> <span class="nf">get_service</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">servicename</span><span class="p">):</span>
|
|
<span class="n">service</span> <span class="o">=</span> <span class="bp">None</span>
|
|
<span class="k">if</span> <span class="n">servicename</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">_services</span><span class="p">:</span>
|
|
<span class="n">service</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_services</span><span class="p">[</span><span class="n">servicename</span><span class="p">]</span>
|
|
<span class="k">return</span> <span class="n">service</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Inventory.add_group_to_service"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Inventory.add_group_to_service">[docs]</a> <span class="k">def</span> <span class="nf">add_group_to_service</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">groupname</span><span class="p">,</span> <span class="n">servicename</span><span class="p">):</span>
|
|
<span class="k">if</span> <span class="n">groupname</span> <span class="ow">not</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">_groups</span><span class="p">:</span>
|
|
<span class="k">raise</span> <span class="n">NotInInventory</span><span class="p">(</span><span class="n">u</span><span class="o">.</span><span class="n">_</span><span class="p">(</span><span class="s1">'Group'</span><span class="p">),</span> <span class="n">groupname</span><span class="p">)</span>
|
|
<span class="k">if</span> <span class="n">servicename</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">_services</span><span class="p">:</span>
|
|
<span class="n">service</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_service</span><span class="p">(</span><span class="n">servicename</span><span class="p">)</span>
|
|
<span class="n">service</span><span class="o">.</span><span class="n">add_groupname</span><span class="p">(</span><span class="n">groupname</span><span class="p">)</span>
|
|
<span class="k">elif</span> <span class="n">servicename</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">_sub_services</span><span class="p">:</span>
|
|
<span class="n">sub_service</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_sub_service</span><span class="p">(</span><span class="n">servicename</span><span class="p">)</span>
|
|
<span class="n">sub_service</span><span class="o">.</span><span class="n">add_groupname</span><span class="p">(</span><span class="n">groupname</span><span class="p">)</span>
|
|
<span class="k">else</span><span class="p">:</span>
|
|
<span class="k">raise</span> <span class="n">NotInInventory</span><span class="p">(</span><span class="n">u</span><span class="o">.</span><span class="n">_</span><span class="p">(</span><span class="s1">'Service'</span><span class="p">),</span> <span class="n">servicename</span><span class="p">)</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Inventory.remove_group_from_service"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Inventory.remove_group_from_service">[docs]</a> <span class="k">def</span> <span class="nf">remove_group_from_service</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">groupname</span><span class="p">,</span> <span class="n">servicename</span><span class="p">):</span>
|
|
<span class="k">if</span> <span class="n">groupname</span> <span class="ow">not</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">_groups</span><span class="p">:</span>
|
|
<span class="k">raise</span> <span class="n">NotInInventory</span><span class="p">(</span><span class="n">u</span><span class="o">.</span><span class="n">_</span><span class="p">(</span><span class="s1">'Group'</span><span class="p">),</span> <span class="n">groupname</span><span class="p">)</span>
|
|
<span class="k">if</span> <span class="n">servicename</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">_services</span><span class="p">:</span>
|
|
<span class="n">service</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_service</span><span class="p">(</span><span class="n">servicename</span><span class="p">)</span>
|
|
<span class="n">service</span><span class="o">.</span><span class="n">remove_groupname</span><span class="p">(</span><span class="n">groupname</span><span class="p">)</span>
|
|
<span class="k">elif</span> <span class="n">servicename</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">_sub_services</span><span class="p">:</span>
|
|
<span class="n">sub_service</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_sub_service</span><span class="p">(</span><span class="n">servicename</span><span class="p">)</span>
|
|
<span class="n">sub_service</span><span class="o">.</span><span class="n">remove_groupname</span><span class="p">(</span><span class="n">groupname</span><span class="p">)</span>
|
|
<span class="k">else</span><span class="p">:</span>
|
|
<span class="k">raise</span> <span class="n">NotInInventory</span><span class="p">(</span><span class="n">u</span><span class="o">.</span><span class="n">_</span><span class="p">(</span><span class="s1">'Service'</span><span class="p">),</span> <span class="n">servicename</span><span class="p">)</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Inventory.create_sub_service"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Inventory.create_sub_service">[docs]</a> <span class="k">def</span> <span class="nf">create_sub_service</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">sub_servicename</span><span class="p">):</span>
|
|
<span class="k">if</span> <span class="n">sub_servicename</span> <span class="ow">not</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">_sub_services</span><span class="p">:</span>
|
|
<span class="n">sub_service</span> <span class="o">=</span> <span class="n">SubService</span><span class="p">(</span><span class="n">sub_servicename</span><span class="p">)</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">_sub_services</span><span class="p">[</span><span class="n">sub_servicename</span><span class="p">]</span> <span class="o">=</span> <span class="n">sub_service</span>
|
|
<span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_sub_services</span><span class="p">[</span><span class="n">sub_servicename</span><span class="p">]</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Inventory.delete_sub_service"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Inventory.delete_sub_service">[docs]</a> <span class="k">def</span> <span class="nf">delete_sub_service</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">sub_servicename</span><span class="p">):</span>
|
|
<span class="k">if</span> <span class="n">sub_servicename</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">_sub_services</span><span class="p">:</span>
|
|
<span class="k">del</span> <span class="bp">self</span><span class="o">.</span><span class="n">_sub_services</span><span class="p">[</span><span class="n">sub_servicename</span><span class="p">]</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Inventory.get_sub_services"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Inventory.get_sub_services">[docs]</a> <span class="k">def</span> <span class="nf">get_sub_services</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
|
|
<span class="k">return</span> <span class="bp">self</span><span class="o">.</span><span class="n">_sub_services</span><span class="o">.</span><span class="n">values</span><span class="p">()</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Inventory.get_sub_service"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Inventory.get_sub_service">[docs]</a> <span class="k">def</span> <span class="nf">get_sub_service</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">sub_servicename</span><span class="p">):</span>
|
|
<span class="n">sub_service</span> <span class="o">=</span> <span class="bp">None</span>
|
|
<span class="k">if</span> <span class="n">sub_servicename</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">_sub_services</span><span class="p">:</span>
|
|
<span class="n">sub_service</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">_sub_services</span><span class="p">[</span><span class="n">sub_servicename</span><span class="p">]</span>
|
|
<span class="k">return</span> <span class="n">sub_service</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Inventory.get_service_sub_services"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Inventory.get_service_sub_services">[docs]</a> <span class="k">def</span> <span class="nf">get_service_sub_services</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
|
|
<span class="sd">"""get services and their sub_services</span>
|
|
|
|
<span class="sd"> return { servicename: [sub_servicenames] }</span>
|
|
<span class="sd"> """</span>
|
|
<span class="n">svc_sub_svcs</span> <span class="o">=</span> <span class="p">{}</span>
|
|
<span class="k">for</span> <span class="n">service</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_services</span><span class="p">():</span>
|
|
<span class="n">svc_sub_svcs</span><span class="p">[</span><span class="n">service</span><span class="o">.</span><span class="n">name</span><span class="p">]</span> <span class="o">=</span> <span class="p">[]</span>
|
|
<span class="n">svc_sub_svcs</span><span class="p">[</span><span class="n">service</span><span class="o">.</span><span class="n">name</span><span class="p">]</span><span class="o">.</span><span class="n">extend</span><span class="p">(</span><span class="n">service</span><span class="o">.</span><span class="n">get_sub_servicenames</span><span class="p">())</span>
|
|
<span class="k">return</span> <span class="n">svc_sub_svcs</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Inventory.set_deploy_mode"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Inventory.set_deploy_mode">[docs]</a> <span class="k">def</span> <span class="nf">set_deploy_mode</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">remote_flag</span><span class="p">):</span>
|
|
<span class="k">if</span> <span class="ow">not</span> <span class="n">remote_flag</span> <span class="ow">and</span> <span class="nb">len</span><span class="p">(</span><span class="bp">self</span><span class="o">.</span><span class="n">_hosts</span><span class="p">)</span> <span class="o">></span> <span class="mi">1</span><span class="p">:</span>
|
|
<span class="k">raise</span> <span class="n">InvalidConfiguration</span><span class="p">(</span>
|
|
<span class="n">u</span><span class="o">.</span><span class="n">_</span><span class="p">(</span><span class="s1">'Cannot set local deploy mode when multiple hosts exist.'</span><span class="p">))</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">remote_mode</span> <span class="o">=</span> <span class="n">remote_flag</span>
|
|
|
|
<span class="k">for</span> <span class="n">group</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_groups</span><span class="p">():</span>
|
|
<span class="n">group</span><span class="o">.</span><span class="n">set_remote</span><span class="p">(</span><span class="n">remote_flag</span><span class="p">)</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Inventory.get_ansible_json"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Inventory.get_ansible_json">[docs]</a> <span class="k">def</span> <span class="nf">get_ansible_json</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">inventory_filter</span><span class="o">=</span><span class="bp">None</span><span class="p">):</span>
|
|
<span class="sd">"""generate json inventory for ansible</span>
|
|
|
|
<span class="sd"> The hosts and groups added to the json output for ansible will be</span>
|
|
<span class="sd"> filtered by the hostnames and groupnames in the deploy filters.</span>
|
|
<span class="sd"> This allows a more targeted deploy to a specific set of hosts or</span>
|
|
<span class="sd"> groups.</span>
|
|
|
|
<span class="sd"> typical ansible json format:</span>
|
|
<span class="sd"> {</span>
|
|
<span class="sd"> 'group': {</span>
|
|
<span class="sd"> 'hosts': [</span>
|
|
<span class="sd"> '192.168.28.71',</span>
|
|
<span class="sd"> '192.168.28.72'</span>
|
|
<span class="sd"> ],</span>
|
|
<span class="sd"> 'vars': {</span>
|
|
<span class="sd"> 'ansible_ssh_user': 'johndoe',</span>
|
|
<span class="sd"> 'ansible_ssh_private_key_file': '~/.ssh/mykey',</span>
|
|
<span class="sd"> 'example_variable': 'value'</span>
|
|
<span class="sd"> }</span>
|
|
<span class="sd"> 'children': [ 'marietta', '5points' ]</span>
|
|
<span class="sd"> },</span>
|
|
<span class="sd"> '_meta': {</span>
|
|
<span class="sd"> 'hostvars': {</span>
|
|
<span class="sd"> '192.168.28.71': {</span>
|
|
<span class="sd"> 'host_specific_var': 'bar'</span>
|
|
<span class="sd"> },</span>
|
|
<span class="sd"> '192.168.28.72': {</span>
|
|
<span class="sd"> 'host_specific_var': 'foo'</span>
|
|
<span class="sd"> }</span>
|
|
<span class="sd"> }</span>
|
|
<span class="sd"> }</span>
|
|
<span class="sd"> }</span>
|
|
<span class="sd"> """</span>
|
|
<span class="n">jdict</span> <span class="o">=</span> <span class="p">{}</span>
|
|
|
|
<span class="c1"># if no filter provided, use all groups, all hosts</span>
|
|
<span class="n">deploy_hostnames</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_hostnames</span><span class="p">()</span>
|
|
<span class="n">deploy_groupnames</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_groupnames</span><span class="p">()</span>
|
|
<span class="k">if</span> <span class="n">inventory_filter</span><span class="p">:</span>
|
|
<span class="k">if</span> <span class="s1">'deploy_hosts'</span> <span class="ow">in</span> <span class="n">inventory_filter</span><span class="p">:</span>
|
|
<span class="n">deploy_hostnames</span> <span class="o">=</span> <span class="n">inventory_filter</span><span class="p">[</span><span class="s1">'deploy_hosts'</span><span class="p">]</span>
|
|
<span class="k">if</span> <span class="s1">'deploy_groups'</span> <span class="ow">in</span> <span class="n">inventory_filter</span><span class="p">:</span>
|
|
<span class="n">deploy_groupnames</span> <span class="o">=</span> <span class="n">inventory_filter</span><span class="p">[</span><span class="s1">'deploy_groups'</span><span class="p">]</span>
|
|
|
|
<span class="c1"># add hostgroups</span>
|
|
<span class="k">for</span> <span class="n">group</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_groups</span><span class="p">():</span>
|
|
<span class="n">jdict</span><span class="p">[</span><span class="n">group</span><span class="o">.</span><span class="n">name</span><span class="p">]</span> <span class="o">=</span> <span class="p">{}</span>
|
|
<span class="n">jdict</span><span class="p">[</span><span class="n">group</span><span class="o">.</span><span class="n">name</span><span class="p">][</span><span class="s1">'hosts'</span><span class="p">]</span> <span class="o">=</span> <span class="p">[]</span>
|
|
|
|
<span class="k">if</span> <span class="n">group</span><span class="o">.</span><span class="n">name</span> <span class="ow">in</span> <span class="n">deploy_groupnames</span><span class="p">:</span>
|
|
<span class="n">jdict</span><span class="p">[</span><span class="n">group</span><span class="o">.</span><span class="n">name</span><span class="p">][</span><span class="s1">'hosts'</span><span class="p">]</span> <span class="o">=</span> \
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">_filter_hosts</span><span class="p">(</span><span class="n">group</span><span class="o">.</span><span class="n">get_hostnames</span><span class="p">(),</span> <span class="n">deploy_hostnames</span><span class="p">)</span>
|
|
<span class="n">jdict</span><span class="p">[</span><span class="n">group</span><span class="o">.</span><span class="n">name</span><span class="p">][</span><span class="s1">'children'</span><span class="p">]</span> <span class="o">=</span> <span class="p">[]</span>
|
|
<span class="n">jdict</span><span class="p">[</span><span class="n">group</span><span class="o">.</span><span class="n">name</span><span class="p">][</span><span class="s1">'vars'</span><span class="p">]</span> <span class="o">=</span> <span class="n">group</span><span class="o">.</span><span class="n">get_vars</span><span class="p">()</span>
|
|
|
|
<span class="c1"># add top-level services and what groups they are in</span>
|
|
<span class="k">for</span> <span class="n">service</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_services</span><span class="p">():</span>
|
|
<span class="n">jdict</span><span class="p">[</span><span class="n">service</span><span class="o">.</span><span class="n">name</span><span class="p">]</span> <span class="o">=</span> <span class="p">{}</span>
|
|
<span class="n">jdict</span><span class="p">[</span><span class="n">service</span><span class="o">.</span><span class="n">name</span><span class="p">][</span><span class="s1">'children'</span><span class="p">]</span> <span class="o">=</span> <span class="n">service</span><span class="o">.</span><span class="n">get_groupnames</span><span class="p">()</span>
|
|
|
|
<span class="c1"># add sub-services and their groups</span>
|
|
<span class="k">for</span> <span class="n">sub_svc</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_sub_services</span><span class="p">():</span>
|
|
<span class="n">jdict</span><span class="p">[</span><span class="n">sub_svc</span><span class="o">.</span><span class="n">name</span><span class="p">]</span> <span class="o">=</span> <span class="p">{}</span>
|
|
<span class="n">groupnames</span> <span class="o">=</span> <span class="n">sub_svc</span><span class="o">.</span><span class="n">get_groupnames</span><span class="p">()</span>
|
|
<span class="k">if</span> <span class="n">groupnames</span><span class="p">:</span>
|
|
<span class="c1"># sub-service is associated with a group(s)</span>
|
|
<span class="n">jdict</span><span class="p">[</span><span class="n">sub_svc</span><span class="o">.</span><span class="n">name</span><span class="p">][</span><span class="s1">'children'</span><span class="p">]</span> <span class="o">=</span> <span class="n">groupnames</span>
|
|
<span class="k">else</span><span class="p">:</span>
|
|
<span class="c1"># sub-service is associated with parent service</span>
|
|
<span class="n">jdict</span><span class="p">[</span><span class="n">sub_svc</span><span class="o">.</span><span class="n">name</span><span class="p">][</span><span class="s1">'children'</span><span class="p">]</span> <span class="o">=</span> \
|
|
<span class="p">[</span><span class="n">sub_svc</span><span class="o">.</span><span class="n">get_parent_servicename</span><span class="p">()]</span>
|
|
|
|
<span class="c1"># temporarily create group containing all hosts. this is needed for</span>
|
|
<span class="c1"># ansible commands that are performed on hosts not yet in groups.</span>
|
|
<span class="n">group</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">add_group</span><span class="p">(</span><span class="s1">'__GLOBAL__'</span><span class="p">)</span>
|
|
<span class="n">jdict</span><span class="p">[</span><span class="n">group</span><span class="o">.</span><span class="n">name</span><span class="p">]</span> <span class="o">=</span> <span class="p">{}</span>
|
|
<span class="n">jdict</span><span class="p">[</span><span class="n">group</span><span class="o">.</span><span class="n">name</span><span class="p">][</span><span class="s1">'hosts'</span><span class="p">]</span> <span class="o">=</span> <span class="n">deploy_hostnames</span>
|
|
<span class="n">jdict</span><span class="p">[</span><span class="n">group</span><span class="o">.</span><span class="n">name</span><span class="p">][</span><span class="s1">'vars'</span><span class="p">]</span> <span class="o">=</span> <span class="n">group</span><span class="o">.</span><span class="n">get_vars</span><span class="p">()</span>
|
|
<span class="bp">self</span><span class="o">.</span><span class="n">remove_group</span><span class="p">(</span><span class="n">group</span><span class="o">.</span><span class="n">name</span><span class="p">)</span>
|
|
|
|
<span class="c1"># process hosts vars</span>
|
|
<span class="n">jdict</span><span class="p">[</span><span class="s1">'_meta'</span><span class="p">]</span> <span class="o">=</span> <span class="p">{}</span>
|
|
<span class="n">jdict</span><span class="p">[</span><span class="s1">'_meta'</span><span class="p">][</span><span class="s1">'hostvars'</span><span class="p">]</span> <span class="o">=</span> <span class="p">{}</span>
|
|
<span class="k">for</span> <span class="n">hostname</span> <span class="ow">in</span> <span class="n">deploy_hostnames</span><span class="p">:</span>
|
|
<span class="n">host</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_host</span><span class="p">(</span><span class="n">hostname</span><span class="p">)</span>
|
|
<span class="k">if</span> <span class="n">host</span><span class="p">:</span>
|
|
<span class="n">jdict</span><span class="p">[</span><span class="s1">'_meta'</span><span class="p">][</span><span class="s1">'hostvars'</span><span class="p">][</span><span class="n">hostname</span><span class="p">]</span> <span class="o">=</span> <span class="n">host</span><span class="o">.</span><span class="n">get_vars</span><span class="p">()</span>
|
|
<span class="k">return</span> <span class="n">json</span><span class="o">.</span><span class="n">dumps</span><span class="p">(</span><span class="n">jdict</span><span class="p">)</span>
|
|
</div>
|
|
<span class="k">def</span> <span class="nf">_filter_hosts</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">initial_hostnames</span><span class="p">,</span> <span class="n">deploy_hostnames</span><span class="p">):</span>
|
|
<span class="sd">"""filter out hosts not in deploy hosts</span>
|
|
|
|
<span class="sd"> Must preserve the ordering of hosts in the group.</span>
|
|
<span class="sd"> """</span>
|
|
<span class="n">filtered_hostnames</span> <span class="o">=</span> <span class="p">[]</span>
|
|
<span class="k">for</span> <span class="n">hostname</span> <span class="ow">in</span> <span class="n">initial_hostnames</span><span class="p">:</span>
|
|
<span class="k">if</span> <span class="n">hostname</span> <span class="ow">in</span> <span class="n">deploy_hostnames</span><span class="p">:</span>
|
|
<span class="n">filtered_hostnames</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">hostname</span><span class="p">)</span>
|
|
<span class="k">return</span> <span class="n">filtered_hostnames</span>
|
|
|
|
<div class="viewcode-block" id="Inventory.create_json_gen_file"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Inventory.create_json_gen_file">[docs]</a> <span class="k">def</span> <span class="nf">create_json_gen_file</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">inventory_filter</span><span class="o">=</span><span class="bp">None</span><span class="p">):</span>
|
|
<span class="sd">"""create json inventory file using filter ({})</span>
|
|
|
|
<span class="sd"> The inventory will be placed in a directory in /tmp,</span>
|
|
<span class="sd"> with the directory name of form kolla_uuid.py,</span>
|
|
<span class="sd"> where uuid is a unique deployment id.</span>
|
|
|
|
<span class="sd"> return path to filtered json generator file</span>
|
|
<span class="sd"> """</span>
|
|
<span class="n">json_out</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_ansible_json</span><span class="p">(</span><span class="n">inventory_filter</span><span class="p">)</span>
|
|
|
|
<span class="n">deploy_id</span> <span class="o">=</span> <span class="nb">str</span><span class="p">(</span><span class="n">uuid</span><span class="o">.</span><span class="n">uuid4</span><span class="p">())</span>
|
|
<span class="n">dirname</span> <span class="o">=</span> <span class="s1">'kolla_</span><span class="si">%s</span><span class="s1">'</span> <span class="o">%</span> <span class="n">deploy_id</span>
|
|
<span class="n">dirpath</span> <span class="o">=</span> <span class="n">os</span><span class="o">.</span><span class="n">path</span><span class="o">.</span><span class="n">join</span><span class="p">(</span><span class="n">tempfile</span><span class="o">.</span><span class="n">gettempdir</span><span class="p">(),</span> <span class="n">dirname</span><span class="p">)</span>
|
|
<span class="n">os</span><span class="o">.</span><span class="n">mkdir</span><span class="p">(</span><span class="n">dirpath</span><span class="p">)</span>
|
|
<span class="n">json_gen_path</span> <span class="o">=</span> <span class="n">os</span><span class="o">.</span><span class="n">path</span><span class="o">.</span><span class="n">join</span><span class="p">(</span><span class="n">dirpath</span><span class="p">,</span> <span class="s1">'temp_inventory.py'</span><span class="p">)</span>
|
|
|
|
<span class="k">with</span> <span class="nb">open</span><span class="p">(</span><span class="n">json_gen_path</span><span class="p">,</span> <span class="s1">'w'</span><span class="p">)</span> <span class="k">as</span> <span class="n">json_gen_file</span><span class="p">:</span>
|
|
<span class="n">json_gen_file</span><span class="o">.</span><span class="n">write</span><span class="p">(</span><span class="s1">'#!/usr/bin/env python</span><span class="se">\n</span><span class="s1">'</span><span class="p">)</span>
|
|
<span class="c1"># the quotes here are significant. The json_out has double quotes</span>
|
|
<span class="c1"># embedded in it so single quotes are needed to wrap it.</span>
|
|
<span class="n">json_gen_file</span><span class="o">.</span><span class="n">write</span><span class="p">(</span><span class="s2">"print('</span><span class="si">%s</span><span class="s2">')"</span> <span class="o">%</span> <span class="n">json_out</span><span class="p">)</span>
|
|
|
|
<span class="c1"># set executable by group</span>
|
|
<span class="n">os</span><span class="o">.</span><span class="n">chmod</span><span class="p">(</span><span class="n">json_gen_path</span><span class="p">,</span> <span class="mi">0</span><span class="n">o555</span><span class="p">)</span> <span class="c1"># nosec</span>
|
|
<span class="k">return</span> <span class="n">json_gen_path</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Inventory.remove_json_gen_file"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Inventory.remove_json_gen_file">[docs]</a> <span class="k">def</span> <span class="nf">remove_json_gen_file</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">path</span><span class="p">):</span>
|
|
<span class="n">remove_temp_inventory</span><span class="p">(</span><span class="n">path</span><span class="p">)</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Inventory.validate_hostnames"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Inventory.validate_hostnames">[docs]</a> <span class="k">def</span> <span class="nf">validate_hostnames</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">hostnames</span><span class="p">):</span>
|
|
<span class="k">if</span> <span class="ow">not</span> <span class="n">hostnames</span><span class="p">:</span>
|
|
<span class="k">raise</span> <span class="n">MissingArgument</span><span class="p">(</span><span class="n">u</span><span class="o">.</span><span class="n">_</span><span class="p">(</span><span class="s1">'Host name(s)'</span><span class="p">))</span>
|
|
<span class="n">invalid_hosts</span> <span class="o">=</span> <span class="p">[]</span>
|
|
<span class="k">for</span> <span class="n">hostname</span> <span class="ow">in</span> <span class="n">hostnames</span><span class="p">:</span>
|
|
<span class="k">if</span> <span class="n">hostname</span> <span class="ow">not</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">_hosts</span><span class="p">:</span>
|
|
<span class="n">invalid_hosts</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">hostname</span><span class="p">)</span>
|
|
<span class="k">if</span> <span class="n">invalid_hosts</span><span class="p">:</span>
|
|
<span class="k">raise</span> <span class="n">NotInInventory</span><span class="p">(</span><span class="n">u</span><span class="o">.</span><span class="n">_</span><span class="p">(</span><span class="s1">'Host'</span><span class="p">),</span> <span class="n">invalid_hosts</span><span class="p">)</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Inventory.validate_groupnames"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Inventory.validate_groupnames">[docs]</a> <span class="k">def</span> <span class="nf">validate_groupnames</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">groupnames</span><span class="p">):</span>
|
|
<span class="k">if</span> <span class="ow">not</span> <span class="n">groupnames</span><span class="p">:</span>
|
|
<span class="k">raise</span> <span class="n">MissingArgument</span><span class="p">(</span><span class="n">u</span><span class="o">.</span><span class="n">_</span><span class="p">(</span><span class="s1">'Group name(s)'</span><span class="p">))</span>
|
|
<span class="n">invalid_groups</span> <span class="o">=</span> <span class="p">[]</span>
|
|
<span class="k">for</span> <span class="n">groupname</span> <span class="ow">in</span> <span class="n">groupnames</span><span class="p">:</span>
|
|
<span class="k">if</span> <span class="n">groupname</span> <span class="ow">not</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">_groups</span><span class="p">:</span>
|
|
<span class="n">invalid_groups</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">groupname</span><span class="p">)</span>
|
|
<span class="k">if</span> <span class="n">invalid_groups</span><span class="p">:</span>
|
|
<span class="k">raise</span> <span class="n">NotInInventory</span><span class="p">(</span><span class="n">u</span><span class="o">.</span><span class="n">_</span><span class="p">(</span><span class="s1">'Group'</span><span class="p">),</span> <span class="n">invalid_groups</span><span class="p">)</span>
|
|
</div>
|
|
<div class="viewcode-block" id="Inventory.validate_servicenames"><a class="viewcode-back" href="../../../kollacli.common.html#kollacli.common.inventory.Inventory.validate_servicenames">[docs]</a> <span class="k">def</span> <span class="nf">validate_servicenames</span><span class="p">(</span><span class="bp">self</span><span class="p">,</span> <span class="n">servicenames</span><span class="p">):</span>
|
|
<span class="k">if</span> <span class="ow">not</span> <span class="n">servicenames</span><span class="p">:</span>
|
|
<span class="k">raise</span> <span class="n">MissingArgument</span><span class="p">(</span><span class="n">u</span><span class="o">.</span><span class="n">_</span><span class="p">(</span><span class="s1">'Service name(s)'</span><span class="p">))</span>
|
|
<span class="n">invalid_services</span> <span class="o">=</span> <span class="p">[]</span>
|
|
<span class="k">for</span> <span class="n">servicename</span> <span class="ow">in</span> <span class="n">servicenames</span><span class="p">:</span>
|
|
<span class="k">if</span> <span class="p">(</span><span class="n">servicename</span> <span class="ow">not</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">_services</span> <span class="ow">and</span>
|
|
<span class="n">servicename</span> <span class="ow">not</span> <span class="ow">in</span> <span class="bp">self</span><span class="o">.</span><span class="n">_sub_services</span><span class="p">):</span>
|
|
<span class="n">invalid_services</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">servicename</span><span class="p">)</span>
|
|
<span class="k">if</span> <span class="n">invalid_services</span><span class="p">:</span>
|
|
<span class="k">raise</span> <span class="n">NotInInventory</span><span class="p">(</span><span class="n">u</span><span class="o">.</span><span class="n">_</span><span class="p">(</span><span class="s1">'Service'</span><span class="p">),</span> <span class="n">invalid_services</span><span class="p">)</span></div></div>
|
|
</pre></div>
|
|
|
|
</div>
|
|
</div>
|
|
</div>
|
|
<div class="sphinxsidebar">
|
|
<div class="sphinxsidebarwrapper">
|
|
<div id="searchbox" style="display: none">
|
|
<h3>Quick search</h3>
|
|
<form class="search" action="../../../search.html" method="get">
|
|
<input type="text" name="q" />
|
|
<input type="submit" value="Go" />
|
|
<input type="hidden" name="check_keywords" value="yes" />
|
|
<input type="hidden" name="area" value="default" />
|
|
</form>
|
|
<p class="searchtip" style="font-size: 90%">
|
|
Enter search terms or a module, class or function name.
|
|
</p>
|
|
</div>
|
|
<script type="text/javascript">$('#searchbox').show(0);</script>
|
|
</div>
|
|
</div>
|
|
<div class="clearer"></div>
|
|
</div>
|
|
<div class="related">
|
|
<h3>Navigation</h3>
|
|
<ul>
|
|
<li class="right" style="margin-right: 10px">
|
|
<a href="../../../genindex.html" title="General Index"
|
|
>index</a></li>
|
|
<li class="right" >
|
|
<a href="../../../py-modindex.html" title="Python Module Index"
|
|
>modules</a> |</li>
|
|
<li><a href="../../../index.html">kollacli 1.0.0.dev401 documentation</a> »</li>
|
|
<li><a href="../../index.html" >Module code</a> »</li>
|
|
</ul>
|
|
</div>
|
|
<div class="footer">
|
|
© Copyright 2016, Oracle.
|
|
Created using <a href="http://sphinx-doc.org/">Sphinx</a> 1.2.3.
|
|
</div>
|
|
</body>
|
|
</html> |