|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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) |
|
|
|
