Changed the way how commands work
This commit is contained in:
Profitroll
@@ -50,113 +50,129 @@ if configGet("enabled", "scheduler", "cache_avatars"):
|
||||
|
||||
|
||||
# Check for birthdays
|
||||
if configGet("enabled", "scheduler", "birthdays"):
|
||||
@scheduler.scheduled_job(trigger="cron", hour=configGet("time", "scheduler", "birthdays"))
|
||||
async def check_birthdays():
|
||||
for entry in col_applications.find():
|
||||
if entry["application"]["2"].strftime("%d.%m") == datetime.now().strftime("%d.%m"):
|
||||
try:
|
||||
tg_user = await app.get_users(entry["user"])
|
||||
await app.send_message( configGet("admin_group"), locale("birthday", "message").format(str(tg_user.first_name), str(tg_user.username), str(relativedelta(datetime.now(), entry["application"]["2"], '%d.%m.%Y').years)) ) # type: ignore
|
||||
logWrite(f"Notified admins about {entry['user']}'s birthday")
|
||||
except Exception as exp:
|
||||
logWrite(f"Could not find user {entry['user']} to send a message about birthday due to '{exp}'")
|
||||
continue
|
||||
logWrite("Birthdays check performed")
|
||||
if configGet("enabled", "features", "applications") is True:
|
||||
if configGet("enabled", "scheduler", "birthdays") is True:
|
||||
@scheduler.scheduled_job(trigger="cron", hour=configGet("time", "scheduler", "birthdays"))
|
||||
async def check_birthdays():
|
||||
for entry in col_applications.find():
|
||||
if entry["application"]["2"].strftime("%d.%m") == datetime.now().strftime("%d.%m"):
|
||||
try:
|
||||
tg_user = await app.get_users(entry["user"])
|
||||
await app.send_message( configGet("admin_group"), locale("birthday", "message").format(str(tg_user.first_name), str(tg_user.username), str(relativedelta(datetime.now(), entry["application"]["2"], '%d.%m.%Y').years)) ) # type: ignore
|
||||
logWrite(f"Notified admins about {entry['user']}'s birthday")
|
||||
except Exception as exp:
|
||||
logWrite(f"Could not find user {entry['user']} to send a message about birthday due to '{exp}'")
|
||||
continue
|
||||
logWrite("Birthdays check performed")
|
||||
|
||||
|
||||
# Check for expired sponsorships
|
||||
if configGet("enabled", "scheduler", "sponsorships"):
|
||||
@scheduler.scheduled_job(trigger="cron", hour=configGet("time", "scheduler", "sponsorships"))
|
||||
async def check_sponsors():
|
||||
for entry in col_sponsorships.find({"sponsorship.expires": {"$lt": datetime.now()+timedelta(days=2)}}):
|
||||
try:
|
||||
tg_user = await app.get_users(entry["user"])
|
||||
until_expiry = relativedelta(datetime.now(), entry["sponsorship"]["expires"]).days
|
||||
await app.send_message( tg_user, locale("sponsorships_expires", "message").format(until_expiry) ) # type: ignore
|
||||
logWrite(f"Notified user that sponsorship expires in {until_expiry} days")
|
||||
except Exception as exp:
|
||||
logWrite(f"Could not find user {entry['user']} notify about sponsorship expiry due to '{exp}'")
|
||||
continue
|
||||
for entry in col_sponsorships.find({"sponsorship.expires": {"$lt": datetime.now()}}):
|
||||
try:
|
||||
holo_user = HoloUser(entry["user"])
|
||||
await app.send_message( entry["user"], locale("sponsorships_expired", "message") ) # type: ignore
|
||||
await holo_user.label_reset(configGet("destination_group"))
|
||||
col_sponsorships.find_one_and_delete({"user": holo_user.id})
|
||||
if configGet("enabled", "features", "sponsorships") is True:
|
||||
if configGet("enabled", "scheduler", "sponsorships") is True:
|
||||
@scheduler.scheduled_job(trigger="cron", hour=configGet("time", "scheduler", "sponsorships"))
|
||||
async def check_sponsors():
|
||||
for entry in col_sponsorships.find({"sponsorship.expires": {"$lt": datetime.now()+timedelta(days=2)}}):
|
||||
try:
|
||||
tg_user = await app.get_users(entry["user"])
|
||||
logWrite(f"Notified user that sponsorship expired")
|
||||
until_expiry = relativedelta(datetime.now(), entry["sponsorship"]["expires"]).days
|
||||
await app.send_message( tg_user, locale("sponsorships_expires", "message").format(until_expiry) ) # type: ignore
|
||||
logWrite(f"Notified user that sponsorship expires in {until_expiry} days")
|
||||
except Exception as exp:
|
||||
logWrite(f"Could not find user {entry['user']} notify about sponsorship expired due to '{exp}'")
|
||||
except Exception as exp:
|
||||
logWrite(f"Could not reset label of user {entry['user']} due to '{exp}'")
|
||||
continue
|
||||
logWrite("Sponsorships check performed")
|
||||
logWrite(f"Could not find user {entry['user']} notify about sponsorship expiry due to '{exp}'")
|
||||
continue
|
||||
for entry in col_sponsorships.find({"sponsorship.expires": {"$lt": datetime.now()}}):
|
||||
try:
|
||||
holo_user = HoloUser(entry["user"])
|
||||
await app.send_message( entry["user"], locale("sponsorships_expired", "message") ) # type: ignore
|
||||
await holo_user.label_reset(configGet("destination_group"))
|
||||
col_sponsorships.find_one_and_delete({"user": holo_user.id})
|
||||
try:
|
||||
tg_user = await app.get_users(entry["user"])
|
||||
logWrite(f"Notified user that sponsorship expired")
|
||||
except Exception as exp:
|
||||
logWrite(f"Could not find user {entry['user']} notify about sponsorship expired due to '{exp}'")
|
||||
except Exception as exp:
|
||||
logWrite(f"Could not reset label of user {entry['user']} due to '{exp}'")
|
||||
continue
|
||||
logWrite("Sponsorships check performed")
|
||||
|
||||
|
||||
# Register all bot commands
|
||||
@scheduler.scheduled_job(trigger="date", run_date=datetime.now()+timedelta(seconds=3))
|
||||
async def commands_register():
|
||||
|
||||
commands = {
|
||||
"users": [],
|
||||
"admins": [],
|
||||
"group_users": [],
|
||||
"group_admins": [],
|
||||
"locales": {}
|
||||
}
|
||||
|
||||
valid_locales = []
|
||||
files_locales = listdir(f'{configGet("locale", "locations")}')
|
||||
|
||||
for entry in files_locales:
|
||||
if entry.endswith(".json"):
|
||||
valid_locales.append(".".join(entry.split(".")[:-1]))
|
||||
commands["locales"][".".join(entry.split(".")[:-1])] = {
|
||||
"users": [],
|
||||
"admins": [],
|
||||
"group_users": [],
|
||||
"group_admins": []
|
||||
}
|
||||
|
||||
config_modules = configGet("features")
|
||||
config_commands = configGet("commands")
|
||||
|
||||
|
||||
for command in config_commands:
|
||||
|
||||
enabled = False
|
||||
|
||||
for module in config_commands[command]["modules"]:
|
||||
if config_modules[module] == module:
|
||||
enabled = True
|
||||
|
||||
if enabled is False:
|
||||
continue
|
||||
|
||||
for permission in config_commands[command]["permissions"]:
|
||||
commands[permission].append(BotCommand(command, locale("commands")[command]))
|
||||
for lc in valid_locales:
|
||||
commands["locales"][lc][permission].append(BotCommand(command, locale("commands", locale=lc)[command]))
|
||||
|
||||
|
||||
# Registering user commands
|
||||
commands_list = []
|
||||
for command in locale("commands"):
|
||||
commands_list.append(BotCommand(command, locale("commands")[command]))
|
||||
await app.set_bot_commands(commands_list)
|
||||
await app.set_bot_commands(commands["users"])
|
||||
logWrite("Registered user commands for default locale")
|
||||
|
||||
# Registering user commands for each locale
|
||||
for lc in valid_locales:
|
||||
commands_list = []
|
||||
for command in locale("commands", locale=lc):
|
||||
commands_list.append(BotCommand(command, locale("commands",locale=lc)[command]))
|
||||
await app.set_bot_commands(commands_list, language_code=lc)
|
||||
await app.set_bot_commands(commands["locales"][lc]["users"], language_code=lc)
|
||||
logWrite(f"Registered user commands for locale {lc}")
|
||||
|
||||
# Registering admin commands
|
||||
commands_admin_list = []
|
||||
for command in locale("commands"):
|
||||
commands_admin_list.append(BotCommand(command, locale("commands")[command]))
|
||||
|
||||
for command in locale("commands_admin"):
|
||||
commands_admin_list.append(BotCommand(command, locale("commands_admin")[command]))
|
||||
|
||||
for admin in configGet("admins"):
|
||||
# Registering admin/owner commands
|
||||
for admin in configGet("admins")+[configGet("owner")]:
|
||||
try:
|
||||
await app.set_bot_commands(commands_admin_list, scope=BotCommandScopeChat(chat_id=admin))
|
||||
logWrite(f"Registered admin commands for admin {admin}")
|
||||
await app.set_bot_commands(commands["admins"], scope=BotCommandScopeChat(chat_id=admin))
|
||||
if admin == configGet("owner"):
|
||||
logWrite(f"Registered admin commands for owner {configGet('owner')}")
|
||||
else:
|
||||
logWrite(f"Registered admin commands for admin {admin}")
|
||||
except bad_request_400.PeerIdInvalid:
|
||||
pass
|
||||
|
||||
try:
|
||||
await app.set_bot_commands(commands_admin_list, scope=BotCommandScopeChat(chat_id=configGet("owner")))
|
||||
logWrite(f"Registered admin commands for owner {configGet('owner')}")
|
||||
except bad_request_400.PeerIdInvalid:
|
||||
logWrite(f"Could not register commands for bot owner. Perhaps user has not started the bot yet.")
|
||||
|
||||
# Registering admin group commands
|
||||
commands_group_admin_list = []
|
||||
for command in locale("commands_group_admin"):
|
||||
commands_group_admin_list.append(BotCommand(command, locale("commands_group_admin")[command]))
|
||||
try:
|
||||
await app.set_bot_commands(commands_group_admin_list, scope=BotCommandScopeChat(chat_id=configGet("admin_group")))
|
||||
await app.set_bot_commands(commands["group_admins"], scope=BotCommandScopeChat(chat_id=configGet("admin_group")))
|
||||
logWrite("Registered admin group commands for default locale")
|
||||
except bad_request_400.ChannelInvalid:
|
||||
logWrite(f"Could not register commands for admin group. Bot is likely not in the group.")
|
||||
|
||||
# Registering destination group commands
|
||||
commands_group_destination_list = []
|
||||
for command in locale("commands_group_destination"):
|
||||
commands_group_destination_list.append(BotCommand(command, locale("commands_group_destination")[command]))
|
||||
try:
|
||||
await app.set_bot_commands(commands_group_destination_list, scope=BotCommandScopeChat(chat_id=configGet("destination_group")))
|
||||
await app.set_bot_commands(commands["group_users"], scope=BotCommandScopeChat(chat_id=configGet("destination_group")))
|
||||
logWrite("Registered destination group commands")
|
||||
except bad_request_400.ChannelInvalid:
|
||||
logWrite(f"Could not register commands for destination group. Bot is likely not in the group.")
|
||||
Reference in New Issue
Block a user