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 FileStream.utils.FileProcessors.broadcast_helper import send_msg
10
- from FileStream.Database import Database
11
  from FileStream.bot import FileStream
 
12
  from FileStream.Exceptions import FileNotFound
13
  from FileStream.config import Telegram, Server
14
- from pyrogram import filters, Client
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']} ReleaseDate {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']} ReleaseDate {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)
 
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