Added #233 and fixed #234

This commit is contained in:
Frisk 2021-11-18 17:22:16 +01:00
parent b6065a8188
commit b63aadb74d
No known key found for this signature in database
GPG key ID: 213F7C15068AF8AC
4 changed files with 11 additions and 11 deletions

View file

@ -36,6 +36,7 @@
"show_abuselog": false, "show_abuselog": false,
"hide_ips": false, "hide_ips": false,
"discord_message_cooldown": 0, "discord_message_cooldown": 0,
"datafile_path": "data.json",
"auto_suppression": { "auto_suppression": {
"enabled": false, "enabled": false,
"db_location": ":memory:" "db_location": ":memory:"

View file

@ -16,19 +16,18 @@
import json import json
import logging import logging
import sys import sys
global settings global settings
from src.argparser import command_args
def load_settings(): def load_settings():
global settings global settings
try: # load settings try: # load settings
with open("settings.json", encoding="utf8") as sfile: settings = json.load(command_args.settings)
settings = json.load(sfile)
if settings["limitrefetch"] < settings["limit"] and settings["limitrefetch"] != -1: if settings["limitrefetch"] < settings["limit"] and settings["limitrefetch"] != -1:
settings["limitrefetch"] = settings["limit"] settings["limitrefetch"] = settings["limit"]
if "user-agent" in settings["header"]: if "user-agent" in settings["header"]:
settings["header"]["user-agent"] = settings["header"]["user-agent"].format(version="1.14.0.1") # set the version in the useragent settings["header"]["user-agent"] = settings["header"]["user-agent"].format(version="1.14.0.2") # set the version in the useragent
except FileNotFoundError: except FileNotFoundError:
logging.critical("No config file could be found. Please make sure settings.json is in the directory.") logging.critical("No config file could be found. Please make sure settings.json is in the directory.")
sys.exit(1) sys.exit(1)
@ -45,4 +44,3 @@ def load_settings():
load_settings() load_settings()

View file

@ -79,7 +79,7 @@ class DataFile:
if self.changed is False: # don't cause unnecessary write operations if self.changed is False: # don't cause unnecessary write operations
return return
try: try:
with open("data.json", "w", encoding="utf-8") as data_file: with open(settings.get("datafile_path", "data.json"), "w", encoding="utf-8") as data_file:
data_file.write(json.dumps(self.data, indent=4)) data_file.write(json.dumps(self.data, indent=4))
self.changed = False self.changed = False
misc_logger.debug("Saving the database succeeded.") misc_logger.debug("Saving the database succeeded.")

View file

@ -19,16 +19,16 @@
# WARNING! SHITTY CODE AHEAD. ENTER ONLY IF YOU ARE SURE YOU CAN TAKE IT # WARNING! SHITTY CODE AHEAD. ENTER ONLY IF YOU ARE SURE YOU CAN TAKE IT
# You have been warned # You have been warned
import time, logging.config, requests, datetime, math, os.path, schedule, sys, re, importlib import time, logging.config, requests, datetime, math, os.path, schedule, sys, re, importlib, argparse
import src.misc import src.misc
import src.configloader
from collections import defaultdict, Counter, OrderedDict from collections import defaultdict, Counter, OrderedDict
from typing import Optional from typing import Optional
import src.api.client import src.api.client
from src.api.context import Context from src.api.context import Context
from src.api.hooks import formatter_hooks, pre_hooks, post_hooks from src.api.hooks import formatter_hooks, pre_hooks, post_hooks
from src.configloader import settings
from src.misc import add_to_dict, datafile, WIKI_API_PATH, LinkParser, run_hooks from src.misc import add_to_dict, datafile, WIKI_API_PATH, LinkParser, run_hooks
from src.api.util import create_article_path, default_message from src.api.util import create_article_path, default_message
from src.discord.queue import send_to_discord from src.discord.queue import send_to_discord
@ -37,6 +37,7 @@ from src.exceptions import MWError, ServerError, MediaWikiError, BadRequest, Cli
from src.i18n import rcgcdw from src.i18n import rcgcdw
from src.wiki import Wiki from src.wiki import Wiki
settings = src.configloader.settings
_ = rcgcdw.gettext _ = rcgcdw.gettext
ngettext = rcgcdw.ngettext ngettext = rcgcdw.ngettext