mirror of
https://github.com/nichkara/InfinitumBotty.git
synced 2026-06-11 14:41:58 +02:00
BaerbelBox
42 lines
1.4 KiB
Python
42 lines
1.4 KiB
Python
from wikipedia import wikipedia
|
|
from FaustBot.Modules.PrivMsgObserverPrototype import PrivMsgObserverPrototype
|
|
|
|
|
|
class WikiObserver(PrivMsgObserverPrototype):
|
|
@staticmethod
|
|
def cmd():
|
|
return [".w"]
|
|
|
|
@staticmethod
|
|
def help():
|
|
return ".w <term> - fragt Wikipediaartikel zu <term> ab"
|
|
|
|
def update_on_priv_msg(self, data, connection):
|
|
|
|
if data['message'].find('.w ') == -1:
|
|
return
|
|
w = wikipedia.set_lang('de')
|
|
q = data['message'].split(' ')
|
|
query = ''
|
|
for word in q:
|
|
if word.strip() != '.w':
|
|
query += word + ' '
|
|
w = wikipedia.search(query)
|
|
if w.__len__() == 0:
|
|
connection.send_back(data['nick'] + ', ' +
|
|
'ich habe dazu keinen eintrag gefunden!',
|
|
data)
|
|
return
|
|
try:
|
|
page = wikipedia.WikipediaPage(w.pop(0))
|
|
except wikipedia.DisambiguationError as error:
|
|
print('disambiguation page')
|
|
page = wikipedia.WikipediaPage(error.args[1][0])
|
|
connection.send_back(data['nick'] + ' ' + page.url, data)
|
|
index = 51 + page.summary[50:350].rfind('. ')
|
|
if index == 50 or index > 230:
|
|
index = page.summary[0:350].rfind(' ')
|
|
connection.send_back(page.summary[0:index], data)
|
|
else:
|
|
connection.send_back(page.summary[0:index], data)
|