159 lines
5.0 KiB
Python
159 lines
5.0 KiB
Python
"""
|
|
Class for interfacing with mediawiki
|
|
"""
|
|
from typing import List
|
|
from urllib.parse import urljoin
|
|
from dataclasses import dataclass
|
|
from wiki_postbot.creds import Mediawiki_Creds
|
|
from wiki_postbot.formats.wiki import WikiPage
|
|
from wiki_postbot.templates.wiki import TemplateMessage
|
|
from wiki_postbot.patterns.wikilink import Wikilink
|
|
from wiki_postbot.actions import Result
|
|
from datetime import datetime
|
|
import requests
|
|
from discord.message import Message, Embed
|
|
import pdb
|
|
|
|
# creds = Mediawiki_Creds.from_json('mediawiki_creds.json')
|
|
|
|
|
|
class Wiki:
|
|
def __init__(self, url:str, api_suffix:str="/api.php", index_page="Discord Messages"):
|
|
self.url = url
|
|
self.api_url = urljoin(self.url, api_suffix)
|
|
self.sess = None
|
|
self.index_page = index_page
|
|
|
|
|
|
def login(self, creds:Mediawiki_Creds):
|
|
# get token to log in
|
|
sess = requests.Session()
|
|
|
|
login_token = sess.get(
|
|
self.api_url,
|
|
params={
|
|
"action":"query",
|
|
"meta":"tokens",
|
|
"type":"login",
|
|
"format":"json"
|
|
},
|
|
verify=False
|
|
).json()['query']['tokens']['logintoken']
|
|
|
|
|
|
login_result = sess.post(
|
|
self.api_url,
|
|
data = {
|
|
"action":"login",
|
|
"lgname":creds.user,
|
|
"lgpassword":creds.password,
|
|
"lgtoken": login_token,
|
|
"format": "json"
|
|
},
|
|
verify=False
|
|
)
|
|
assert login_result.json()['login']['result'] == "Success"
|
|
self.sess = sess
|
|
|
|
def get_page(self, page:str) -> WikiPage:
|
|
|
|
content = self.sess.get(
|
|
self.api_url,
|
|
params={
|
|
'action':'parse',
|
|
'page': page,
|
|
'prop': 'wikitext',
|
|
'formatversion':'2',
|
|
'format':'json'
|
|
}
|
|
).json()
|
|
return WikiPage.from_source(title=content['parse']['title'], source=content['parse']['wikitext'])
|
|
|
|
def insert_text(self, page, section, text):
|
|
|
|
# TODO: Move finding section IDs into the page class!
|
|
page_text = self.get_page(page)
|
|
|
|
sections = page_text.content.get_sections()
|
|
matching_section = -1
|
|
for i, page_section in enumerate(sections):
|
|
if page_section.title is not None and page_section.title.strip().lower() == section.lower():
|
|
matching_section = i
|
|
break
|
|
|
|
|
|
token = self.sess.get(
|
|
self.api_url,
|
|
params={
|
|
"action": "query",
|
|
"meta": "tokens",
|
|
"format": "json"
|
|
},
|
|
verify=False
|
|
).json()['query']['tokens']['csrftoken']
|
|
|
|
if matching_section >= 0:
|
|
print(f'found matching section {matching_section}')
|
|
result = self.sess.post(
|
|
self.api_url,
|
|
data={
|
|
"action":"edit",
|
|
"title":page,
|
|
"section":str(matching_section),
|
|
"appendtext":text,
|
|
"format":"json",
|
|
"token":token
|
|
}
|
|
)
|
|
else:
|
|
print('making new section')
|
|
result = self.sess.post(
|
|
self.api_url,
|
|
data={
|
|
"action":"edit",
|
|
"title":page,
|
|
"section":"new",
|
|
"sectiontitle":section,
|
|
"appendtext":text,
|
|
"format":"json",
|
|
"token":token
|
|
}
|
|
)
|
|
return result
|
|
|
|
def handle_discord(self, msg:Message) -> Result:
|
|
"""
|
|
Not being precious about this, just implementing
|
|
and will worry about generality later!
|
|
"""
|
|
# Get message in mediawiki template formatting
|
|
template_str = TemplateMessage.format_discord(msg)
|
|
|
|
# parse wikilinks, add to each page
|
|
wikilinks = Wikilink.parse(msg.content)
|
|
errored_pages = []
|
|
for link in wikilinks:
|
|
if link.section is None:
|
|
section = "Discord"
|
|
else:
|
|
section = link.section
|
|
|
|
res = self.insert_text(link.link, section, template_str)
|
|
if res.json()['edit']['result'] != 'Success':
|
|
errored_pages.append(res.json())
|
|
|
|
# Add to index page (only once)
|
|
self.add_to_index(template_str)
|
|
|
|
if len(errored_pages) == 0:
|
|
# gather links for a reply
|
|
reply = '\n'.join([f"[{l.link}]({urljoin(self.url, l.link.replace(' ', '_'))})" for l in wikilinks])
|
|
|
|
return Result(ok=True, log=f"Successfully posted message to {[l.link for l in wikilinks]}", reply=reply)
|
|
else:
|
|
return Result(ok=False, log=f"Got exceptions: {errored_pages}")
|
|
|
|
def add_to_index(self, message):
|
|
section = datetime.today().strftime("%y-%m-%d")
|
|
self.insert_text(page=self.index_page, section=section, text=message)
|