smmry: add keywords options

This commit is contained in:
nemunaire 2018-09-26 06:45:44 +02:00
parent 445a66ea90
commit 10b8ce8940

View File

@ -38,6 +38,7 @@ def load(context):
"WORDS/CMD": "" "WORDS/CMD": ""
}, },
keywords={ keywords={
"keywords?=X": "Returns keywords instead of summary (count optional)",
"length=7": "The number of sentences returned, default 7", "length=7": "The number of sentences returned, default 7",
"break": "inserts the string [BREAK] between sentences", "break": "inserts the string [BREAK] between sentences",
"ignore_length": "returns summary regardless of quality or length", "ignore_length": "returns summary regardless of quality or length",
@ -64,6 +65,7 @@ def cmd_smmry(msg):
if "quote_avoid" in msg.kwargs: URL += "&SM_QUOTE_AVOID" if "quote_avoid" in msg.kwargs: URL += "&SM_QUOTE_AVOID"
if "question_avoid" in msg.kwargs: URL += "&SM_QUESTION_AVOID" if "question_avoid" in msg.kwargs: URL += "&SM_QUESTION_AVOID"
if "exclamation_avoid" in msg.kwargs: URL += "&SM_EXCLAMATION_AVOID" if "exclamation_avoid" in msg.kwargs: URL += "&SM_EXCLAMATION_AVOID"
if "keywords" in msg.kwargs and msg.kwargs["keywords"] is not None and int(msg.kwargs["keywords"]) > 0: URL += "&SM_KEYWORD_COUNT=" + msg.kwargs["keywords"]
res = Response(channel=msg.channel) res = Response(channel=msg.channel)
@ -103,6 +105,9 @@ def cmd_smmry(msg):
title = "Unknown error" title = "Unknown error"
raise IMException(title + ": " + smmry['sm_api_message'].lower()) raise IMException(title + ": " + smmry['sm_api_message'].lower())
if "keywords" in msg.kwargs:
smmry["sm_api_content"] = ", ".join(smmry["sm_api_keyword_array"])
if "sm_api_title" in smmry and smmry["sm_api_title"] != "": if "sm_api_title" in smmry and smmry["sm_api_title"] != "":
res.append_message(smmry["sm_api_content"], title=smmry["sm_api_title"]) res.append_message(smmry["sm_api_content"], title=smmry["sm_api_title"])
else: else: