diff --git a/README.md b/README.md index 1c09fab..76278f6 100644 --- a/README.md +++ b/README.md @@ -60,6 +60,23 @@ You can see config file with all the comments below: "size": 512, "location": "logs" }, + "features": { + "applications": { + "enabled": true + }, + "sponsorships": { + "enabled": true + }, + "warnings": { + "enabled": false + }, + "invites_check": { + "enabled": true + }, + "dinovoice": { + "enabled": false + } + }, "scheduler": { "birthdays": { "time": 9, @@ -85,6 +102,119 @@ You can see config file with all the comments below: "locations": { "cache": "cache", "locale": "locale" + }, + "commands": { + "rules": { + "permissions": [ + "users", + "admins" + ], + "modules": [ + "general" + ] + }, + "cancel": { + "permissions": [ + "users", + "admins" + ], + "modules": [ + "applications", + "sponsorships" + ] + }, + "nearby": { + "permissions": [ + "users", + "admins", + "group_users", + "group_admins" + ], + "modules": [ + "applications" + ] + }, + "warn": { + "permissions": [ + "group_users" + ], + "modules": [ + "warnings" + ] + }, + "reapply": { + "permissions": [ + "users", + "admins" + ], + "modules": [ + "applications" + ] + }, + "sponsorship": { + "permissions": [ + "users", + "admins" + ], + "modules": [ + "sponsorships" + ] + }, + "reboot": { + "permissions": [ + "admins", + "group_admins" + ], + "modules": [ + "general" + ] + }, + "label": { + "permissions": [ + "admins", + "group_admins" + ], + "modules": [ + "applications" + ] + }, + "message": { + "permissions": [ + "admins", + "group_admins" + ], + "modules": [ + "general" + ] + }, + "identify": { + "permissions": [ + "admins", + "group_admins" + ], + "modules": [ + "applications", + "sponsorships" + ] + }, + "application": { + "permissions": [ + "admins", + "group_admins" + ], + "modules": [ + "applications" + ] + }, + "applications": { + "permissions": [ + "admins", + "group_admins" + ], + "modules": [ + "applications" + ] + } } } ``` @@ -99,4 +229,4 @@ After all of that you're good to go! Happy using :) * [x] Check sponsorship on Holo girls * [x] /nearby command * [x] Complete messenger between user and admins -* [x] Get application by id and user_id \ No newline at end of file +* [x] Get application by id and user_id diff --git a/config_example.json b/config_example.json index f1bdd55..2861396 100644 --- a/config_example.json +++ b/config_example.json @@ -72,5 +72,118 @@ "locations": { "cache": "cache", "locale": "locale" + }, + "commands": { + "rules": { + "permissions": [ + "users", + "admins" + ], + "modules": [ + "general" + ] + }, + "cancel": { + "permissions": [ + "users", + "admins" + ], + "modules": [ + "applications", + "sponsorships" + ] + }, + "nearby": { + "permissions": [ + "users", + "admins", + "group_users", + "group_admins" + ], + "modules": [ + "applications" + ] + }, + "warn": { + "permissions": [ + "group_users" + ], + "modules": [ + "warnings" + ] + }, + "reapply": { + "permissions": [ + "users", + "admins" + ], + "modules": [ + "applications" + ] + }, + "sponsorship": { + "permissions": [ + "users", + "admins" + ], + "modules": [ + "sponsorships" + ] + }, + "reboot": { + "permissions": [ + "admins", + "group_admins" + ], + "modules": [ + "general" + ] + }, + "label": { + "permissions": [ + "admins", + "group_admins" + ], + "modules": [ + "applications" + ] + }, + "message": { + "permissions": [ + "admins", + "group_admins" + ], + "modules": [ + "general" + ] + }, + "identify": { + "permissions": [ + "admins", + "group_admins" + ], + "modules": [ + "applications", + "sponsorships" + ] + }, + "application": { + "permissions": [ + "admins", + "group_admins" + ], + "modules": [ + "applications" + ] + }, + "applications": { + "permissions": [ + "admins", + "group_admins" + ], + "modules": [ + "applications" + ] + } } } \ No newline at end of file diff --git a/locale/uk.json b/locale/uk.json index 2ead0de..c5674a8 100644 --- a/locale/uk.json +++ b/locale/uk.json @@ -238,33 +238,18 @@ ], "rules_additional": "Додаткові правила, які несуть рекомендаційний характер, та не мають явних покарань за порушення:\n1️⃣) У чаті немає заборони на російську мову. Ми поважаємо кожного українця і не бажаємо розпалювати мовні конфлікти.\n2️⃣) У чаті немає заборони на російський контент. Але, майте на увазі, що учасники, здебільшого, не будуть зацікавлені у тому, щоб обговорювати його і він може бути проігнорованим.\n3️⃣) Не зловживайте матами. Намагайтесь спілкуватись чистою мовою.\n4️⃣) Поважайте авторські права контентмейкерів. Якщо ви знаходите арт, анімацію, музику тощо, на офіційних ресурсах (pixiv, twitter, deviantart тощо), відправляйте на нього посилання.\nЯкщо хтось із учасників відправив арт із не офіційного ресурсу і ви бажаєте дізнатись його автора, відправте у відповідь повідомлення із текстом `/search` на повідомлення із артом.", "commands": { - "rules": "Правила спільноти", + "application": "Переглянути анкету користувача", + "applications": "Отримати всі анкети як JSON", "cancel": "Відмінити актуальну дію", + "identify": "Дізнатись дані про користувача за айді", + "label": "Встановити нікнейм користувачу", + "message": "Надіслати користувачу повідомлення", "nearby": "Показати користувачів поблизу", "reapply": "Повторно заповнити анкету", - "sponsorship": "Отримати роль за спонсорство" - }, - "commands_admin": { "reboot": "Перезапустити бота", - "label": "Встановити нікнейм користувачу", - "message": "Надіслати користувачу повідомлення", - "identify": "Дізнатись дані про користувача за айді", - "warnings": "Переглянути попередження користувача", - "application": "Переглянути анкету користувача", - "applications": "Отримати всі анкети як JSON" - }, - "commands_group_admin": { - "reboot": "Перезапустити бота", - "label": "Встановити нікнейм користувачу", - "nearby": "Показати користувачів поблизу", - "message": "Надіслати користувачу повідомлення", - "identify": "Дізнатись дані про користувача за айді", - "warnings": "Переглянути попередження користувача", - "application": "Переглянути анкету користувача", - "applications": "Отримати всі анкети як JSON" - }, - "commands_group_destination": { + "rules": "Правила спільноти", + "sponsorship": "Отримати роль за спонсорство", "warn": "Попередити користувача", - "nearby": "Показати користувачів поблизу" + "warnings": "Переглянути попередження користувача" } } \ No newline at end of file diff --git a/modules/scheduled.py b/modules/scheduled.py index ce986ef..b5ea0c2 100644 --- a/modules/scheduled.py +++ b/modules/scheduled.py @@ -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.") \ No newline at end of file