External-Mirrors/awx
2
0
Fork 0
mirror of https://github.com/ansible/awx.git synced 2026-05-19 14:57:39 -02:30
Code Issues Packages Projects Releases Wiki Activity

add tests for our custom log aggregator/handler

Browse Source
This commit is contained in:
Ryan Petrello
2017-02-08 13:17:04 -05:00
parent 56303c144c
commit 2c447942d5
5 changed files with 195 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
awx/main/utils/handlers.py
View File

@@ -116,8 +116,16 @@ class HTTPSHandler(logging.Handler):
return self.enabled_loggers is None or logger_name.split('.')[-1] not in self.enabled_loggers
def emit(self, record):
"""
Emit a log record. When ``self.async`` is True, returns a list of
``concurrent.futures.Future`` objects.
See:
https://docs.python.org/3/library/concurrent.futures.html#future-objects
http://pythonhosted.org/futures/
"""
if self.skip_log(record.name):
return
return []
try:
payload = self.format(record)
host = self.get_http_host()
@@ -129,18 +137,21 @@ class HTTPSHandler(logging.Handler):
module_name = payload_data['module_name']
if module_name in ['services', 'packages', 'files']:
facts_dict = payload_data.pop(module_name)
async_futures = []
for key in facts_dict:
fact_payload = copy(payload_data)
fact_payload.update(facts_dict[key])
self.session.post(host, **self.get_post_kwargs(fact_payload))
return
async_futures.append(
self.session.post(host, **self.get_post_kwargs(fact_payload))
)
return async_futures
if self.async:
self.session.post(host, **self.get_post_kwargs(payload))
else:
requests.post(host, auth=requests.auth.HTTPBasicAuth(self.username, self.password), **self.get_post_kwargs(payload))
return [self.session.post(host, **self.get_post_kwargs(payload))]
requests.post(host, auth=requests.auth.HTTPBasicAuth(self.username, self.password), **self.get_post_kwargs(payload))
return []
except (KeyboardInterrupt, SystemExit):
raise
except:
self.handleError(record)