BinaryONe
commited on
Commit
·
6028c68
1
Parent(s):
11198dc
Changes in Database
Browse files
FileStream/Database/database.py
CHANGED
|
@@ -32,12 +32,12 @@ class Database:
|
|
| 32 |
user = UserSchema(id, details)
|
| 33 |
await self.users.insert_one(user)
|
| 34 |
|
| 35 |
-
async def add_admin(self, id):
|
| 36 |
user= await self.get_user(id)
|
| 37 |
if user:
|
| 38 |
await self.users.update_one({"_id": user['_id']}, {"$set": {"access":"ADMIN" }})
|
| 39 |
else:
|
| 40 |
-
user = UserSchema(id)
|
| 41 |
user['access']="ADMIN"
|
| 42 |
await self.users.insert_one(user)
|
| 43 |
|
|
|
|
| 32 |
user = UserSchema(id, details)
|
| 33 |
await self.users.insert_one(user)
|
| 34 |
|
| 35 |
+
async def add_admin(self, id, details=None):
|
| 36 |
user= await self.get_user(id)
|
| 37 |
if user:
|
| 38 |
await self.users.update_one({"_id": user['_id']}, {"$set": {"access":"ADMIN" }})
|
| 39 |
else:
|
| 40 |
+
user = UserSchema(id,details)
|
| 41 |
user['access']="ADMIN"
|
| 42 |
await self.users.insert_one(user)
|
| 43 |
|
FileStream/bot/plugins/Admin/admin.py
CHANGED
|
@@ -6,18 +6,28 @@ import asyncio
|
|
| 6 |
import aiofiles
|
| 7 |
import datetime
|
| 8 |
|
| 9 |
-
from
|
| 10 |
-
from
|
| 11 |
from FileStream.bot import FileStream
|
|
|
|
| 12 |
from FileStream.Exceptions import FileNotFound
|
| 13 |
from FileStream.config import Telegram, Server
|
| 14 |
-
|
| 15 |
-
from pyrogram.types import Message
|
| 16 |
from pyrogram.enums.parse_mode import ParseMode
|
|
|
|
|
|
|
| 17 |
|
| 18 |
db = Database(Telegram.DATABASE_URL, Telegram.SESSION_NAME)
|
| 19 |
broadcast_ids = {}
|
| 20 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 21 |
|
| 22 |
#Authorize User to Use the Services
|
| 23 |
@FileStream.on_message(filters.command("auth") & filters.private & filters.user(Telegram.OWNER_ID))
|
|
|
|
| 6 |
import aiofiles
|
| 7 |
import datetime
|
| 8 |
|
| 9 |
+
from pyrogram.types import Message
|
| 10 |
+
from pyrogram import filters, Client
|
| 11 |
from FileStream.bot import FileStream
|
| 12 |
+
from FileStream.Database import Database
|
| 13 |
from FileStream.Exceptions import FileNotFound
|
| 14 |
from FileStream.config import Telegram, Server
|
| 15 |
+
|
|
|
|
| 16 |
from pyrogram.enums.parse_mode import ParseMode
|
| 17 |
+
from FileStream.utils.FileProcessors.translation import LANG, BUTTON
|
| 18 |
+
from FileStream.utils.FileProcessors.broadcast_helper import send_msg
|
| 19 |
|
| 20 |
db = Database(Telegram.DATABASE_URL, Telegram.SESSION_NAME)
|
| 21 |
broadcast_ids = {}
|
| 22 |
|
| 23 |
+
@FileStream.on_message(filters.command("admin") & filters.private & filters.user(Telegram.OWNER_ID))
|
| 24 |
+
async def iamadmin(c: Client, m: Message):
|
| 25 |
+
if await db.is_admin(m.from_user.id):
|
| 26 |
+
await m.reply_text(
|
| 27 |
+
text=LANG.ADMIN_GUIDELINES_TEXT_Y.format("/add_user <UserID>","/add_admin <UserID>"),
|
| 28 |
+
parse_mode=ParseMode.MARKDOWN,
|
| 29 |
+
quote=True)
|
| 30 |
+
|
| 31 |
|
| 32 |
#Authorize User to Use the Services
|
| 33 |
@FileStream.on_message(filters.command("auth") & filters.private & filters.user(Telegram.OWNER_ID))
|
FileStream/bot/plugins/FileHandlers/callback.py
CHANGED
|
@@ -58,8 +58,10 @@ async def cb_data(bot: Client, update: CallbackQuery):
|
|
| 58 |
|
| 59 |
elif usr_cmd[0] == "N/A":
|
| 60 |
await update.answer("N/A", True)
|
|
|
|
| 61 |
elif usr_cmd[0] == "close":
|
| 62 |
await update.message.delete()
|
|
|
|
| 63 |
elif usr_cmd[0] == "back":
|
| 64 |
try:
|
| 65 |
user_id = str(usr_cmd[1])
|
|
@@ -86,9 +88,11 @@ async def cb_data(bot: Client, update: CallbackQuery):
|
|
| 86 |
InlineKeyboardButton(
|
| 87 |
"ɴᴏ", callback_data=f"myfile_{usr_cmd[1]}_{usr_cmd[2]}")
|
| 88 |
]]))
|
|
|
|
| 89 |
elif usr_cmd[0] == "msgdelyes":
|
| 90 |
await delete_user_file(usr_cmd[1], int(usr_cmd[2]), update)
|
| 91 |
return
|
|
|
|
| 92 |
elif usr_cmd[0] == "msgdelpvt":
|
| 93 |
await update.message.edit_caption(
|
| 94 |
caption="**Cᴏɴғɪʀᴍ ʏᴏᴜ ᴡᴀɴᴛ ᴛᴏ ᴅᴇʟᴇᴛᴇ ᴛʜᴇ Fɪʟᴇ**\n\n",
|
|
@@ -98,6 +102,7 @@ async def cb_data(bot: Client, update: CallbackQuery):
|
|
| 98 |
InlineKeyboardButton("ɴᴏ",
|
| 99 |
callback_data=f"mainstream_{usr_cmd[1]}")
|
| 100 |
]]))
|
|
|
|
| 101 |
elif usr_cmd[0] == "msgdelpvtyes":
|
| 102 |
await delete_user_filex(usr_cmd[1], update)
|
| 103 |
return
|
|
@@ -111,8 +116,9 @@ async def cb_data(bot: Client, update: CallbackQuery):
|
|
| 111 |
disable_web_page_preview=True,
|
| 112 |
reply_markup=reply_markup,
|
| 113 |
)
|
|
|
|
| 114 |
elif usr_cmd[0] == "pubup":
|
| 115 |
-
|
| 116 |
try:
|
| 117 |
print("Public Upload :",usr_cmd)
|
| 118 |
user_id = str(usr_cmd[1])
|
|
@@ -178,6 +184,7 @@ async def cb_data(bot: Client, update: CallbackQuery):
|
|
| 178 |
chat_id=Telegram.ULOG_GROUP,
|
| 179 |
text=f"An error occurred: {str(e)}"
|
| 180 |
)
|
|
|
|
| 181 |
elif usr_cmd[0] == "tempup":
|
| 182 |
try:
|
| 183 |
print("Public Upload :",usr_cmd)
|
|
|
|
| 58 |
|
| 59 |
elif usr_cmd[0] == "N/A":
|
| 60 |
await update.answer("N/A", True)
|
| 61 |
+
|
| 62 |
elif usr_cmd[0] == "close":
|
| 63 |
await update.message.delete()
|
| 64 |
+
|
| 65 |
elif usr_cmd[0] == "back":
|
| 66 |
try:
|
| 67 |
user_id = str(usr_cmd[1])
|
|
|
|
| 88 |
InlineKeyboardButton(
|
| 89 |
"ɴᴏ", callback_data=f"myfile_{usr_cmd[1]}_{usr_cmd[2]}")
|
| 90 |
]]))
|
| 91 |
+
|
| 92 |
elif usr_cmd[0] == "msgdelyes":
|
| 93 |
await delete_user_file(usr_cmd[1], int(usr_cmd[2]), update)
|
| 94 |
return
|
| 95 |
+
|
| 96 |
elif usr_cmd[0] == "msgdelpvt":
|
| 97 |
await update.message.edit_caption(
|
| 98 |
caption="**Cᴏɴғɪʀᴍ ʏᴏᴜ ᴡᴀɴᴛ ᴛᴏ ᴅᴇʟᴇᴛᴇ ᴛʜᴇ Fɪʟᴇ**\n\n",
|
|
|
|
| 102 |
InlineKeyboardButton("ɴᴏ",
|
| 103 |
callback_data=f"mainstream_{usr_cmd[1]}")
|
| 104 |
]]))
|
| 105 |
+
|
| 106 |
elif usr_cmd[0] == "msgdelpvtyes":
|
| 107 |
await delete_user_filex(usr_cmd[1], update)
|
| 108 |
return
|
|
|
|
| 116 |
disable_web_page_preview=True,
|
| 117 |
reply_markup=reply_markup,
|
| 118 |
)
|
| 119 |
+
|
| 120 |
elif usr_cmd[0] == "pubup":
|
| 121 |
+
|
| 122 |
try:
|
| 123 |
print("Public Upload :",usr_cmd)
|
| 124 |
user_id = str(usr_cmd[1])
|
|
|
|
| 184 |
chat_id=Telegram.ULOG_GROUP,
|
| 185 |
text=f"An error occurred: {str(e)}"
|
| 186 |
)
|
| 187 |
+
|
| 188 |
elif usr_cmd[0] == "tempup":
|
| 189 |
try:
|
| 190 |
print("Public Upload :",usr_cmd)
|
FileStream/bot/plugins/startup.py
CHANGED
|
@@ -42,6 +42,7 @@ async def start(bot: Client, message: Message, response):
|
|
| 42 |
BotCommand("filebank", "ꜰɪʟᴇ ʙᴀɴᴋ"),
|
| 43 |
BotCommand("files", "ᴍʏ ᴘᴜʙʟɪᴄ ꜰɪʟᴇꜱ"),
|
| 44 |
BotCommand("myfiles", "ᴍʏ ᴘʀɪᴠᴀᴛᴇ ꜱᴘᴀᴄᴇ"),
|
|
|
|
| 45 |
])
|
| 46 |
await menu(bot, message)
|
| 47 |
if Telegram.START_PIC:
|
|
@@ -109,13 +110,11 @@ async def start(bot: Client, message: Message, response):
|
|
| 109 |
async def about_handler(bot, message):
|
| 110 |
if Telegram.START_PIC:
|
| 111 |
await message.reply_photo(photo=Telegram.START_PIC,
|
| 112 |
-
caption=LANG.ABOUT_TEXT.format(
|
| 113 |
-
FileStream.fname, __version__),
|
| 114 |
parse_mode=ParseMode.HTML,
|
| 115 |
reply_markup=BUTTON.ABOUT_BUTTONS)
|
| 116 |
else:
|
| 117 |
-
await message.reply_text(text=LANG.ABOUT_TEXT.format(
|
| 118 |
-
FileStream.fname, __version__),
|
| 119 |
disable_web_page_preview=True,
|
| 120 |
reply_markup=BUTTON.ABOUT_BUTTONS)
|
| 121 |
|
|
|
|
| 42 |
BotCommand("filebank", "ꜰɪʟᴇ ʙᴀɴᴋ"),
|
| 43 |
BotCommand("files", "ᴍʏ ᴘᴜʙʟɪᴄ ꜰɪʟᴇꜱ"),
|
| 44 |
BotCommand("myfiles", "ᴍʏ ᴘʀɪᴠᴀᴛᴇ ꜱᴘᴀᴄᴇ"),
|
| 45 |
+
BotCommand("admin", "ɪ ᴀᴍ ᴀᴅᴍɪɴ")
|
| 46 |
])
|
| 47 |
await menu(bot, message)
|
| 48 |
if Telegram.START_PIC:
|
|
|
|
| 110 |
async def about_handler(bot, message):
|
| 111 |
if Telegram.START_PIC:
|
| 112 |
await message.reply_photo(photo=Telegram.START_PIC,
|
| 113 |
+
caption=LANG.ABOUT_TEXT.format(FileStream.fname, __version__),
|
|
|
|
| 114 |
parse_mode=ParseMode.HTML,
|
| 115 |
reply_markup=BUTTON.ABOUT_BUTTONS)
|
| 116 |
else:
|
| 117 |
+
await message.reply_text(text=LANG.ABOUT_TEXT.format(FileStream.fname, __version__),
|
|
|
|
| 118 |
disable_web_page_preview=True,
|
| 119 |
reply_markup=BUTTON.ABOUT_BUTTONS)
|
| 120 |
|
FileStream/utils/FileProcessors/file_properties.py
CHANGED
|
@@ -31,14 +31,14 @@ async def send_file(client: Client, db_id, file_info , message, send_to):
|
|
| 31 |
# quote=True)
|
| 32 |
await log_msg.edit_text(
|
| 33 |
text=
|
| 34 |
-
f"**{file_info['type']} :{file_info['title']}
|
| 35 |
disable_web_page_preview=True,
|
| 36 |
parse_mode=ParseMode.MARKDOWN,
|
| 37 |
)
|
| 38 |
else:
|
| 39 |
await log_msg.edit_text(
|
| 40 |
text=
|
| 41 |
-
f"**{file_info['type']} :{file_info['title']}
|
| 42 |
disable_web_page_preview=True,
|
| 43 |
parse_mode=ParseMode.MARKDOWN,
|
| 44 |
quote=True)
|
|
|
|
| 31 |
# quote=True)
|
| 32 |
await log_msg.edit_text(
|
| 33 |
text=
|
| 34 |
+
f"**{file_info['type']} :{file_info['title']} \nReleaseDate {file_info['release_date']} \n**Description{file_info['description']} \n**Genre {file_info['genre']}\n\nRᴇQᴜᴇꜱᴛᴇᴅ ʙʏ :** {message.from_user.first_name} \n**Cʜᴀɴɴᴇʟ ɪᴅ :** `{message.chat.id}`\n**Fɪʟᴇ ɪᴅ :** `{db_id}`",
|
| 35 |
disable_web_page_preview=True,
|
| 36 |
parse_mode=ParseMode.MARKDOWN,
|
| 37 |
)
|
| 38 |
else:
|
| 39 |
await log_msg.edit_text(
|
| 40 |
text=
|
| 41 |
+
f"**{file_info['type']} :{file_info['title']} \nReleaseDate {file_info['release_date']} \n Description{file_info['description']} \n Genre {file_info['genre']}\n\nRᴇQᴜᴇꜱᴛᴇᴅ ʙʏ :** {message.chat.title} \n**Cʜᴀɴɴᴇʟ ɪᴅ :** `{message.chat.id}`\n**Fɪʟᴇ ɪᴅ :** `{db_id}`",
|
| 42 |
disable_web_page_preview=True,
|
| 43 |
parse_mode=ParseMode.MARKDOWN,
|
| 44 |
quote=True)
|
FileStream/utils/FileProcessors/translation.py
CHANGED
|
@@ -47,7 +47,11 @@ class LANG(object):
|
|
| 47 |
<b>📂 𝖣𝖾𝗋𝗂𝗏𝖾𝖽 𝖭𝖺𝗆𝖾 :</b> <b>{}</b>\n
|
| 48 |
<b>📦 Fɪʟᴇ ꜱɪᴢᴇ :</b> <code>{}</code>\n
|
| 49 |
"""
|
| 50 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 51 |
STREAM_TEXT_X = """
|
| 52 |
<i>Chose where you wand to upload this file !</i>\n
|
| 53 |
<b>📂 Fɪʟᴇ ɴᴀᴍᴇ :</b> <b>{}</b>\n
|
|
|
|
| 47 |
<b>📂 𝖣𝖾𝗋𝗂𝗏𝖾𝖽 𝖭𝖺𝗆𝖾 :</b> <b>{}</b>\n
|
| 48 |
<b>📦 Fɪʟᴇ ꜱɪᴢᴇ :</b> <code>{}</code>\n
|
| 49 |
"""
|
| 50 |
+
ADMIN_GUIDELINES_TEXT_Y = """
|
| 51 |
+
<b>🪂 ✩░▒▓▆▅▃▂▁🅰🅳🅼🅸🅽 🅶🆄🅸🅳🅴🅻🅸🅽🅴🆂▁▂▃▅▆▓▒░✩ 🪂</b>\n
|
| 52 |
+
<b>🔗 ᴄᴏᴍᴍᴀɴᴅ 1:</b> <b>{}</b>\n
|
| 53 |
+
<b>🔗 ᴄᴏᴍᴍᴀɴᴅ 2 :</b> <b>{}</b>\n
|
| 54 |
+
"""
|
| 55 |
STREAM_TEXT_X = """
|
| 56 |
<i>Chose where you wand to upload this file !</i>\n
|
| 57 |
<b>📂 Fɪʟᴇ ɴᴀᴍᴇ :</b> <b>{}</b>\n
|