mirror of https://github.com/zulip/zulip.git
119 lines
3.6 KiB
Python
119 lines
3.6 KiB
Python
"""
|
|
This bot uses the python library `howdoi` which is not a
|
|
dependency of Zulip. To use this module, you will have to
|
|
install it in your local machine. In your terminal, enter
|
|
the following command:
|
|
|
|
$ sudo pip install howdoi --upgrade
|
|
|
|
Note:
|
|
* You might have to use `pip3` if you are using python 3.
|
|
* The install command would also download any dependency
|
|
required by `howdoi`.
|
|
"""
|
|
|
|
from __future__ import absolute_import
|
|
|
|
import sys
|
|
import logging
|
|
from textwrap import fill
|
|
try:
|
|
from howdoi.howdoi import howdoi
|
|
except ImportError:
|
|
logging.error("Dependency missing!!\n%s" % (__doc__))
|
|
sys.exit(0)
|
|
|
|
|
|
class HowdoiHandler(object):
|
|
'''
|
|
This plugin facilitates searching Stack Overflow for
|
|
techanical answers based on the Python library `howdoi`.
|
|
To get the best possible answer, only include keywords
|
|
in your questions.
|
|
|
|
There are two possible commands:
|
|
* @mention-bot howdowe > This would return the answer to the same
|
|
stream that it was called from.
|
|
|
|
* @mention-bot howdoi > The bot would send a private message to the
|
|
user containing the answer.
|
|
|
|
By default, howdoi only returns the coding section of the
|
|
first search result if possible, to see the full answer
|
|
from Stack Overflow, append a '!' to the commands.
|
|
(ie '@mention-bot howdoi!', '@mention-bot howdowe!')
|
|
'''
|
|
|
|
MAX_LINE_LENGTH = 85
|
|
|
|
def usage(self):
|
|
return '''
|
|
This plugin will allow users to get techanical
|
|
answers from Stackoverflow. Users should preface
|
|
their questions with one of the following:
|
|
|
|
* @mention-bot howdowe > Answer to the same stream
|
|
* @mention-bot howdoi > Answer via private message
|
|
|
|
* @mention-bot howdowe! OR @mention-bot howdoi! > Full answer from SO
|
|
'''
|
|
|
|
def line_wrap(self, string, length):
|
|
lines = string.split("\n")
|
|
|
|
wrapped = [(fill(line) if len(line) > length else line)
|
|
for line in lines]
|
|
|
|
return "\n".join(wrapped).strip()
|
|
|
|
def get_answer(self, command, query):
|
|
question = query[len(command):].strip()
|
|
result = howdoi(dict(
|
|
query=question,
|
|
num_answers=1,
|
|
pos=1,
|
|
all=command[-1] == '!',
|
|
color=False
|
|
))
|
|
_answer = self.line_wrap(result, HowdoiHandler.MAX_LINE_LENGTH)
|
|
|
|
answer = "Answer to '%s':\n```\n%s\n```" % (question, _answer)
|
|
|
|
return answer
|
|
|
|
def handle_message(self, message, bot_handler, state_handler):
|
|
question = message['content'].strip()
|
|
|
|
if question.startswith('howdowe!'):
|
|
bot_handler.send_message(dict(
|
|
type='stream',
|
|
to=message['display_recipient'],
|
|
subject=message['subject'],
|
|
content=self.get_answer('howdowe!', question)
|
|
))
|
|
|
|
elif question.startswith('howdoi!'):
|
|
bot_handler.send_message(dict(
|
|
type='private',
|
|
to=message['sender_email'],
|
|
content=self.get_answer('howdoi!', question)
|
|
))
|
|
|
|
elif question.startswith('howdowe'):
|
|
bot_handler.send_message(dict(
|
|
type='stream',
|
|
to=message['display_recipient'],
|
|
subject=message['subject'],
|
|
content=self.get_answer('howdowe', question)
|
|
))
|
|
|
|
elif question.startswith('howdoi'):
|
|
bot_handler.send_message(dict(
|
|
type='private',
|
|
to=message['sender_email'],
|
|
content=self.get_answer('howdoi', question)
|
|
))
|
|
|
|
|
|
handler_class = HowdoiHandler
|