import asyncio import re from io import BytesIO from os import remove import aiofiles from gpytranslate import Translator from pyrogram import enums, filters from pyrogram.errors import MessageTooLong, PeerIdInvalid from pyrogram.types import InlineKeyboardButton, InlineKeyboardMarkup, Message from wikipedia import summary from wikipedia.exceptions import DisambiguationError, PageError from Powers import * from Powers.bot_class import Gojo from Powers.database.users_db import Users from Powers.utils.clean_file import remove_markdown_and_html from Powers.utils.custom_filters import command from Powers.utils.extract_user import extract_user from Powers.utils.http_helper import * from Powers.utils.parser import mention_html @Gojo.on_message(command("wiki")) async def wiki(_, m: Message): if len(m.text.split()) <= 1: return await m.reply_text( text="Please check help on how to use this this command." ) search = m.text.split(None, 1)[1] try: res = summary(search) except DisambiguationError as de: return await m.reply_text( f"Disambiguated pages found! Adjust your query accordingly.\n{de}", parse_mode=enums.ParseMode.HTML, ) except PageError as pe: return await m.reply_text(f"{pe}", parse_mode=enums.ParseMode.HTML) if res: result = f"{search}\n\n" result += f"{res}\n" result += f"""Read more...""" try: return await m.reply_text( result, parse_mode=enums.ParseMode.HTML, disable_web_page_preview=True, ) except MessageTooLong: with BytesIO(str.encode(await remove_markdown_and_html(result))) as f: f.name = "result.txt" return await m.reply_document( document=f, quote=True, parse_mode=enums.ParseMode.HTML, ) await m.stop_propagation() @Gojo.on_message(command("gdpr")) async def gdpr_remove(_, m: Message): if m.from_user.id in SUPPORT_STAFF: await m.reply_text( "You're in my support staff, I cannot do that unless you are no longer a part of it!", ) return Users(m.from_user.id).delete_user() await m.reply_text( "Your personal data has been deleted.\n" "Note that this will not unban you from any chats, as that is telegram data, not Gojo data." " Flooding, warns, and gbans are also preserved, as of " "[this](https://ico.org.uk/for-organisations/guide-to-the-general-data-protection-regulation-gdpr/individual-rights/right-to-erasure/)," " which clearly states that the right to erasure does not apply 'for the performance of a task carried out in the public interest', " "as is the case for the aforementioned pieces of data.", disable_web_page_preview=True, ) await m.stop_propagation() @Gojo.on_message( command("lyrics") & (filters.group | filters.private), ) async def get_lyrics(_, m: Message): if not is_genius_lyrics: await m.reply_text("Genius lyrics api is missing") return if len(m.text.split()) <= 1: await m.reply_text(text="Please check help on how to use this this command.") return query = m.text.split(None, 1)[1] song = "" if not query: await m.edit_text(text="You haven't specified which song to look for!") return song_name = query em = await m.reply_text(text=f"Finding lyrics for {song_name}...") try: song = genius_lyrics.search_song(query) except Exception as e: await em.delete() await m.reply_text("Connection error try again after sometime") return if song: if song.lyrics: reply = song.lyrics reply = reply.split("\n",1)[1] artist = song.artist else: reply = "Couldn't find any lyrics for that song!" else: reply = "Song not found!" try: await em.edit_text(f"**{query.capitalize()} by {artist}**\n`{reply}`") except MessageTooLong: with BytesIO(str.encode(await remove_markdown_and_html(reply))) as f: f.name = "lyrics.txt" await m.reply_document( document=f, ) await em.delete() return @Gojo.on_message( command("id") & (filters.group | filters.private), ) async def id_info(c: Gojo, m: Message): ChatType = enums.ChatType if m.chat.type in [ChatType.SUPERGROUP, ChatType.GROUP] and not m.reply_to_message: await m.reply_text(text=f"This Group's ID is {m.chat.id}\nYour ID {m.from_user.id}") return if m.chat.type == ChatType.PRIVATE and not m.reply_to_message: await m.reply_text(text=f"Your ID is {m.chat.id}.") return user_id, _, _ = await extract_user(c, m) if user_id: if m.reply_to_message and m.reply_to_message.forward_from: user1 = m.reply_to_message.from_user user2 = m.reply_to_message.forward_from orig_sender = ((await mention_html(user2.first_name, user2.id)),) orig_id = (f"{user2.id}",) fwd_sender = ((await mention_html(user1.first_name, user1.id)),) fwd_id = (f"{user1.id}",) await m.reply_text( text=f"""Original Sender - {orig_sender} ({orig_id}) Forwarder - {fwd_sender} ({fwd_id})""", parse_mode=enums.ParseMode.HTML, ) else: try: user = await c.get_users(user_id) except PeerIdInvalid: await m.reply_text( text="""Failed to get user Peer ID invalid, I haven't seen this user anywhere earlier, maybe username would help to know them!""" ) return await m.reply_text( f"{(await mention_html(user.first_name, user.id))}'s ID is {user.id}.", parse_mode=enums.ParseMode.HTML, ) elif m.chat.type == ChatType.PRIVATE: text=f"Your ID is {m.chat.id}." if m.reply_to_message: if m.forward_from: text+=f"Forwarded from user ID {m.forward_from.id}." elif m.forward_from_chat: text+=f"Forwarded from user ID {m.forward_from_chat.id}." await m.reply_text() else: await m.reply_text(text=f"This Group's ID is {m.chat.id}\nYour ID {m.from_user.id}") return @Gojo.on_message( command("gifid") & (filters.group | filters.private), ) async def get_gifid(_, m: Message): if m.reply_to_message and m.reply_to_message.animation: LOGGER.info(f"{m.from_user.id} used gifid cmd in {m.chat.id}") await m.reply_text( f"Gif ID:\n{m.reply_to_message.animation.file_id}", parse_mode=enums.ParseMode.HTML, ) else: await m.reply_text(text="Please reply to a gif to get it's ID.") return @Gojo.on_message( command(["github", "git"]) & (filters.group | filters.private), ) async def github(_, m: Message): if len(m.text.split()) == 2: username = m.text.split(maxsplit=1)[1] LOGGER.info(f"{m.from_user.id} used github cmd in {m.chat.id}") else: await m.reply_text( f"Usage: {Config.PREFIX_HANDLER}github username", ) return username = username.split("/")[-1] URL = f"https://api.github.com/users/{username}" try: r = await get(URL, timeout=5) except asyncio.TimeoutError: return await m.reply_text("request timeout") except Exception as e: return await m.reply_text(f"ERROR: `{e}`") avtar = r.get("avatar_url", None) url = r.get("html_url", None) name = r.get("name", None) company = r.get("company", None) followers = r.get("followers", 0) following = r.get("following", 0) public_repos = r.get("public_repos", 0) bio = r.get("bio", None) created_at = r.get("created_at", "Not Found") location = r.get("location", None) email = r.get("email", None) updated_at = r.get("updated_at", "Not Found") blog = r.get("blog", None) twitter = r.get("twitter_username", None) REPLY = "" if name: REPLY += f"🧑‍💻 GitHub Info of {name}:" if url: REPLY += f"\n📎 URL: {username}" REPLY += f"\n🔑 Public Repos: {public_repos}" REPLY += f"\n🧲 Followers: {followers}" REPLY += f"\n✨ Following: {following}" if email: REPLY += f"\n✉️ Email: {email}" if company: org_url = company.strip("@") REPLY += f"\n™️ Organization: {company}" if blog: bname = blog.split(".")[-2] bname = bname.split("/")[-1] REPLY += f"\n📝 Blog: {bname}" if twitter: REPLY += f"\n⚜️ Twitter: {twitter}" if location: REPLY += f"\n🚀 Location: {location}" REPLY += f"\n💫 Created at: {created_at}" REPLY += f"\n⌚️ Updated at: {updated_at}" if bio: REPLY += f"\n\n🎯 Bio: {bio}" if avtar: return await m.reply_photo(photo=f"{avtar}", caption=REPLY) await m.reply_text(REPLY) return pattern = re.compile(r"^text/|json$|yaml$|xml$|toml$|x-sh$|x-shellscript$") BASE = "https://batbin.me/" async def paste(content: str): resp = await post(f"{BASE}api/v2/paste", data=content) if not resp["success"]: return return BASE + resp["message"] @Gojo.on_message(command("paste")) async def paste_func(_, message: Message): r = message.reply_to_message m = await message.reply_text("Pasting...") if not r: content = message.text.split(None, 1)[1] if r: if not r.text and not r.document: return await m.edit("Only text and documents are supported") if r.text: content = str(r.text) if r.document: if r.document.file_size > 40000: return await m.edit("You can only paste files smaller than 40KB.") if not pattern.search(r.document.mime_type): return await m.edit("Only text files can be pasted.") doc = await message.reply_to_message.download() async with aiofiles.open(doc, mode="r") as f: content = await f.read() remove(doc) link = await paste(content) kb = [[InlineKeyboardButton(text="Paste Link ", url=link)]] await m.delete() try: await message.reply_text("Here's your paste", reply_markup=InlineKeyboardMarkup(kb)) except Exception as e: if link: return await message.reply_text(f"Here's your paste:\n [link]({link})",) return await message.reply_text(f"Failed to post. Due to following error:\n{e}") @Gojo.on_message(command("tr")) async def tr(_, message): trl = Translator() if message.reply_to_message and ( message.reply_to_message.text or message.reply_to_message.caption ): if len(message.text.split()) == 1: target_lang = "en" else: target_lang = message.text.split()[1] if message.reply_to_message.text: text = message.reply_to_message.text else: text = message.reply_to_message.caption else: if len(message.text.split()) <= 2: await message.reply_text( "Provide lang code.\n[Available options](https://telegra.ph/Lang-Codes-02-22).\nUsage: /tr en", ) return target_lang = message.text.split(None, 2)[1] text = message.text.split(None, 2)[2] detectlang = await trl.detect(text) try: tekstr = await trl(text, targetlang=target_lang) except ValueError as err: await message.reply_text(f"Error: {str(err)}") return return await message.reply_text( f"Translated: from {detectlang} to {target_lang} \n``{tekstr.text}``", ) __PLUGIN__ = "utils" _DISABLE_CMDS_ = ["paste", "wiki", "id", "gifid", "tr", "github", "git"] __alt_name__ = ["util", "misc", "tools"] __HELP__ = """ **Utils** Some utils provided by bot to make your tasks easy! • /id: Get the current group id. If used by replying to a message, get that user's id. • /info: Get information about a user. • /gifid: Reply to a gif to me to tell you its file ID. • /lyrics `` : Find your song and give the lyrics of the song • /wiki: ``: wiki your query. • /tr ``: Translates the text and then replies to you with the language you have specifed, works as a reply to message. • /git ``: Search for the user using github api! • /weebify `` or ``: To weebify the text. **Example:** `/git iamgojoof6eyes`: this fetches the information about a user from the database."""