mirror of
https://github.com/nichkara/InfinitumBotty.git
synced 2026-06-10 22:26:23 +02:00
95 lines
3.8 KiB
Python
95 lines
3.8 KiB
Python
from FaustBot.Communication.Connection import Connection
|
|
from FaustBot.Model.Config import Config
|
|
from FaustBot.Model.GlossaryProvider import GlossaryProvider
|
|
from FaustBot.Modules.PrivMsgObserverPrototype import PrivMsgObserverPrototype
|
|
from FaustBot.Modules.WikiObserver import WikiObserver
|
|
|
|
class GlossaryModule(PrivMsgObserverPrototype):
|
|
@staticmethod
|
|
def cmd():
|
|
return [GlossaryModule._ADD_EXPLANATION,
|
|
GlossaryModule._REMOVE_EXPLANATION,
|
|
GlossaryModule._QUERY_EXPLANATION]
|
|
|
|
@staticmethod
|
|
def help():
|
|
return None
|
|
|
|
_QUERY_EXPLANATION = '.?'
|
|
_REMOVE_EXPLANATION = '.?-'
|
|
_ADD_EXPLANATION = '.?+'
|
|
|
|
def __init__(self, config: Config):
|
|
super().__init__()
|
|
self._config = config
|
|
|
|
def update_on_priv_msg(self, data, connection: Connection):
|
|
msg = data['messageCaseSensitive']
|
|
if msg.startswith(GlossaryModule._REMOVE_EXPLANATION):
|
|
self._remove_query(data, connection)
|
|
elif msg.startswith(GlossaryModule._ADD_EXPLANATION):
|
|
self._add_query(data, connection)
|
|
elif msg.startswith(GlossaryModule._QUERY_EXPLANATION):
|
|
self._answer_query(data, connection)
|
|
|
|
def _answer_query(self, data, connection: Connection):
|
|
"""
|
|
:param data:
|
|
:param connection:
|
|
:return:
|
|
"""
|
|
glossary_provider = GlossaryProvider()
|
|
split = data['message'].split(GlossaryModule._QUERY_EXPLANATION)
|
|
if not len(split) == 2:
|
|
return
|
|
answer = glossary_provider.get_explanation(split[1].strip())
|
|
if answer is None or answer[1] is None or answer[1].strip() == '':
|
|
if split[1].strip() == '':
|
|
return
|
|
# connection.send_back("Tut mir leid, " + data['nick'] + ". Für " + split[1].strip() +
|
|
# " habe ich noch keinen Eintrag. Aber Wikipedia sagt dazu:", data)
|
|
wikiObserver = WikiObserver()
|
|
wikiObserver.config = self.config
|
|
data2 = data.copy()
|
|
data2['message'] = '.w '+split[1]+" \r\n"
|
|
wikiObserver.update_on_priv_msg(data2, connection)
|
|
else:
|
|
connection.send_back(data['nick'] + ": " + split[1] + " - " + answer[1], data)
|
|
|
|
def _remove_query(self, data, connection: Connection):
|
|
"""
|
|
|
|
:param data:
|
|
:param connection:
|
|
:return:
|
|
"""
|
|
if not self._is_idented_mod(data, connection):
|
|
connection.send_back("Dir fehlen die Berechtigungen zum Löschen von Einträgen, " + data['nick'] + ".", data)
|
|
return
|
|
glossary_provider = GlossaryProvider()
|
|
split = data['message'].split(GlossaryModule._REMOVE_EXPLANATION)
|
|
if not len(split) == 2:
|
|
return
|
|
glossary_provider.delete_explanation(split[1])
|
|
connection.send_back("Der Eintrag zu " + split[1] + " wurde gelöscht, " + data['nick'] + ".", data)
|
|
|
|
def _add_query(self, data, connection: Connection):
|
|
"""
|
|
|
|
:param data:
|
|
:param connection:
|
|
:return:
|
|
"""
|
|
if not self._is_idented_mod(data, connection):
|
|
connection.send_back("Dir fehlen leider die Rechte zum Hinzufügen von Einträgen, " + data['nick'] + ".",
|
|
data)
|
|
return
|
|
msg = data['messageCaseSensitive'].split(GlossaryModule._ADD_EXPLANATION)[1].strip()
|
|
split = msg.split(' ', 1)
|
|
glossary_provider = GlossaryProvider()
|
|
glossary_provider.save_or_replace(split[0], split[1])
|
|
connection.send_back(data['nick'] + ": der Eintrag zu " + split[0] + " wurde gespeichert.", data)
|
|
|
|
def _is_idented_mod(self, data: dict, connection: Connection):
|
|
return data['nick'] in self._config.mods and connection.is_idented(data['nick'])
|