mirror of
https://github.com/Nick80835/microbot
synced 2025-08-31 14:38:04 +00:00
remove functions that were moved to ChatWrapper
This commit is contained in:
@@ -100,59 +100,6 @@ class Database():
|
|||||||
chat.save()
|
chat.save()
|
||||||
return chat
|
return chat
|
||||||
|
|
||||||
# custom prefix functions
|
|
||||||
def get_prefix(self, chat_id: int) -> str:
|
|
||||||
return self.get_chat(chat_id).custom_prefix
|
|
||||||
|
|
||||||
def set_prefix(self, chat_id: int, prefix: str):
|
|
||||||
chat = self.get_chat(chat_id)
|
|
||||||
chat.custom_prefix = prefix
|
|
||||||
chat.save()
|
|
||||||
|
|
||||||
# fun command functions
|
|
||||||
def fun_enabled(self, chat_id: int) -> bool:
|
|
||||||
return self.get_chat(chat_id).fun_enabled
|
|
||||||
|
|
||||||
def set_fun(self, chat_id: int, enabled: bool):
|
|
||||||
chat = self.get_chat(chat_id)
|
|
||||||
chat.fun_enabled = enabled
|
|
||||||
chat.save()
|
|
||||||
|
|
||||||
# nsfw command functions
|
|
||||||
def nsfw_enabled(self, chat_id: int) -> bool:
|
|
||||||
return self.get_chat(chat_id).nsfw_enabled
|
|
||||||
|
|
||||||
def set_nsfw(self, chat_id: int, enabled: bool):
|
|
||||||
chat = self.get_chat(chat_id)
|
|
||||||
chat.nsfw_enabled = enabled
|
|
||||||
chat.save()
|
|
||||||
|
|
||||||
# disable/enable command functions
|
|
||||||
@staticmethod
|
|
||||||
def _get_disabled_commands(chat: Chat) -> list:
|
|
||||||
return ujson.loads(chat.disabled_commands)
|
|
||||||
|
|
||||||
def disabled_commands(self, chat_id: int) -> list:
|
|
||||||
return self._get_disabled_commands(self.get_chat(chat_id))
|
|
||||||
|
|
||||||
def enable_command(self, chat_id: int, command: str):
|
|
||||||
chat = self.get_chat(chat_id)
|
|
||||||
disabled_commands = self._get_disabled_commands(chat)
|
|
||||||
|
|
||||||
if command in disabled_commands:
|
|
||||||
disabled_commands.remove(command)
|
|
||||||
chat.disabled_commands = ujson.dumps(disabled_commands)
|
|
||||||
chat.save()
|
|
||||||
|
|
||||||
def disable_command(self, chat_id: int, command: str):
|
|
||||||
chat = self.get_chat(chat_id)
|
|
||||||
disabled_commands = self._get_disabled_commands(chat)
|
|
||||||
|
|
||||||
if command not in disabled_commands:
|
|
||||||
disabled_commands.append(command)
|
|
||||||
chat.disabled_commands = ujson.dumps(disabled_commands)
|
|
||||||
chat.save()
|
|
||||||
|
|
||||||
# sudo functions
|
# sudo functions
|
||||||
@staticmethod
|
@staticmethod
|
||||||
def get_sudo_list() -> list:
|
def get_sudo_list() -> list:
|
||||||
|
Reference in New Issue
Block a user