Spaces:
Running
Running
| # Ultroid - UserBot | |
| # Copyright (C) 2021-2026 TeamUltroid | |
| # | |
| # This file is a part of < https://github.com/TeamUltroid/Ultroid/ > | |
| # PLease read the GNU Affero General Public License in | |
| # <https://github.com/TeamUltroid/pyUltroid/blob/main/LICENSE>. | |
| from .. import udB | |
| def get_stuff(key=None): | |
| return udB.get_key(key) or {} | |
| def add_welcome(chat, msg, media, button): | |
| ok = get_stuff("WELCOME") | |
| ok.update({chat: {"welcome": msg, "media": media, "button": button}}) | |
| return udB.set_key("WELCOME", ok) | |
| def get_welcome(chat): | |
| ok = get_stuff("WELCOME") | |
| return ok.get(chat) | |
| def delete_welcome(chat): | |
| ok = get_stuff("WELCOME") | |
| if ok.get(chat): | |
| ok.pop(chat) | |
| return udB.set_key("WELCOME", ok) | |
| def add_goodbye(chat, msg, media, button): | |
| ok = get_stuff("GOODBYE") | |
| ok.update({chat: {"goodbye": msg, "media": media, "button": button}}) | |
| return udB.set_key("GOODBYE", ok) | |
| def get_goodbye(chat): | |
| ok = get_stuff("GOODBYE") | |
| return ok.get(chat) | |
| def delete_goodbye(chat): | |
| ok = get_stuff("GOODBYE") | |
| if ok.get(chat): | |
| ok.pop(chat) | |
| return udB.set_key("GOODBYE", ok) | |
| def add_thanks(chat): | |
| x = get_stuff("THANK_MEMBERS") | |
| x.update({chat: True}) | |
| return udB.set_key("THANK_MEMBERS", x) | |
| def remove_thanks(chat): | |
| x = get_stuff("THANK_MEMBERS") | |
| if x.get(chat): | |
| x.pop(chat) | |
| return udB.set_key("THANK_MEMBERS", x) | |
| def must_thank(chat): | |
| x = get_stuff("THANK_MEMBERS") | |
| return x.get(chat) | |