def addwhitelist(update: Update, context: CallbackContext) -> str:
message = update.effective_message
user = update.effective_user
chat = update.effective_chat
bot, args = context.bot, context.args
user_id = extract_user(message, args)
user_member = bot.getChat(user_id)
rt = ""
reply = check_user_id(user_id, bot)
if reply:
message.reply_text(reply)
return ""
with open(ELEVATED_USERS_FILE, "r") as infile:
data = json.load(infile)
if user_id in DEV_USERS:
message.reply_text("Huh? he is more than whitelist!")
return ""
if user_id in SUDO_USERS:
if user.id in DEV_USERS:
rt += "Demoted from sudo to whitelist user"
data["sudos"].remove(user_id)
SUDO_USERS.remove(user_id)
else:
message.reply_text("This user is already sudo")
return ""
if user_id in SUPPORT_USERS:
rt += "Demoted from support user to whitelist user"
data["supports"].remove(user_id)
SUPPORT_USERS.remove(user_id)
if user_id in WHITELIST_USERS:
message.reply_text("This user is already a whitelist user")
return ""
data["whitelists"].append(user_id)
WHITELIST_USERS.append(user_id)
with open(ELEVATED_USERS_FILE, "w") as outfile:
json.dump(data, outfile, indent=4)
update.effective_message.reply_text(
rt + f"\nSuccessfully promoted {user_member.first_name} to whitelist user!"
)
log_message = (
f"#WHITELIST\n"
f"<b>Admin:</b> {mention_html(user.id, html.escape(user.first_name))} \n"
f"<b>User:</b> {mention_html(user_member.id, html.escape(user_member.first_name))}"
)
if chat.type != "private":
log_message = f"<b>{html.escape(chat.title)}:</b>\n" + log_message
return log_message
def addtiger(update: Update, context: CallbackContext) -> str:
message = update.effective_message
user = update.effective_user
chat = update.effective_chat
bot, args = context.bot, context.args
user_id = extract_user(message, args)
user_member = bot.getChat(user_id)
rt = ""
reply = check_user_id(user_id, bot)
if reply:
message.reply_text(reply)
return ""
with open(ELEVATED_USERS_FILE, 'r') as infile:
data = json.load(infile)
if user_id in SUDO_USERS:
rt += "This member is a Dragon Disaster, Demoting to Tiger."
data['sudos'].remove(user_id)
SUDO_USERS.remove(user_id)
if user_id in SUPPORT_USERS:
rt += "This user is already a Demon Disaster, Demoting to Tiger."
data['supports'].remove(user_id)
SUPPORT_USERS.remove(user_id)
if user_id in WHITELIST_USERS:
rt += "This user is already a Wolf Disaster, Demoting to Tiger."
data['whitelists'].remove(user_id)
WHITELIST_USERS.remove(user_id)
if user_id in TIGER_USERS:
message.reply_text("This user is already a Tiger.")
return ""
data['tigers'].append(user_id)
TIGER_USERS.append(user_id)
with open(ELEVATED_USERS_FILE, 'w') as outfile:
json.dump(data, outfile, indent=4)
update.effective_message.reply_text(
rt +
f"\nSuccessfully promoted {user_member.first_name} to a Tiger Disaster!"
)
log_message = (
f"#TIGER\n"
f"<b>Admin:</b> {mention_html(user.id, user.first_name)} \n"
f"<b>User:</b> {mention_html(user_member.id, user_member.first_name)}")
if chat.type != 'private':
log_message = f"<b>{html.escape(chat.title)}:</b>\n" + log_message
return log_message
def removesudo(update: Update, context: CallbackContext) -> str:
message = update.effective_message
user = update.effective_user
chat = update.effective_chat
bot, args = context.bot, context.args
user_id = extract_user(message, args)
user_member = bot.getChat(user_id)
reply = check_user_id(user_id, bot)
if reply:
message.reply_text(reply)
return ""
with open(ELEVATED_USERS_FILE, "r") as infile:
data = json.load(infile)
if user_id in DEV_USERS:
message.reply_text("Huh? he is more than sudo!")
return ""
if user_id in SUDO_USERS:
message.reply_text("Demoted to normal user")
SUDO_USERS.remove(user_id)
data["sudos"].remove(user_id)
with open(ELEVATED_USERS_FILE, "w") as outfile:
json.dump(data, outfile, indent=4)
log_message = (
f"#UNSUDO\n"
f"<b>Admin:</b> {mention_html(user.id, html.escape(user.first_name))}\n"
f"<b>User:</b> {mention_html(user_member.id, html.escape(user_member.first_name))}"
)
if chat.type != "private":
log_message = "<b>{}:</b>\n".format(html.escape(
chat.title)) + log_message
return log_message
else:
message.reply_text("This user is not sudo")
return ""
def removesudo(update: Update, context: CallbackContext) -> str:
message = update.effective_message
user = update.effective_user
chat = update.effective_chat
bot, args = context.bot, context.args
user_id = extract_user(message, args)
user_member = bot.getChat(user_id)
reply = check_user_id(user_id, bot)
if reply:
message.reply_text(reply)
return ""
with open(ELEVATED_USERS_FILE, 'r') as infile:
data = json.load(infile)
if user_id in SUDO_USERS:
message.reply_text("Requested HA to demote this user to Civilian")
SUDO_USERS.remove(user_id)
data['sudos'].remove(user_id)
with open(ELEVATED_USERS_FILE, 'w') as outfile:
json.dump(data, outfile, indent=4)
log_message = (
f"#UNSUDO\n"
f"<b>Admin:</b> {mention_html(user.id, user.first_name)}\n"
f"<b>User:</b> {mention_html(user_member.id, user_member.first_name)}"
)
if chat.type != 'private':
log_message = "<b>{}:</b>\n".format(html.escape(
chat.title)) + log_message
return log_message
else:
message.reply_text("This user is not a Dragon Disaster!")
return ""