check_for_typos.py

"""
Checks all translations of one language for typos.
Runs the corrector on it and prints out the result but doesn't write any changes to the mediawiki system.

Usage:
python3 check_for_typos.py language_code
"""

import argparse
from configparser import ConfigParser
import logging
from os.path import abspath, dirname, join
import sys
from typing import List

from pywikitools.correctbot.bot import CorrectBot


def parse_arguments() -> argparse.Namespace:
    """
    Parses the arguments given from outside

    Returns:
        argparse.Namespace: parsed arguments
    """
    log_levels: List[str] = ['debug', 'info', 'warning', 'error']

    parser = argparse.ArgumentParser()
    parser.add_argument("language_code", help="Language code")
    parser.add_argument("-l", "--loglevel", choices=log_levels, default="warning", help="set loglevel for the script")
    parser.add_argument("--only", help="Only apply the correction rule with the specified method name")
    return parser.parse_args()


if __name__ == "__main__":
    args = parse_arguments()
    root = logging.getLogger()
    root.setLevel(logging.DEBUG)
    sh = logging.StreamHandler(sys.stdout)
    fformatter = logging.Formatter('%(levelname)s: %(message)s')
    sh.setFormatter(fformatter)
    numeric_level = getattr(logging, args.loglevel.upper(), None)
    assert isinstance(numeric_level, int)
    sh.setLevel(numeric_level)
    root.addHandler(sh)

    config = ConfigParser()
    config.read(join(dirname(abspath(__file__)), "config.ini"))
    correctbot = CorrectBot(config, simulate=True)

    apply_only_rule = None
    if args.only is not None:
        apply_only_rule = str(args.only)

    for worksheet in correctbot.fortraininglib.get_worksheet_list():
        correctbot.check_page(worksheet, args.language_code, apply_only_rule)
        print(f"{worksheet}: {correctbot.get_correction_counter()} corrections")
        if correctbot.get_correction_counter() > 0 or correctbot.get_suggestion_counter() > 0:
            print(correctbot.get_correction_stats())
            print(correctbot.get_correction_diff())
            print(correctbot.get_suggestion_stats())
            print(correctbot.get_suggestion_diff())
        if correctbot.get_warning_counter() > 0:
            print(correctbot.get_warnings())