09971ca76d
--------- Co-authored-by: Amit Sharma <48654350+buddhhu@users.noreply.github.com> Co-authored-by: Aditya <me@xditya.me> Co-authored-by: Dark <darkbeamer.official@gmail.com> Co-authored-by: buddhhu <amitsharma123234@gmail.com> Co-authored-by: Kaif <88398455+kaif-00z@users.noreply.github.com> Co-authored-by: 1Danish-00 <danish@ultroid.tech> Co-authored-by: TechiError <techierror@gmail.com> Co-authored-by: Dark <59723913+DarkBeamerYT@users.noreply.github.com>
241 lines
7.7 KiB
Python
241 lines
7.7 KiB
Python
# Ultroid - UserBot
|
||
# Copyright (C) 2021-2022 TeamUltroid
|
||
#
|
||
# This file is a part of < https://github.com/TeamUltroid/Ultroid/ >
|
||
# PLease read the GNU Affero General Public License in
|
||
# <https://www.github.com/TeamUltroid/Ultroid/blob/main/LICENSE/>.
|
||
|
||
from datetime import datetime
|
||
|
||
from pytz import timezone as tz
|
||
from telethon import Button, events
|
||
from telethon.errors.rpcerrorlist import MessageDeleteForbiddenError
|
||
from telethon.utils import get_display_name
|
||
|
||
from pyUltroid._misc import SUDO_M, owner_and_sudos
|
||
from pyUltroid.dB.base import KeyManager
|
||
from pyUltroid.fns.helper import inline_mention
|
||
from strings import get_string
|
||
|
||
from . import *
|
||
|
||
Owner_info_msg = udB.get_key("BOT_INFO_START")
|
||
custom_info = True
|
||
if Owner_info_msg is None:
|
||
custom_info = False
|
||
Owner_info_msg = f"""
|
||
**Owner** - {OWNER_NAME}
|
||
**OwnerID** - `{OWNER_ID}`
|
||
|
||
**Message Forwards** - {udB.get_key("PMBOT")}
|
||
|
||
**Ultroid [v{ultroid_version}](https://github.com/TeamUltroid/Ultroid), powered by @TeamUltroid**
|
||
"""
|
||
|
||
|
||
_settings = [
|
||
[
|
||
Button.inline("API Kᴇʏs", data="cbs_apiset"),
|
||
Button.inline("Pᴍ Bᴏᴛ", data="cbs_chatbot"),
|
||
],
|
||
[
|
||
Button.inline("Aʟɪᴠᴇ", data="cbs_alvcstm"),
|
||
Button.inline("PᴍPᴇʀᴍɪᴛ", data="cbs_ppmset"),
|
||
],
|
||
[
|
||
Button.inline("Fᴇᴀᴛᴜʀᴇs", data="cbs_otvars"),
|
||
Button.inline("VC Sᴏɴɢ Bᴏᴛ", data="cbs_vcb"),
|
||
],
|
||
[Button.inline("« Bᴀᴄᴋ", data="mainmenu")],
|
||
]
|
||
|
||
_start = [
|
||
[
|
||
Button.inline("Lᴀɴɢᴜᴀɢᴇ 🌐", data="lang"),
|
||
Button.inline("Sᴇᴛᴛɪɴɢs ⚙️", data="setter"),
|
||
],
|
||
[
|
||
Button.inline("Sᴛᴀᴛs ✨", data="stat"),
|
||
Button.inline("Bʀᴏᴀᴅᴄᴀsᴛ 📻", data="bcast"),
|
||
],
|
||
[Button.inline("TɪᴍᴇZᴏɴᴇ 🌎", data="tz")],
|
||
]
|
||
|
||
|
||
@callback("ownerinfo")
|
||
async def own(event):
|
||
msg = Owner_info_msg.format(
|
||
mention=event.sender.mention, me=inline_mention(ultroid_bot.me)
|
||
)
|
||
if custom_info:
|
||
msg += "\n\n• Powered by **@TeamUltroid**"
|
||
await event.edit(
|
||
msg,
|
||
buttons=[Button.inline("Close", data="closeit")],
|
||
link_preview=False,
|
||
)
|
||
|
||
|
||
@callback("closeit")
|
||
async def closet(lol):
|
||
try:
|
||
await lol.delete()
|
||
except MessageDeleteForbiddenError:
|
||
await lol.answer("MESSAGE_TOO_OLD", alert=True)
|
||
|
||
|
||
@asst_cmd(pattern="start( (.*)|$)", forwards=False, func=lambda x: not x.is_group)
|
||
async def ultroid(event):
|
||
args = event.pattern_match.group(1).strip()
|
||
keym = KeyManager("BOT_USERS", cast=list)
|
||
if not keym.contains(event.sender_id) and event.sender_id not in owner_and_sudos():
|
||
keym.add(event.sender_id)
|
||
kak_uiw = udB.get_key("OFF_START_LOG")
|
||
if not kak_uiw or kak_uiw != True:
|
||
msg = f"{inline_mention(event.sender)} `[{event.sender_id}]` started your [Assistant bot](@{asst.me.username})."
|
||
buttons = [[Button.inline("Info", "itkkstyo")]]
|
||
if event.sender.username:
|
||
buttons[0].append(
|
||
Button.mention(
|
||
"User", await event.client.get_input_entity(event.sender_id)
|
||
)
|
||
)
|
||
await event.client.send_message(
|
||
udB.get_key("LOG_CHANNEL"), msg, buttons=buttons
|
||
)
|
||
if event.sender_id not in SUDO_M.fullsudos:
|
||
ok = ""
|
||
me = inline_mention(ultroid_bot.me)
|
||
mention = inline_mention(event.sender)
|
||
if args and args != "set":
|
||
await get_stored_file(event, args)
|
||
if not udB.get_key("STARTMSG"):
|
||
if udB.get_key("PMBOT"):
|
||
ok = "You can contact my master using this bot!!\n\nSend your Message, I will Deliver it To Master."
|
||
await event.reply(
|
||
f"Hey there {mention}, this is Ultroid Assistant of {me}!\n\n{ok}",
|
||
file=udB.get_key("STARTMEDIA"),
|
||
buttons=[Button.inline("Info.", data="ownerinfo")]
|
||
if Owner_info_msg
|
||
else None,
|
||
)
|
||
else:
|
||
await event.reply(
|
||
udB.get_key("STARTMSG").format(me=me, mention=mention),
|
||
file=udB.get_key("STARTMEDIA"),
|
||
buttons=[Button.inline("Info.", data="ownerinfo")]
|
||
if Owner_info_msg
|
||
else None,
|
||
)
|
||
else:
|
||
name = get_display_name(event.sender)
|
||
if args == "set":
|
||
await event.reply(
|
||
"Choose from the below options -",
|
||
buttons=_settings,
|
||
)
|
||
elif args:
|
||
await get_stored_file(event, args)
|
||
else:
|
||
await event.reply(
|
||
get_string("ast_3").format(name),
|
||
buttons=_start,
|
||
)
|
||
|
||
|
||
@callback("itkkstyo", owner=True)
|
||
async def ekekdhdb(e):
|
||
text = f"When New Visitor will visit your Assistant Bot. You will get this log message!\n\nTo Disable : {HNDLR}setdb OFF_START_LOG True"
|
||
await e.answer(text, alert=True)
|
||
|
||
|
||
@callback("mainmenu", owner=True, func=lambda x: not x.is_group)
|
||
async def ultroid(event):
|
||
await event.edit(
|
||
get_string("ast_3").format(OWNER_NAME),
|
||
buttons=_start,
|
||
)
|
||
|
||
|
||
@callback("stat", owner=True)
|
||
async def botstat(event):
|
||
ok = len(udB.get_key("BOT_USERS") or [])
|
||
msg = """Ultroid Assistant - Stats
|
||
Total Users - {}""".format(
|
||
ok,
|
||
)
|
||
await event.answer(msg, cache_time=0, alert=True)
|
||
|
||
|
||
@callback("bcast", owner=True)
|
||
async def bdcast(event):
|
||
keym = KeyManager("BOT_USERS", cast=list)
|
||
total = keym.count()
|
||
await event.edit(f"• Broadcast to {total} users.")
|
||
async with event.client.conversation(OWNER_ID) as conv:
|
||
await conv.send_message(
|
||
"Enter your broadcast message.\nUse /cancel to stop the broadcast.",
|
||
)
|
||
response = await conv.get_response()
|
||
if response.message == "/cancel":
|
||
return await conv.send_message("Cancelled!!")
|
||
success = 0
|
||
fail = 0
|
||
await conv.send_message(f"Starting a broadcast to {total} users...")
|
||
start = datetime.now()
|
||
for i in keym.get():
|
||
try:
|
||
await asst.send_message(int(i), response)
|
||
success += 1
|
||
except BaseException:
|
||
fail += 1
|
||
end = datetime.now()
|
||
time_taken = (end - start).seconds
|
||
await conv.send_message(
|
||
f"""
|
||
**Broadcast completed in {time_taken} seconds.**
|
||
Total Users in Bot - {total}
|
||
**Sent to** : `{success} users.`
|
||
**Failed for** : `{fail} user(s).`""",
|
||
)
|
||
|
||
|
||
@callback("setter", owner=True)
|
||
async def setting(event):
|
||
await event.edit(
|
||
"Choose from the below options -",
|
||
buttons=_settings,
|
||
)
|
||
|
||
|
||
@callback("tz", owner=True)
|
||
async def timezone_(event):
|
||
await event.delete()
|
||
pru = event.sender_id
|
||
var = "TIMEZONE"
|
||
name = "Timezone"
|
||
async with event.client.conversation(pru) as conv:
|
||
await conv.send_message(
|
||
"Send Your TimeZone From This List [Check From Here](http://www.timezoneconverter.com/cgi-bin/findzone.tzc)"
|
||
)
|
||
response = conv.wait_event(events.NewMessage(chats=pru))
|
||
response = await response
|
||
themssg = response.message.message
|
||
if themssg == "/cancel":
|
||
return await conv.send_message(
|
||
"Cancelled!!",
|
||
buttons=get_back_button("mainmenu"),
|
||
)
|
||
try:
|
||
tz(themssg)
|
||
await setit(event, var, themssg)
|
||
await conv.send_message(
|
||
f"{name} changed to {themssg}\n",
|
||
buttons=get_back_button("mainmenu"),
|
||
)
|
||
except BaseException:
|
||
await conv.send_message(
|
||
"Wrong TimeZone, Try again",
|
||
buttons=get_back_button("mainmenu"),
|
||
)
|