File size: 6,022 Bytes
056f521 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 |
# <============================================== IMPORTS =========================================================>
import asyncio
import html
import json
import re
from typing import Optional
import requests
from telegram import (
CallbackQuery,
Chat,
InlineKeyboardButton,
InlineKeyboardMarkup,
Update,
User,
)
from telegram.constants import ParseMode
from telegram.error import BadRequest, Forbidden, RetryAfter
from telegram.ext import (
CallbackQueryHandler,
CommandHandler,
ContextTypes,
MessageHandler,
filters,
)
from telegram.helpers import mention_html
import Database.sql.kuki_sql as sql
from Mikobot import function
from Mikobot.plugins.log_channel import gloggable
# <=======================================================================================================>
# <================================================ FUNCTION =======================================================>
@gloggable
async def kukirm(update: Update, context: ContextTypes.DEFAULT_TYPE):
query: Optional[CallbackQuery] = update.callback_query
user: Optional[User] = update.effective_user
if match := re.match(r"rm_chat\((.+?)\)", query.data):
user_id = match[1]
chat: Optional[Chat] = update.effective_chat
if is_kuki := sql.rem_kuki(chat.id):
sql.rem_kuki(user_id)
return (
f"<b>{html.escape(chat.title)}:</b>\n"
f"AI_DISABLED\n"
f"<b>Admin:</b> {mention_html(user.id, html.escape(user.first_name))}\n"
)
else:
await update.effective_message.edit_text(
f"Chatbot disable by {mention_html(user.id, user.first_name)}.",
parse_mode=ParseMode.HTML,
)
return ""
@gloggable
async def kukiadd(update: Update, context: ContextTypes.DEFAULT_TYPE):
query: Optional[CallbackQuery] = update.callback_query
user: Optional[User] = update.effective_user
if match := re.match(r"add_chat\((.+?)\)", query.data):
user_id = match[1]
chat: Optional[Chat] = update.effective_chat
if is_kuki := sql.set_kuki(chat.id):
sql.set_kuki(user_id)
return (
f"<b>{html.escape(chat.title)}:</b>\n"
f"AI_ENABLE\n"
f"<b>Admin:</b> {mention_html(user.id, html.escape(user.first_name))}\n"
)
else:
await update.effective_message.edit_text(
f"Hey Darling Chatbot enable by {mention_html(user.id, user.first_name)}.",
parse_mode=ParseMode.HTML,
)
return ""
@gloggable
async def kuki(update: Update, context: ContextTypes.DEFAULT_TYPE):
update.effective_user
message = update.effective_message
msg = "Choose an option"
keyboard = InlineKeyboardMarkup(
[
[InlineKeyboardButton(text="Enable", callback_data="add_chat({})")],
[InlineKeyboardButton(text="Disable", callback_data="rm_chat({})")],
]
)
await message.reply_text(
msg,
reply_markup=keyboard,
parse_mode=ParseMode.HTML,
)
async def kuki_message(context: ContextTypes.DEFAULT_TYPE, message):
reply_message = message.reply_to_message
if message.text.lower() == "kuki":
return True
if reply_message:
if reply_message.from_user.id == (await context.bot.get_me()).id:
return True
else:
return False
async def chatbot(update: Update, context: ContextTypes.DEFAULT_TYPE):
update.effective_user
message = update.effective_message
chat_id = update.effective_chat.id
bot = context.bot
is_kuki = sql.is_kuki(chat_id)
if not is_kuki:
return
if message.text and not message.document:
if not await kuki_message(context, message):
return
Message = message.text
await bot.send_chat_action(chat_id, action="typing")
kukiurl = requests.get(
f"http://api.brainshop.ai/get?bid=176809&key=lbMN8CXTGzhn1NKG&uid=[user]&msg={Message}"
)
Kuki = json.loads(kukiurl.text)
kuki = Kuki["cnt"]
await asyncio.sleep(0.3)
await message.reply_text(kuki)
async def list_all_chats(update: Update, context: ContextTypes.DEFAULT_TYPE):
chats = sql.get_all_kuki_chats()
text = "<b>Neko Enabled Chats</b>\n"
for chat in chats:
try:
x = await context.bot.get_chat(int(*chat))
name = x.title or x.first_name
text += f"• <code>{name}</code>\n"
except (BadRequest, Forbidden):
sql.rem_kuki(*chat)
except RetryAfter as e:
await asyncio.sleep(e.retry_after)
await update.effective_message.reply_text(text, parse_mode="HTML")
# <=================================================== HELP ====================================================>
__help__ = """
➠ *Admins only command*:
» /chatbot: shows chatbot panel.
"""
__mod_name__ = "CHATBOT"
# <================================================ HANDLER =======================================================>
CHATBOTK_HANDLER = CommandHandler("chatbot", kuki, block=False)
ADD_CHAT_HANDLER = CallbackQueryHandler(kukiadd, pattern=r"add_chat", block=False)
RM_CHAT_HANDLER = CallbackQueryHandler(kukirm, pattern=r"rm_chat", block=False)
CHATBOT_HANDLER = MessageHandler(
filters.TEXT
& (~filters.Regex(r"^#[^\s]+") & ~filters.Regex(r"^!") & ~filters.Regex(r"^\/")),
chatbot,
block=False,
)
LIST_ALL_CHATS_HANDLER = CommandHandler("allchats", list_all_chats, block=False)
function(ADD_CHAT_HANDLER)
function(CHATBOTK_HANDLER)
function(RM_CHAT_HANDLER)
function(LIST_ALL_CHATS_HANDLER)
function(CHATBOT_HANDLER)
__handlers__ = [
ADD_CHAT_HANDLER,
CHATBOTK_HANDLER,
RM_CHAT_HANDLER,
LIST_ALL_CHATS_HANDLER,
CHATBOT_HANDLER,
]
# <================================================ END =======================================================>
|