Merge pull request subinps#34 from subinps/develop

live quality change

plugins/callback.py

@@ -487,9 +487,10 @@ async def cb_handler(client: Client, query: CallbackQuery):
  Config.IS_VIDEO = set_config(Config.IS_VIDEO)
  await query.message.edit_reply_markup(reply_markup=await settings_panel())
  data=Config.DATA.get('FILE_DATA')
- if not data.get('dur', 0) or \
- data.get('dur') == 0:
+ if not data \
+ or data.get('dur', 0) == 0:
  await restart_playout()
+ return
  k, reply = await seek_file(0)
  if k == False:
  await restart_playout()

plugins/commands.py

@@ -41,12 +41,14 @@
  is_admin, 
  chat_filter,
  sudo_filter,
- delete_messages
+ delete_messages,
+ seek_file
 )
 from pyrogram import (
  Client, 
  filters
 )
+
 IST = pytz.timezone(Config.TIME_ZONE)
 if Config.DATABASE_URI:
  from utils import db
@@ -249,7 +251,7 @@ async def set_heroku_var(client, message):
  await delete_messages([message, m])
  return
 
- if Config.DATABASE_URI and var in ["STARTUP_STREAM", "CHAT", "LOG_GROUP", "REPLY_MESSAGE", "DELAY", "RECORDING_DUMP"]: 
+ if Config.DATABASE_URI and var in ["STARTUP_STREAM", "CHAT", "LOG_GROUP", "REPLY_MESSAGE", "DELAY", "RECORDING_DUMP", "QUALITY"]: 
  await m.edit("Mongo DB Found, Setting up config vars...")
  await asyncio.sleep(2) 
  if not value:
@@ -264,13 +266,27 @@ async def set_heroku_var(client, message):
  await delete_messages([message, m]) 
  return
  else:
- if var in ["CHAT", "LOG_GROUP", "RECORDING_DUMP"]:
+ if var in ["CHAT", "LOG_GROUP", "RECORDING_DUMP", "QUALITY"]:
  try:
  value=int(value)
  except:
- await m.edit("You should give me a chat id . It should be an interger.")
- await delete_messages([message, m])
- return
+ if var == "QUALITY":
+ if not value.lower() in ["low", "medium", "high"]:
+ await m.edit("You should specify a value between 10 - 100.")
+ await delete_messages([message, m])
+ return
+ else:
+ value = value.lower()
+ if value == "high":
+ value = 100
+ elif value == "medium":
+ value = 66.9
+ elif value == "low":
+ value = 50
+ else:
+ await m.edit("You should give me a chat id . It should be an interger.")
+ await delete_messages([message, m])
+ return
  if var == "CHAT":
  await leave_call()
  Config.ADMIN_CACHE=False
@@ -280,6 +296,16 @@ async def set_heroku_var(client, message):
  Config.CHAT=int(value)
  await restart()
  await edit_config(var, int(value))
+ if var == "QUALITY":
+ if Config.CALL_STATUS:
+ data=Config.DATA.get('FILE_DATA')
+ if not data \
+ or data.get('dur', 0) == 0:
+ await restart_playout()
+ return
+ k, reply = await seek_file(0)
+ if k == False:
+ await restart_playout()
  await m.edit(f"Succesfully changed {var} to {value}")
  await delete_messages([message, m])
  return

utils/utils.py

@@ -1610,6 +1610,7 @@ async def sync_to_db():
  await db.edit_config("RECORDING_DUMP", Config.RECORDING_DUMP)
  await db.edit_config("RECORDING_TITLE", Config.RECORDING_TITLE)
  await db.edit_config("HAS_SCHEDULE", Config.HAS_SCHEDULE)
+ await db.edit_config("CUSTOM_QUALITY", Config.CUSTOM_QUALITY)
 
 
 async def sync_from_db():
@@ -1638,6 +1639,7 @@ async def sync_from_db():
  Config.RECORDING_DUMP = await db.get_config("RECORDING_DUMP")
  Config.RECORDING_TITLE = await db.get_config("RECORDING_TITLE")
  Config.HAS_SCHEDULE = await db.get_config("HAS_SCHEDULE")
+ Config.CUSTOM_QUALITY = await db.get_config("CUSTOM_QUALITY")
 
 async def add_to_db_playlist(song):
  if Config.DATABASE_URI:
@@ -1696,6 +1698,8 @@ async def check_db():
  db.add_config("RECORDING_TITLE", Config.RECORDING_TITLE)
  if not await db.is_saved('HAS_SCHEDULE'):
  db.add_config("HAS_SCHEDULE", Config.HAS_SCHEDULE)
+ if not await db.is_saved("CUSTOM_QUALITY"):
+ db.add_config("CUSTOM_QUALITY", Config.CUSTOM_QUALITY)
 
 
 async def is_audio(file):
@@ -1876,6 +1880,8 @@ async def edit_config(var, value):
  Config.REPLY_MESSAGE = value
  elif var == "RECORDING_DUMP":
  Config.RECORDING_DUMP = value
+ elif var == "QUALITY":
+ Config.CUSTOM_QUALITY = value
  await sync_to_db()