File size: 4,700 Bytes
8db43b6 | 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 | # This file is a part of TG-FileStreamBot
import math
from pyrogram import filters, Client
from pyrogram.types import InlineKeyboardMarkup, InlineKeyboardButton, Message
from pyrogram.enums.parse_mode import ParseMode
from WebStreamer import __version__
from WebStreamer.bot import StreamBot
from WebStreamer.server.exceptions import FIleNotFound
from WebStreamer.utils.bot_utils import is_user_accepted_tos, validate_user
from WebStreamer.vars import Var
from WebStreamer.utils.database import Database
from WebStreamer.utils.Translation import Language, BUTTON
db = Database(Var.DATABASE_URL, Var.SESSION_NAME)
@StreamBot.on_message(filters.command('start') & filters.private)
async def start(_: Client, message: Message):
lang = Language(message)
if not await validate_user(message, lang):
return
# usr_cmd = message.text.split(" ")
# if usr_cmd[-1] == "/start":
await message.reply_text(
text=lang.START_TEXT.format(message.from_user.mention),
parse_mode=ParseMode.HTML,
disable_web_page_preview=True,
reply_markup=BUTTON.START_BUTTONS
)
@StreamBot.on_message(filters.command("about") & filters.private)
async def about(_: Client, message: Message):
lang = Language(message)
if not await validate_user(message, lang):
return
await message.reply_text(
text=lang.ABOUT_TEXT.format(__version__),
disable_web_page_preview=True,
reply_markup=BUTTON.ABOUT_BUTTONS
)
@StreamBot.on_message((filters.command('help')) & filters.private)
async def help_handler(_: Client, message: Message):
lang = Language(message)
if not await validate_user(message, lang):
return
await message.reply_text(
text=lang.HELP_TEXT.format(Var.UPDATES_CHANNEL),
parse_mode=ParseMode.HTML,
disable_web_page_preview=True,
reply_markup=BUTTON.HELP_BUTTONS
)
# ---------------------------------------------------------------------------------------------------
@StreamBot.on_message(filters.command('myfiles') & filters.private)
async def my_files(_: Client, message: Message):
if not await validate_user(message):
return
user_files, total_files=await db.find_files(message.from_user.id, [1,10])
file_list=[]
async for x in user_files:
file_list.append([InlineKeyboardButton(x["file_name"], callback_data=f"myfile_{x['_id']}_{1}")])
if total_files > 10:
file_list.append(
[
InlineKeyboardButton("<<", callback_data="N/A"),
InlineKeyboardButton(f"1/{math.ceil(total_files/10)}", callback_data="N/A"),
InlineKeyboardButton(">>", callback_data="userfiles_2")
]
)
if not file_list:
file_list.append([InlineKeyboardButton("Empty", callback_data="N/A")])
await message.reply_photo(photo=Var.IMAGE_FILEID,
caption="Total files: {total_files}",
reply_markup=InlineKeyboardMarkup(file_list))
@StreamBot.on_message(filters.command('tos') & filters.private)
async def tos_handler(_: Client, message: Message):
if not Var.TOS:
await message.reply_text("This bot does not have any terms of service.")
return
if await is_user_accepted_tos(message):
await message.reply_text(
Var.TOS,
reply_markup=InlineKeyboardMarkup([[InlineKeyboardButton("✅ I accepted the TOS", callback_data="N/A")]])
)
@StreamBot.on_message(filters.command('info') & filters.private)
async def info_handler(_: Client, message: Message):
lang = Language(message)
if not await validate_user(message, lang):
return
i_cmd=message.text.split()
if (message.from_user.id == Var.OWNER_ID) and (len(i_cmd) > 1):
message.from_user.id=int(i_cmd[1])
user = await db.get_user(message.from_user.id)
files=await db.total_files(message.from_user.id)
links="N/A"
if (user.get("Plan") == "Free") and (Var.LINK_LIMIT):
links=Var.LINK_LIMIT-files
await message.reply_text(lang.INFO_TEXT.format(message.from_user.id, user.get("Plan"), files, links))
@StreamBot.on_message(filters.command('getfile') & filters.private & filters.user(Var.OWNER_ID))
async def getfile(_: Client, message: Message):
if not await validate_user(message):
return
usr_cmd=message.text.split()[1:]
if len(usr_cmd) < 1:
return await message.reply_text("Invalid Format\nUsage: `/getfile _id`")
for x in usr_cmd:
try:
myfile = await db.get_file(x)
await message.reply_cached_media(myfile['file_id'])
except FIleNotFound:
await message.reply_text(f"{x} :File Not Found")
|