67 lines
2.1 KiB
Python
Executable file
67 lines
2.1 KiB
Python
Executable file
#!/usr/bin/env python
|
|
import sys
|
|
from os import path
|
|
import logging
|
|
|
|
sys.path.append(path.join(path.dirname(__file__), '../api'))
|
|
import zulip
|
|
sys.path.append(path.join(path.dirname(__file__), '../zproject'))
|
|
import local_settings
|
|
|
|
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/zulip/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 = zulip.Client(
|
|
email="feedback@zulip.com",
|
|
api_key=local_settings.FEEDBACK_BOT_KEY,
|
|
verbose=True,
|
|
site="https://api.zulip.com")
|
|
staging_client = zulip.Client(
|
|
email="feedback@zulip.com",
|
|
api_key=local_settings.FEEDBACK_BOT_KEY,
|
|
verbose=True,
|
|
site="https://staging.zulip.com/api")
|
|
|
|
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)
|