improve function

- added userbot auto join
- must check if the bot was promoted as admin first before playing video
This commit is contained in:
levina 2021-11-02 15:07:40 +07:00 committed by GitHub
parent e6db2a9b8c
commit d3dcf1b7bd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -2,15 +2,17 @@
# Commit Start Date 20/10/2021 # Commit Start Date 20/10/2021
# Finished On 28/10/2021 # Finished On 28/10/2021
import asyncio
import re import re
import asyncio
from config import BOT_USERNAME, GROUP_SUPPORT, IMG_1, IMG_2, UPDATES_CHANNEL
from driver.filters import command, other_filters
from driver.queues import QUEUE, add_to_queue
from driver.veez import call_py
from pyrogram import Client from pyrogram import Client
from driver.veez import call_py
from program import BOT_ID, USERBOT_ID
from driver.queues import QUEUE, add_to_queue
from driver.filters import command, other_filters
from pyrogram.errors import UserAlreadyParticipant, UserNotParticipant
from pyrogram.types import InlineKeyboardButton, InlineKeyboardMarkup, Message from pyrogram.types import InlineKeyboardButton, InlineKeyboardMarkup, Message
from config import BOT_USERNAME, GROUP_SUPPORT, IMG_1, IMG_2, UPDATES_CHANNEL, ASSISTANT_NAME
from pytgcalls import StreamType from pytgcalls import StreamType
from pytgcalls.types.input_stream import AudioPiped from pytgcalls.types.input_stream import AudioPiped
from youtubesearchpython import VideosSearch from youtubesearchpython import VideosSearch
@ -65,6 +67,52 @@ async def play(_, m: Message):
] ]
) )
chat_title = m.chat.title
a = await _.get_chat_member(m.chat.id, BOT_ID)
if a.status != "administrator":
await m.reply_text(f"💡 To use me, I need to be an **Administrator** with the following **permissions**:\n\n» ❌ __Delete messages__\n» ❌ __Ban users__\n» ❌ __Add users__\n» ❌ __Manage voice chat__\n\nData is **updated** automatically after you **promote me**")
return
if not a.can_manage_voice_chats:
await m.reply_text(
"missing required permission:"
+ "\n\n» ❌ __Manage voice chat__")
return
if not a.can_delete_messages:
await m.reply_text(
"missing required permission:"
+ "\n\n» ❌ __Delete messages__")
return
if not a.can_invite_users:
await m.reply_text(
"missing required permission:"
+ "\n\n» ❌ __Add users__")
return
if not a.can_restrict_members:
await m.reply_text(
"missing required permission:"
+ "\n\n» ❌ __Ban users__")
return
try:
b = await _.get_chat_member(m.chat.id, USERBOT_ID)
if b.status == "kicked":
await m.reply_text(f"@{ASSISTANT_NAME} **is banned in group** {chat_title}\n\n» **unban the userbot first if you want to use this bot.**")
return
except UserNotParticipant:
if m.chat.username:
try:
await call_py.join_chat(f"{m.chat.username}")
except Exception as e:
await m.reply_text(f"❌ **userbot failed to join**\n\n**reason**:{e}")
return
else:
try:
pope = await _.export_chat_invite_link(m.chat.id)
pepo = await _.revoke_chat_invite_link(m.chat.id, pope)
await call_py.join_chat(pepo.invite_link)
except UserAlreadyParticipant:
pass
except Exception as e:
return await m.reply_text(f"❌ **userbot failed to join**\n\n**reason**:{e}")
replied = m.reply_to_message replied = m.reply_to_message
chat_id = m.chat.id chat_id = m.chat.id
if replied: if replied:
@ -216,6 +264,52 @@ async def stream(_, m: Message):
] ]
) )
chat_title = m.chat.title
a = await _.get_chat_member(m.chat.id, BOT_ID)
if a.status != "administrator":
await m.reply_text(f"💡 To use me, I need to be an **Administrator** with the following **permissions**:\n\n» ❌ __Delete messages__\n» ❌ __Ban users__\n» ❌ __Add users__\n» ❌ __Manage voice chat__\n\nData is **updated** automatically after you **promote me**")
return
if not a.can_manage_voice_chats:
await m.reply_text(
"missing required permission:"
+ "\n\n» ❌ __Manage voice chat__")
return
if not a.can_delete_messages:
await m.reply_text(
"missing required permission:"
+ "\n\n» ❌ __Delete messages__")
return
if not a.can_invite_users:
await m.reply_text(
"missing required permission:"
+ "\n\n» ❌ __Add users__")
return
if not a.can_restrict_members:
await m.reply_text(
"missing required permission:"
+ "\n\n» ❌ __Ban users__")
return
try:
b = await _.get_chat_member(m.chat.id, USERBOT_ID)
if b.status == "kicked":
await m.reply_text(f"@{ASSISTANT_NAME} **is banned in group** {chat_title}\n\n» **unban the userbot first if you want to use this bot.**")
return
except UserNotParticipant:
if m.chat.username:
try:
await call_py.join_chat(f"{m.chat.username}")
except Exception as e:
await m.reply_text(f"❌ **userbot failed to join**\n\n**reason**:{e}")
return
else:
try:
pope = await _.export_chat_invite_link(m.chat.id)
pepo = await _.revoke_chat_invite_link(m.chat.id, pope)
await call_py.join_chat(pepo.invite_link)
except UserAlreadyParticipant:
pass
except Exception as e:
return await m.reply_text(f"❌ **userbot failed to join**\n\n**reason**:{e}")
chat_id = m.chat.id chat_id = m.chat.id
if len(m.command) < 2: if len(m.command) < 2:
await m.reply("» give me a live-link/m3u8 url/youtube link to stream.") await m.reply("» give me a live-link/m3u8 url/youtube link to stream.")