Python - 只能在两个关键字列表中检测到关键字时发送消息？

Question

我有一个 python 脚本，它从关键字列表keywords = ['camera', 'nikon'] 中检测一个关键字，然后向 Slack 发送一条消息，如下所示

检测到关键字相机

'Reddit 帖子网址'

'包含关键字的reddit评论'

如果脚本从第二个关键字列表color_keywords = ['red', 'blue'] 中检测到一个关键字，那么它会发布以下内容

检测到关键字相机

'Reddit 帖子网址'

'包含关键字的reddit评论'

检测到颜色

我的问题是，我是否能够以某种方式拥有脚本，以便仅在找到每个关键字列表中的关键字时才发送消息？ 所以如果它只从第一个列表中找到一个关键字，它将被忽略，如果它从第二个列表中找到一个，它也将被忽略。但如果它从 BOTH 列表中找到一个关键字，它会将消息发送到 slack。

下面是我当前的代码

MSG_TEMPLATE = """Keyword *{keyword}* detected
https://www.reddit.com{permalink}
```{comment_body}```"""

keywords = ['camera', 'nikon', 'canon']  
color_keywords = ['blue', 'red']

with open(save_path, 'r') as fp:
    alerted_comments = json.load(fp)

    for comment in comment_stream:
        if comment.id in alerted_comments:
            continue

        if comment.author:  # if comment author hasn't deleted
            if comment.author.name in ignore_users:
                continue

        if any(kw.lower() in comment.body.lower() for kw in keywords):
            found_kws = [kw for kw in keywords if kw.lower() in comment.body.lower()]

            msg = MSG_TEMPLATE.format(
                keyword=found_kws[0],
                permalink=comment.permalink,
                comment_body=comment.body
            )

            if any(kw.lower() in comment.body.lower() for kw in color_keywords):
                msg += "\n<!here> *A color was detected*"

            slack_data = {'text': msg, 'mrkdwn': True,}

            response = requests.post('https://hooks.slack.com/services/TB7AH6U2G/xxxxxxx/0KOjl9251TZExxxxxxxx',
                                             data=json.dumps(slack_data), headers={'Content-Type': 'application/json'})

任何帮助将不胜感激！

Answer 1

当然！为简洁起见，摘录以下代码：

def find_keywords(comment, word_list):
    """:returns: List of matching keywords present in the comment, or the empty list"""
    return [word for word in word_list if word.lower() in comment.body.lower()]

for comment in comment_stream:
    if not should_be_ignored(comment):
        found_kws = find_keywords(comment, keywords)
        found_colors = find_keywords(comment, color_keywords)

        if found_kws and found_colors:
            # At this point, we're guaranteed to have *both* one or more keywords *and* one or more colors
            send_message(comment, found_kws, found_colors)

这里的关键见解是：您首先创建匹配列表，然后随后检查它们以决定是否要发送消息。在这种情况下，只有两个列表都不为空，您才会继续发送消息。

（当然，should_be_ignored() 和 send_message() 的实现留给读者作为练习。:)）

编辑：原代码的完整实现：​​

def send_message(comment, keywords, colors):
    assert keywords and colors, "At this point, we should *only* be calling this function if we have at least one keyword and one color"

    MSG_TEMPLATE = """Keyword *{keyword}* and color *{color}* detected
    https://www.reddit.com{permalink}
    ```{comment_body}```"""
    msg = MSG_TEMPLATE.format(
        keyword=keywords[0],
        color=colors[0],
        permalink=comment.permalink,
        comment_body=comment.body
    )
    slack_data = {'text': msg, 'mrkdwn': True,}
    response = requests.post('https://hooks.slack.com/services/TB7AH6U2G/xxxxxxx/0KOjl9251TZExxxxxxxx',
                             data=json.dumps(slack_data), headers={'Content-Type': 'application/json'})


def should_be_ignored(comment, alerted):
    return comment.id in alerted or (comment.author and comment.author.name in ignore_users)


def find_keywords(comment, word_list):
    """:returns: List of matching keywords present in the comment, or the empty list"""
    return [word for word in word_list if word.lower() in comment.body.lower()]


keywords = ['camera', 'nikon', 'canon']  
color_keywords = ['blue', 'red']

with open(save_path, 'r') as fp:
    alerted_comments = json.load(fp)

for comment in comment_stream:
    if not should_be_ignored(comment, alerted_comments):
        found_kws = find_keywords(comment, keywords)
        found_colors = find_keywords(comment, color_keywords)

        if found_kws and found_colors:
            # At this point, we're guaranteed to have *both* one or more keywords *and* one or more colors
            send_message(comment, found_kws, found_colors)

请注意，我所做的一切（除了我们在发送消息之前同时具有颜色和关键字的新要求）是将您的一些业务逻辑提取到 @987654327 @ 和 send_message() 函数，希望能阐明代码主体的意图。这应该是您开始使用的示例的直接替代品。