Word_Swap.py

"""
Word swap module
"""
import re
import logging
import html
import difflib

from telegram import Bot, Update
from telegram.ext import Filters, MessageHandler, Updater

import core
PLUGINVERSION = 2
from difflib import Differ

def appendBoldChanges(s1, s2):
    """
    Adds <b></b> tags to words that are changed
    https://stackoverflow.com/a/10775310
    """
    l1 = s1.split(' ')
    l2 = s2.split(' ')
    dif = list(Differ().compare(l1, l2))
    return " ".join(['<b>'+i[2:]+'</b>' if i[:1] == '+' else i[2:] for i in dif 
                                                           if not i[:1] in '-?'])
# Always name this variable as `plugin`
# If you dont, module loader will fail to load the plugin!
plugin = core.Plugin()
@plugin.command(command="/s/",
                description="Swaps word in message",
                inline_supported=True,
                hidden=False)
def wordsw(bot: Bot, update: Update, user, args):
    """
    Example usage:
    User A
    Hi

    User B
    [In reply to User A]
    /s/Hi/Bye

    OctoBot
    Hello, User A
    Did you mean:
    Bye
    """
    msg = update.message
    txt = msg.text
    if msg.reply_to_message is not None:
        if txt.startswith("/s"):
            if not msg.reply_to_message.from_user.name == bot.getMe().name:
                origword = txt[2]
                swap = txt[3]
                groups = [b.replace("\/", "/") for b in re.split(r"(?<!\\)/", txt)]
                find = groups[2]
                replacement = groups[3]
                if len(groups) > 4:
                    flags = groups[4]
                else:
                    flags = ""
                if len(origword) > 0:
                    find_re = re.compile("{}{}".format("(?{})".format(flags.replace("g", "")) if flags.replace("g", "") != "" else "", find))
                    mod_msg = html.escape(find_re.sub(replacement, msg.reply_to_message.text, count=int("g" not in flags)))
                    mod_msg = appendBoldChanges(msg.reply_to_message.text, mod_msg)
                    text = "Hello, {}\nDid you mean:\n{}".format(
                            msg.reply_to_message.from_user.first_name,
                            mod_msg
                    )
                    return core.message(text=text, parse_mode="HTML")