From 51ffce69a3b7c84a89feabbef4fd1e09b030b342 Mon Sep 17 00:00:00 2001 From: yasirarism <55983182+yasirarism@users.noreply.github.com> Date: Sat, 26 Aug 2023 22:18:27 +0700 Subject: [PATCH] Salah escape text --- misskaty/plugins/dev.py | 2 ++ misskaty/plugins/inline_search.py | 4 ++-- misskaty/plugins/misc_tools.py | 8 ++++---- 3 files changed, 8 insertions(+), 6 deletions(-) diff --git a/misskaty/plugins/dev.py b/misskaty/plugins/dev.py index 775917f9..1756ac9d 100644 --- a/misskaty/plugins/dev.py +++ b/misskaty/plugins/dev.py @@ -8,6 +8,7 @@ import pickle import re import sys import traceback +from bs4 import BeautifulSoup from datetime import datetime from inspect import getfullargspec from shutil import disk_usage @@ -420,6 +421,7 @@ async def cmd_eval(self: Client, ctx: Message, strings) -> Optional[str]: "fetch": fetch, "replied": ctx.reply_to_message, "requests": requests, + "soup": BeautifulSoup, "help": _help, } eval_vars.update(var) diff --git a/misskaty/plugins/inline_search.py b/misskaty/plugins/inline_search.py index 59b56ea8..40695b87 100644 --- a/misskaty/plugins/inline_search.py +++ b/misskaty/plugins/inline_search.py @@ -236,8 +236,8 @@ async def inline_menu(_, inline_query: InlineQuery): snippet = result.select_one("#rso .lyLwlc").text except: snippet = "-" - message_text = f"{title}\n" - message_text += f"Deskription: {html.escape(snippet)}" + message_text = f"{html.escape(title)}\n" + message_text += f"Deskription: {html.escape(snippet)}\nGoogleSearch by @{self.me.username}" data.append( InlineQueryResultArticle( title=f"{title}", diff --git a/misskaty/plugins/misc_tools.py b/misskaty/plugins/misc_tools.py index 633cd2e6..42377b66 100644 --- a/misskaty/plugins/misc_tools.py +++ b/misskaty/plugins/misc_tools.py @@ -208,7 +208,7 @@ async def gsearch(_, message): # collect data data = [] - for result in soup.select(".tF2Cxc"): + for result in soup.select(".tF2Cxc"):hmm link = result.select_one(".yuRUbf a")["href"] title = result.select_one(".DKV0Md").text try: @@ -219,16 +219,16 @@ async def gsearch(_, message): # appending data to an array data.append( { - "title": title, + "title": html.escape(title), "link": link, - "snippet": snippet, + "snippet": html.escape(snippet), } ) arr = json.dumps(data, indent=2, ensure_ascii=False) parse = json.loads(arr) total = len(parse) res = "".join( - f"{i['title']}\n{html.escape(i['snippet'])}\n\n" + f"{i['title']}\n{i['snippet']}\n\n" for i in parse ) except Exception: