mirror of
https://github.com/ansible/awx.git
synced 2026-04-05 18:19:21 -02:30
926d6a6525
* We include a special header value in the user agent when tower proxies requests to get per-host rules. * This extends that header logic to when we fetch plans (playbooks)
99 lines
3.3 KiB
Python
99 lines
3.3 KiB
Python
from __future__ import (absolute_import, division, print_function)
|
|
__metaclass__ = type
|
|
|
|
import os
|
|
import requests
|
|
|
|
from ansible.plugins.action import ActionBase
|
|
|
|
|
|
class ActionModule(ActionBase):
|
|
|
|
def save_playbook(self, proj_path, plan, content):
|
|
fname = '{}-{}.yml'.format(plan.get('name', None) or 'insights-plan', plan['maintenance_id'])
|
|
file_path = os.path.join(proj_path, fname)
|
|
with open(file_path, 'w') as f:
|
|
f.write(content)
|
|
|
|
def is_stale(self, proj_path, etag):
|
|
file_path = os.path.join(proj_path, '.version')
|
|
try:
|
|
f = open(file_path, 'r')
|
|
version = f.read()
|
|
f.close()
|
|
return version != etag
|
|
except IOError:
|
|
return True
|
|
|
|
def write_version(self, proj_path, etag):
|
|
file_path = os.path.join(proj_path, '.version')
|
|
with open(file_path, 'w') as f:
|
|
f.write(etag)
|
|
|
|
def run(self, tmp=None, task_vars=None):
|
|
|
|
self._supports_check_mode = False
|
|
|
|
result = super(ActionModule, self).run(tmp, task_vars)
|
|
|
|
insights_url = self._task.args.get('insights_url', None)
|
|
username = self._task.args.get('username', None)
|
|
password = self._task.args.get('password', None)
|
|
proj_path = self._task.args.get('project_path', None)
|
|
license = self._task.args.get('awx_license_type', None)
|
|
awx_version = self._task.args.get('awx_version', None)
|
|
|
|
session = requests.Session()
|
|
session.auth = requests.auth.HTTPBasicAuth(username, password)
|
|
headers = {
|
|
'Content-Type': 'application/json',
|
|
'User-Agent': '{} {} ({})'.format(
|
|
'AWX' if license == 'open' else 'Red Hat Ansible Tower',
|
|
awx_version,
|
|
license
|
|
)
|
|
}
|
|
|
|
|
|
url = '{}/r/insights/v3/maintenance?ansible=true'.format(insights_url)
|
|
|
|
res = session.get(url, headers=headers, timeout=120)
|
|
|
|
if res.status_code != 200:
|
|
result['failed'] = True
|
|
result['msg'] = (
|
|
'Expected {} to return a status code of 200 but returned status '
|
|
'code "{}" instead with content "{}".'.format(url, res.status_code, res.content)
|
|
)
|
|
return result
|
|
|
|
if 'ETag' in res.headers:
|
|
version = res.headers['ETag']
|
|
if version.startswith('"') and version.endswith('"'):
|
|
version = version[1:-1]
|
|
else:
|
|
version = "ETAG_NOT_FOUND"
|
|
|
|
if not self.is_stale(proj_path, version):
|
|
result['changed'] = False
|
|
result['version'] = version
|
|
return result
|
|
|
|
for item in res.json():
|
|
url = '{}/r/insights/v3/maintenance/{}/playbook'.format(insights_url, item['maintenance_id'])
|
|
res = session.get(url, timeout=120)
|
|
if res.status_code != 200:
|
|
result['failed'] = True
|
|
result['msg'] = (
|
|
'Expected {} to return a status code of 200 but returned status '
|
|
'code "{}" instead with content "{}".'.format(url, res.status_code, res.content)
|
|
)
|
|
return result
|
|
self.save_playbook(proj_path, item, res.content)
|
|
|
|
self.write_version(proj_path, version)
|
|
|
|
result['changed'] = True
|
|
result['version'] = version
|
|
return result
|