49 lines
1.6 KiB
Python
49 lines
1.6 KiB
Python
|
|
import click
|
|
import json
|
|
import logging
|
|
import requests
|
|
from datetime import datetime
|
|
from certidude import config
|
|
|
|
|
|
def publish(event_type, event_data=''):
|
|
"""
|
|
Publish event on nchan EventSource publisher
|
|
"""
|
|
assert event_type, "No event type specified"
|
|
|
|
if not isinstance(event_data, str):
|
|
from certidude.decorators import MyEncoder
|
|
event_data = json.dumps(event_data, cls=MyEncoder)
|
|
|
|
url = config.EVENT_SOURCE_PUBLISH % config.EVENT_SOURCE_TOKEN
|
|
click.echo("Publishing %s event '%s' on %s" % (event_type, event_data, url))
|
|
|
|
try:
|
|
notification = requests.post(
|
|
url,
|
|
data=event_data,
|
|
headers={"X-EventSource-Event": event_type, "User-Agent": "Certidude API"})
|
|
if notification.status_code == requests.codes.created:
|
|
pass # Sent to client
|
|
elif notification.status_code == requests.codes.accepted:
|
|
pass # Buffered in nchan
|
|
else:
|
|
click.echo("Failed to submit event to push server, server responded %d" % (
|
|
notification.status_code))
|
|
except requests.exceptions.ConnectionError:
|
|
click.echo("Failed to submit event to push server, connection error")
|
|
|
|
|
|
class EventSourceLogHandler(logging.Handler):
|
|
"""
|
|
To be used with Python log handling framework for publishing log entries
|
|
"""
|
|
def emit(self, record):
|
|
publish("log-entry", dict(
|
|
created = datetime.utcfromtimestamp(record.created),
|
|
message = record.msg % record.args,
|
|
severity = record.levelname.lower()))
|
|
|