#!/usr/bin/env python import sys from os import path import logging sys.path.append(path.join(path.dirname(__file__), '../api')) import humbug class StreamLogger(object): """ Give a file-like interface to a logger instance. """ def __init__(self, logger, log_level=logging.INFO): self.logger = logger self.log_level = log_level self.buffer = "" def write(self, message): self.buffer += message if message[-1] == "\n": self.logger.log(self.log_level, self.buffer.rstrip()) self.buffer = "" logging.basicConfig(filename="/var/log/humbug/feedback-bot.log", level=logging.DEBUG, format="%(asctime)s %(levelname)s\t%(message)s") # The API prints to stdout, so capture and format those messages as well. stdout_logger = StreamLogger(logging.getLogger("stdout"), logging.INFO) sys.stdout = stdout_logger stderr_logger = StreamLogger(logging.getLogger("stderr"), logging.ERROR) sys.stderr = stderr_logger prod_client = humbug.Client( email="feedback@humbughq.com", api_key="xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx", verbose=True, site="https://humbughq.com") staging_client = humbug.Client( email="feedback@humbughq.com", api_key="xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx", verbose=True, site="https://staging.humbughq.com") def forward_message(message): if message["type"] != "private" or len(message["display_recipient"]) != 2: return if "sender_domain" in message: subject = "feedback: %s (%s)" % (message["sender_email"], message["sender_domain"]) else: subject = "feedback from %s" % message["sender_email"], if len(subject) > 60: subject = subject[:57].rstrip() + "..." forwarded_message = { "type": "stream", "to": "support", "subject": subject, "content": message["content"], } staging_client.send_message(forwarded_message) logging.info("Forwarded feedback from %s" % (message["sender_email"],)) prod_client.call_on_each_message(forward_message)