Spaces:
Running
Running
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
|