mirror of
https://gitlab.com/chicken-riders/RcGcDw.git
synced 2025-02-23 00:24:09 +00:00
222 lines
7.3 KiB
Python
222 lines
7.3 KiB
Python
# -*- coding: utf-8 -*-
|
||
|
||
# Recent changes Gamepedia compatible Discord webhook is a project for using a webhook as recent changes page from MediaWiki.
|
||
# Copyright (C) 2018 Frisk
|
||
|
||
# This program is free software: you can redistribute it and/or modify
|
||
# it under the terms of the GNU General Public License as published by
|
||
# the Free Software Foundation, either version 3 of the License, or
|
||
# (at your option) any later version.
|
||
|
||
# This program is distributed in the hope that it will be useful,
|
||
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
# GNU General Public License for more details.
|
||
|
||
# You should have received a copy of the GNU General Public License
|
||
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||
|
||
import json, logging, sys, re
|
||
from html.parser import HTMLParser
|
||
from configloader import settings
|
||
import gettext
|
||
|
||
# Initialize translation
|
||
|
||
t = gettext.translation('misc', localedir='locale', languages=[settings["lang"]])
|
||
_ = t.gettext
|
||
|
||
# Create a custom logger
|
||
|
||
misc_logger = logging.getLogger("rcgcdw.misc")
|
||
|
||
data_template = {"rcid": 99999999999,
|
||
"daily_overview": {"edits": None, "new_files": None, "admin_actions": None, "bytes_changed": None,
|
||
"new_articles": None, "unique_editors": None, "day_score": None, "days_tracked": 0}}
|
||
|
||
|
||
def generate_datafile():
|
||
"""Generate a data.json file from a template."""
|
||
try:
|
||
with open("data.json", 'w') as data:
|
||
data.write(json.dumps(data_template, indent=4))
|
||
except PermissionError:
|
||
misc_logger.critical("Could not create a data file (no permissions). No way to store last edit.")
|
||
sys.exit(1)
|
||
|
||
|
||
def load_datafile() -> object:
|
||
"""Read a data.json file and return a dictionary with contents
|
||
:rtype: object
|
||
"""
|
||
try:
|
||
with open("data.json") as data:
|
||
return json.loads(data.read())
|
||
except FileNotFoundError:
|
||
generate_datafile()
|
||
misc_logger.info("The data file could not be found. Generating a new one...")
|
||
return data_template
|
||
|
||
|
||
def save_datafile(data):
|
||
"""Overwrites the data.json file with given dictionary"""
|
||
try:
|
||
with open("data.json", "w") as data_file:
|
||
data_file.write(json.dumps(data, indent=4))
|
||
except PermissionError:
|
||
misc_logger.critical("Could not modify a data file (no permissions). No way to store last edit.")
|
||
sys.exit(1)
|
||
|
||
|
||
def weighted_average(value, weight, new_value):
|
||
"""Calculates weighted average of value number with weight weight and new_value with weight 1"""
|
||
return round(((value * weight) + new_value) / (weight + 1), 2)
|
||
|
||
|
||
def link_formatter(link):
|
||
"""Formats a link to not embed it"""
|
||
return "<" + re.sub(r"([ )])", "\\\\\\1", link) + ">"
|
||
|
||
|
||
class ContentParser(HTMLParser):
|
||
more = _("\n__And more__")
|
||
current_tag = ""
|
||
small_prev_ins = ""
|
||
small_prev_del = ""
|
||
ins_length = len(more)
|
||
del_length = len(more)
|
||
added = False
|
||
|
||
def handle_starttag(self, tagname, attribs):
|
||
if tagname == "ins" or tagname == "del":
|
||
self.current_tag = tagname
|
||
if tagname == "td" and 'diff-addedline' in attribs[0]:
|
||
self.current_tag = tagname + "a"
|
||
if tagname == "td" and 'diff-deletedline' in attribs[0]:
|
||
self.current_tag = tagname + "d"
|
||
if tagname == "td" and 'diff-marker' in attribs[0]:
|
||
self.added = True
|
||
|
||
def handle_data(self, data):
|
||
data = re.sub(r"([`_*~<>{}@/|\\])", "\\\\\\1", data, 0)
|
||
if self.current_tag == "ins" and self.ins_length <= 1000:
|
||
self.ins_length += len("**" + data + '**')
|
||
if self.ins_length <= 1000:
|
||
self.small_prev_ins = self.small_prev_ins + "**" + data + '**'
|
||
else:
|
||
self.small_prev_ins = self.small_prev_ins + self.more
|
||
if self.current_tag == "del" and self.del_length <= 1000:
|
||
self.del_length += len("~~" + data + '~~')
|
||
if self.del_length <= 1000:
|
||
self.small_prev_del = self.small_prev_del + "~~" + data + '~~'
|
||
else:
|
||
self.small_prev_del = self.small_prev_del + self.more
|
||
if (self.current_tag == "afterins" or self.current_tag == "tda") and self.ins_length <= 1000:
|
||
self.ins_length += len(data)
|
||
if self.ins_length <= 1000:
|
||
self.small_prev_ins = self.small_prev_ins + data
|
||
else:
|
||
self.small_prev_ins = self.small_prev_ins + self.more
|
||
if (self.current_tag == "afterdel" or self.current_tag == "tdd") and self.del_length <= 1000:
|
||
self.del_length += len(data)
|
||
if self.del_length <= 1000:
|
||
self.small_prev_del = self.small_prev_del + data
|
||
else:
|
||
self.small_prev_del = self.small_prev_del + self.more
|
||
if self.added:
|
||
if data == '+' and self.ins_length <= 1000:
|
||
self.ins_length += 1
|
||
if self.ins_length <= 1000:
|
||
self.small_prev_ins = self.small_prev_ins + '\n'
|
||
else:
|
||
self.small_prev_ins = self.small_prev_ins + self.more
|
||
if data == '−' and self.del_length <= 1000:
|
||
self.del_length += 1
|
||
if self.del_length <= 1000:
|
||
self.small_prev_del = self.small_prev_del + '\n'
|
||
else:
|
||
self.small_prev_del = self.small_prev_del + self.more
|
||
self.added = False
|
||
|
||
def handle_endtag(self, tagname):
|
||
if tagname == "ins":
|
||
self.current_tag = "afterins"
|
||
elif tagname == "del":
|
||
self.current_tag = "afterdel"
|
||
else:
|
||
self.current_tag = ""
|
||
|
||
|
||
class LinkParser(HTMLParser):
|
||
new_string = ""
|
||
recent_href = ""
|
||
|
||
def handle_starttag(self, tag, attrs):
|
||
for attr in attrs:
|
||
if attr[0] == 'href':
|
||
self.recent_href = attr[1]
|
||
if self.recent_href.startswith("//"):
|
||
self.recent_href = "https:{rest}".format(rest=self.recent_href)
|
||
elif not self.recent_href.startswith("http"):
|
||
self.recent_href = "https://{wiki}.gamepedia.com".format(wiki=settings["wiki"]) + self.recent_href
|
||
self.recent_href = self.recent_href.replace(")", "\\)")
|
||
|
||
def handle_data(self, data):
|
||
if self.recent_href:
|
||
self.new_string = self.new_string + "[{}](<{}>)".format(data, self.recent_href)
|
||
self.recent_href = ""
|
||
else:
|
||
self.new_string = self.new_string + data
|
||
|
||
def handle_comment(self, data):
|
||
self.new_string = self.new_string + data
|
||
|
||
def handle_endtag(self, tag):
|
||
misc_logger.debug(self.new_string)
|
||
|
||
|
||
def safe_read(request, *keys):
|
||
if request is None:
|
||
return None
|
||
try:
|
||
request = request.json()
|
||
for item in keys:
|
||
request = request[item]
|
||
except KeyError:
|
||
misc_logger.warning(
|
||
"Failure while extracting data from request on key {key} in {change}".format(key=item, change=request))
|
||
return None
|
||
except ValueError:
|
||
misc_logger.warning("Failure while extracting data from request in {change}".format(change=request))
|
||
return None
|
||
return request
|
||
|
||
|
||
def handle_discord_http(code, formatted_embed, result):
|
||
if 300 > code > 199: # message went through
|
||
return 0
|
||
elif code == 400: # HTTP BAD REQUEST result.status_code, data, result, header
|
||
misc_logger.error(
|
||
"Following message has been rejected by Discord, please submit a bug on our bugtracker adding it:")
|
||
misc_logger.error(formatted_embed)
|
||
misc_logger.error(result.text)
|
||
return 1
|
||
elif code == 401 or code == 404: # HTTP UNAUTHORIZED AND NOT FOUND
|
||
misc_logger.error("Webhook URL is invalid or no longer in use, please replace it with proper one.")
|
||
sys.exit(1)
|
||
elif code == 429:
|
||
misc_logger.error("We are sending too many requests to the Discord, slowing down...")
|
||
return 2
|
||
elif 499 < code < 600:
|
||
misc_logger.error(
|
||
"Discord have trouble processing the event, and because the HTTP code returned is {} it means we blame them.".format(
|
||
code))
|
||
return 3
|
||
|
||
|
||
def add_to_dict(dictionary, key):
|
||
if key in dictionary:
|
||
dictionary[key] += 1
|
||
else:
|
||
dictionary[key] = 1
|
||
return dictionary |